arXiv:2507.19205cs.LG2025-07被引 1

用物理先验增强图神经网络,提升高能物理触发系统中动量估计的精度与效率。

Physics-Informed Graph Neural Networks for Transverse Momentum Estimation in CMS Trigger Systems

  • 基于探测器结构和物理量设计四类图构建策略,融合物理先验信息。
  • 在CMS触发数据集上实现0.8525的最小绝对误差,参数量减少超55%。
  • 适合高能物理实时计算、资源受限场景下的动量估计任务。

高能物理中实时粒子横向动量($p_T$)估计需在严格硬件约束下兼顾效率与精度。静态机器学习模型在高堆叠环境下性能下降,且缺乏物理感知优化;通用图神经网络(GNN)常忽略关键领域结构,影响$p_T$回归稳定性。本文提出一种物理引导的GNN框架,通过四种不同图构造策略——站点为节点、特征为节点、弯角中心、赝快度($η$)中心——系统编码探测器几何与物理可观测量。该框架结合新型消息传递层(MPL),包含内部注意力机制与门控更新,并引入融合$p_T$分布先验的领域特定损失函数。联合设计方法在多项指标上优于现有基线:基于站点信息的EdgeConv模型在CMS触发数据集上达到0.8525的最优平均绝对误差(MAE),参数量比深度学习基线减少≥55%,尤其优于TabNet;而$η$-中心配置也表现出更高精度与相当效率。结果证明物理引导的GNN在资源受限触发系统中的应用潜力。

原文摘要 · Abstract (English)

Real-time particle transverse momentum ($p_T$) estimation in high-energy physics demands algorithms that are both efficient and accurate under strict hardware constraints. Static machine learning models degrade under high pileup and lack physics-aware optimization, while generic graph neural networks (GNNs) often neglect domain structure critical for robust $p_T$ regression. We propose a physics-informed GNN framework that systematically encodes detector geometry and physical observables through four distinct graph construction strategies that systematically encode detector geometry and physical observables: station-as-node, feature-as-node, bending angle-centric, and pseudorapidity ($η$)-centric representations. This framework integrates these tailored graph structures with a novel Message Passing Layer (MPL), featuring intra-message attention and gated updates, and domain-specific loss functions incorporating $p_{T}$-distribution priors. Our co-design methodology yields superior accuracy-efficiency trade-offs compared to existing baselines. Extensive experiments on the CMS Trigger Dataset validate the approach: a station-informed EdgeConv model achieves a state-of-the-art MAE of 0.8525 with $\ge55\%$ fewer parameters than deep learning baselines, especially TabNet, while an $η$-centric MPL configuration also demonstrates improved accuracy with comparable efficiency. These results establish the promise of physics-guided GNNs for deployment in resource-constrained trigger systems.

图神经网络动量估计高能物理实时系统

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