在FPGA上实现亚微秒级Transformer,实现实时粒子对撞机触发
Sub-microsecond Transformers for Jet Tagging on FPGAs
- 用高精度量化与分布式算术优化,将完整Transformer部署到单个FPGA
- 达到100纳秒量级延迟,性能优于传统基线模型
- 开源支持多头与线性注意力,推动高速机器学习应用
我们首次在FPGA上实现了亚微秒级的Transformer,用于欧洲核子研究中心大型强子对撞机(LHC)的粒子喷注分类任务,取得了与当前先进方法相当的性能。尽管Transformer在现代机器学习中表现优异,但其计算复杂度长期限制了其在实时系统中的应用,如对撞机实验的硬件触发系统。本工作首次将Transformer应用于FPGA上的喷注分类,通过高粒度量化和分布式算术优化,使整个模型可部署于单个FPGA,实现约100纳秒的延迟,同时保持优异性能。我们还向hls4ml添加了多头注意力和线性注意力支持,使该成果可被更广泛的快速机器学习社区使用。这项工作推进了高亮度LHC下一代触发系统的发展,为高能物理及其他领域实现实时Transformer应用铺平道路。
原文摘要 · Abstract (English)
We present the first sub-microsecond transformer implementation on an FPGA achieving competitive performance for state-of-the-art high-energy physics benchmarks. Transformers have shown exceptional performance on multiple tasks in modern machine learning applications, including jet tagging at the CERN Large Hadron Collider (LHC). However, their computational complexity prohibits use in real-time applications, such as the hardware trigger system of the collider experiments up until now. In this work, we demonstrate the first application of transformers for jet tagging on FPGAs, achieving $\mathcal{O}(100)$ nanosecond latency with superior performance compared to alternative baseline models. We leverage high-granularity quantization and distributed arithmetic optimization to fit the entire transformer model on a single FPGA, achieving the required throughput and latency. Furthermore, we add multi-head attention and linear attention support to hls4ml, making our work accessible to the broader fast machine learning community. This work advances the next-generation trigger systems for the High Luminosity LHC, enabling the use of transformers for real-time applications in high-energy physics and beyond.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。