文章提出LEAF,一种面向FPGA的查找表(LUT)感知高效注意力机制,是首个纳秒级LUT感知Transformer。针对Transformer注意力机制硬件实现成本高、难以兼顾模型规模与亚微秒实时需求(如CERN场景)的问题,LEAF通过LUT感知训练将查询、键、值投影及输出前馈层转化为FPGA原生层,并仅对注意力块中权重恒定的矩阵-向量乘法做表化处理,其余部分保留算术形式,从而绕开注意力因两个激活相乘而难以表化的障碍。在标准喷注标记基准上,其准确率提升1.8个百分点,同时LUT用量比此前同类亚微秒Transformer减少2.9倍;最高效的64粒子输入linformer在28k LUT、35纳秒下达到80.4%分类准确率。该工作为超低延迟Transformer推理提供了可行的硬件设计路径。
| LEAF | 一种面向FPGA的查找表感知高效注意力机制,是首个纳秒级LUT感知Transformer。 |
| LUT-aware training | 一种训练方法,使模型权重适应FPGA查找表(LUT)的实现方式,以优化硬件效率。 |
| Linformer | 一种Transformer变体,通过低秩近似降低注意力机制的计算复杂度。 |
| FPGA | 现场可编程门阵列,一种可编程硬件设备,常用于低延迟推理加速。 |
| Jet tagging | 高能物理中的一项任务,用于识别粒子对撞产生的喷注类型,常作为机器学习基准。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅