用物理约束提升粒子探测器模拟的扩散模型精度。
Lantern: Conflict-Aware Gradient Blending for Physics-Guided Diffusion Models in Calorimeter Simulation
- 设计冲突感知梯度融合机制,融合物理损失与去噪目标。
- 在CaloChallenge数据集上,物理一致性指标提升2-100倍。
- 适合高能物理仿真、需高保真模拟的研究者使用。
高亮度大型强子对撞机(HL-LHC)中,量能器簇射的蒙特卡洛模拟是主要瓶颈。扩散模型作为快速高保真的替代方法应运而生,但其去噪目标仅基于统计,无法保证物理准确性。现有物理引导生成方法依赖封闭形式定律或每样本硬约束,而簇射过程无单样本偏微分方程描述,仅能量守恒提供一个标量约束。标准指标忽略量能器层间与体素间的相关性,仅在物理特征空间比较簇射。本文提出层间与体素级相关性保真度的统一归一化评分——相关性弗罗贝尼乌斯距离(CFD)。通过引入两个物理感知辅助损失:基于计数统计的方差稳定体素残差损失,以及基于探测器几何图拉普拉斯损失,结合GradBlend梯度融合策略,在锚定去噪梯度幅值的同时让辅助损失控制方向,构建Lantern模型。在CaloChallenge Dataset 2上,使用PCGrad、GradNorm、IMTL-G、ConFIG等任务对称规则注入物理损失,使最终物理距离(FPD)相较纯去噪提升2-100倍;而GradBlend可无退化地融入相同信号。加入拉普拉斯损失后,Lantern同时优化了FPD与CFD。消融实验表明,与去噪冲突的体素残差损失需在末端进行纯去噪阶段以保持簇射保真度,而无冲突的拉普拉斯损失则对调度不敏感。
原文摘要 · Abstract (English)
Monte Carlo simulation of calorimeter showers is a principal bottleneck for the High-Luminosity LHC, and diffusion models have emerged as fast, high-fidelity surrogates. Their denoising objective is purely statistical, however: a model can minimize it while placing the physics wrong. Existing physics-informed generative methods cannot close this gap, because they assume a closed-form law, a governing PDE residual or a hard per-sample constraint, that a shower does not supply: no per-sample PDE governs a stochastic cascade, and energy conservation fixes only one scalar per shower. Standard metrics ignore the correlation structure across calorimeter layers and voxels, comparing showers only in a physics feature space. We address both gaps. We introduce the Correlation Frobenius Distance (CFD), a single normalized score for correlation fidelity at layer-wise and voxel-wise scales. We then encode the soft per-sample structure available in a shower as two physics-aware auxiliary losses: a variance-stabilized voxel residual loss grounded in counting statistics, and a graph Laplacian loss over the detector geometry. We combine both with denoising through GradBlend, which anchors the step magnitude to the denoising gradient while letting the auxiliary steer its direction, yielding Lantern, a physics-guided diffusion surrogate. On CaloChallenge Dataset 2, injecting the physics losses through task-symmetric rules such as PCGrad, GradNorm, IMTL-G, and ConFIG inflates FPD by 2-100x relative to denoising alone, whereas GradBlend admits the same signal without regression and, with the Laplacian loss, Lantern improves both FPD and CFD. Our ablation on the auxiliary loss scheduler shows that the voxel residual loss, whose gradient conflicts with denoising, requires a terminal denoising-only phase to preserve shower fidelity, whereas the non-conflicting Laplacian loss is insensitive to the schedule.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。