Attention neural network accelerator and acceleration method

By employing multi-modal prediction and sparse workload balancing techniques in attention neural networks, the problem of wasted computational and hardware resources caused by the sparsity of prediction data in existing methods is solved, achieving efficient neural network acceleration and improving speed and energy efficiency.

CN122414263APending Publication Date: 2026-07-17XI AN JIAOTONG UNIV
View PDF 0 Cites 0 Cited by

Patent Information

Application Number
CN202610540631.5
Authority / Receiving Office
CN · China
Patent Type
Applications(China)
Current Assignee / Owner
Filing Date
2026-04-22
Publication Date
2026-07-17

AI Technical Summary

Technical Problem

Existing methods consume additional computational and hardware resources due to the sparsity of the predicted data, hindering the efficient deployment of attention-based neural networks on resource-constrained devices.

Method used

A multi-mode prediction unit is used to perform bitwise AND operations on the most significant mantissa of the input feature matrix and weight matrix using AND gates and adders to generate estimates and a skip mask. Combined with a sparse workload balancing unit and a precise computation unit, a unified acceleration of the attention computation layer and the linear layer is achieved.

Benefits of technology

It reduces the workload of the precision computing unit, lowers computational complexity and hardware resource consumption, improves inference speed and energy efficiency, and maintains the accuracy of the model.

✦ Generated by Eureka AI based on patent content.

Smart Images

  • Figure CN122414263A_ABST
    Figure CN122414263A_ABST
Patent Text Reader

Abstract

本发明公开了一种注意力神经网络加速器及加速方法,涉及人工智能与机器学习技术领域,包括:多模式预测单元,通过与门和加法器对注意力计算层和线性层中的输入特征矩阵和权重矩阵的尾数最高有效位进行按位与运算和累加,得到多个估计值;并基于估计值生成对应的跳过掩码;稀疏工作负载均衡单元,基于跳过掩码生成一个包含未跳过的乘法计算及其原始相对位置的压缩列表;精确计算单元,执行压缩列表中未跳过的乘法运算;加法树单元,用于对输出矩阵的部分和进行累加,得到输出特征矩阵。本发明能够精准识别并跳过对最终输出贡献较小的乘法运算,从而减少了实际工作负载,解决了现有方法通常会因预测数据稀疏性而消耗额外的计算和硬件资源的问题。
Need to check novelty before this filing date? Find Prior Art