🔥 今日值得一读 FPGA上首个纳秒级LUT感知Transformer,准确率涨1.8个点、LUT省2.9倍,35纳秒搞定分类!
LEAF: LUT-aware Efficient Attention on FPGAs
arXiv LG (cs.LG) 🔥 重点 #FPGA#注意力机制#硬件加速 🕐 10-21 12:00
👨‍💼 主理人解读 · 为什么值得关注
可在FPGA上部署纳秒级低延迟Transformer注意力,适用于CERN等实时触发系统。

📖 AI 总结

文章提出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高能物理中的一项任务,用于识别粒子对撞产生的喷注类型,常作为机器学习基准。
infoAI 公众号二维码

📱 每天一份 AI 前沿日报

关注公众号,每天 09:00 推送 · 不错过任何重磅