在抗辐射FPGA上实现25纳秒低延迟机器学习,助力高能物理实验
Enabling Low-Latency Machine learning on Radiation-Hard FPGAs with hls4ml
- 用轻量自编码器压缩32点时间读数至二维隐空间
- 10位量化使模型性能损失极小,实现超低延迟25纳秒
- 拓展hls4ml支持抗辐射FPGA,推动高辐射环境机器学习应用
本文展示了在辐射硬化FPGA上实现超快速机器学习应用的端到端方案,适用于未来高能物理实验。以LHCb升级二期计划中的PicoCal量能器为测试案例,提出三项贡献:首先,开发轻量自编码器将32样本时间读数压缩至二维隐空间;其次,提出硬件感知量化策略,模型权重可降至10位且性能损失极小;第三,针对高能物理领域主流工具hls4ml缺乏对辐射硬化FPGA支持的问题,开发新后端,实现将ML模型自动转换为Microchip PolarFire系列FPGA的高层次综合(HLS)项目。在目标PolarFire FPGA上合成自编码器,验证延迟可达25纳秒,资源占用低,可部署于FPGA固有保护逻辑中。该扩展显著推动了机器学习在高辐射环境下的应用。
原文摘要 · Abstract (English)
This paper presents an end-to-end demonstration of a viable, ultra-fast, radiation-hard machine learning (ML) application on FPGAs, which could be used in future high-energy physics experiments. We present a three-fold contribution, with the PicoCal calorimeter, planned for the LHCb Upgrade II experiment, used as a test case. First, we develop a lightweight autoencoder to compress a 32-sample timing readout, representative of that of the PicoCal, into a two-dimensional latent space. Second, we introduce a systematic, hardware-aware quantization strategy and show that the model can be reduced to 10-bit weights with minimal performance loss. Third, as a barrier to the adoption of on-detector ML is the lack of support for radiation-hard FPGAs in the High-Energy Physics community's standard ML synthesis tool, hls4ml, we develop a new backend for this library. This new back-end enables the automatic translation of ML models into High-Level Synthesis (HLS) projects for the Microchip PolarFire family of FPGAs, one of the few commercially available and radiation hard FPGAs. We present the synthesis of the autoencoder on a target PolarFire FPGA, which indicates that a latency of 25 ns can be achieved. We show that the resources utilized are low enough that the model can be placed within the inherently protected logic of the FPGA. Our extension to hls4ml is a significant contribution, paving the way for broader adoption of ML on FPGAs in high-radiation environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。