arXiv:2507.18811cs.LG2025-07被引 3

用流匹配模型加速阿尔契实验零度量能器仿真,速度提升百倍以上。

Even Faster Simulations with Flow Matching: A Study of Zero Degree Calorimeter Responses

  • 采用低参数流匹配方法构建快速生成模型。
  • 零中子/质子探测器仿真误差低于1.3,推理速度达0.026毫秒/样本。
  • 适合高能物理仿真加速,代码已开源。

近年来,生成神经网络特别是流匹配(FM)技术在生成高保真样本的同时显著降低了计算成本。本研究利用FM开发了阿尔契实验中零度量能器的替代仿真模型。提出一种高效训练策略,使生成模型以极低参数量实现快速训练。该方法在中子(ZN)和质子(ZP)探测器仿真中均达到当前最优保真度,同时大幅降低计算开销。FM模型在ZN仿真中实现1.27的Wasserstein距离,单样本推理时间仅0.46毫秒,优于现有最佳结果(1.20,约109毫秒)。潜空间FM模型进一步将采样时间压缩至0.026毫秒/样本,精度损失极小。对于ZP仿真,其Wasserstein距离为1.30,优于当前最佳的2.08。源码已公开于https://github.com/m-wojnar/faster_zdc。

原文摘要 · Abstract (English)

Recent advances in generative neural networks, particularly flow matching (FM), have enabled the generation of high-fidelity samples while significantly reducing computational costs. A promising application of these models is accelerating simulations in high-energy physics (HEP), helping research institutions meet their increasing computational demands. In this work, we leverage FM to develop surrogate models for fast simulations of zero degree calorimeters in the ALICE experiment. We present an effective training strategy that enables the training of fast generative models with an exceptionally low number of parameters. This approach achieves state-of-the-art simulation fidelity for both neutron (ZN) and proton (ZP) detectors, while offering substantial reductions in computational costs compared to existing methods. Our FM model achieves a Wasserstein distance of 1.27 for the ZN simulation with an inference time of 0.46 ms per sample, compared to the current best of 1.20 with an inference time of approximately 109 ms. The latent FM model further improves the inference speed, reducing the sampling time to 0.026 ms per sample, with a minimal trade-off in accuracy. Similarly, our approach achieves a Wasserstein distance of 1.30 for the ZP simulation, outperforming the current best of 2.08. The source code is available at https://github.com/m-wojnar/faster_zdc.

流匹配高能物理仿真加速生成模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。