arXiv:2512.07420hep-phcs.LG2025-12被引 1

KIGNet用物理启发的图结构提升喷注分类可解释性,性能超越现有方法。

KIGNet: Physics-Motivated Multi-Graph Representation Learning for Explainable Jet Tagging

  • 构建四种基于物理量的图表示,融合动量、角度、质量等信息
  • 梯度归因显示76%注意力集中于角度和横向动量,符合量子色动力学规律
  • 在真实探测数据上显著提升表征结构,适合高能物理可解释建模

喷注识别在高能对撞机实验数据分析中至关重要。尽管深度学习提升了喷注分类性能,但缺乏可解释性。本文提出运动学交互图网络(KIGNet),通过为每个喷注构建四种图表示来整合运动学变量:角分离(Δ)、相对横向动量(k_T)、动量分数(z)和不变质量平方(m²)。其中前三者源于吕恩喷注平面,基于微扰量子色动力学因子化;第四项补充了重味识别的质量敏感性。利用梯度加权类激活映射(Grad-CAM)分析发现,角分离与相对横向动量贡献了约76%的归因(分别为40.72%和35.67%),动量分数与不变质量贡献剩余24%。该分布与训练数据中量子色动力学辐射的软-共线结构一致,表明网络学习的是物理可解释特征而非虚假关联。在JetClass数据集上,KIGNet实现95.07%宏准确率、96.61%宏AUC与81.52%宏AUPR,相较最优基线分别提升2.45%、3.40%与19.11%。在真实CMS碰撞数据的Aspen Open Jets数据集上,其潜空间表征更结构化,戴维斯-鲍尔丁指数下降52.15%(0.8395 → 0.4017),邓恩指数上升42.33%(0.0189 → 0.0269),证明物理启发编码能有效推广至实验探测条件。

原文摘要 · Abstract (English)

Jet identification plays a central role in analyzing data from high-energy collider experiments. While deep learning has improved jet classification, it often lacks interpretability. We introduce the Kinematic Interaction Graph Network (KIGNet), a graph neural network that integrates kinematic variables into jet classification by constructing four graph representations per jet, each weighted by a distinct variable: angular separation ($Δ$), relative transverse momentum ($k_T$), momentum fraction ($z$), and invariant mass squared ($m^2$). Three of these ($Δ$, $k_T$, $z$) are motivated by the Lund jet plane, grounded in perturbative QCD factorization; the fourth ($m^2$) adds complementary mass-scale sensitivity for heavy-flavor identification. Using Gradient-weighted Class Activation Mapping (Grad-CAM), we determine which variables dominate classification. Angular separation and relative transverse momentum account for about 76% of the total Grad-CAM attribution (40.72% and 35.67%), with momentum fraction and invariant mass contributing the remaining 24%. This hierarchy is consistent with the soft-collinear structure of QCD radiation in the training data, showing that the network learns physically interpretable representations rather than spurious correlations. On the JetClass dataset, KIGNet achieves a macro-accuracy of 95.07%, macro-AUC of 96.61%, and macro-AUPR of 81.52%, relative improvements of 2.45%, 3.40%, and 19.11% over the state-of-the-art baseline. On the Aspen Open Jets dataset of real CMS collision data, KIGNet produces substantially more structured latent representations than the baseline, reducing the Davies-Bouldin Index by 52.15% ($0.8395 \rightarrow 0.4017$) and increasing the Dunn Index by 42.33% ($0.0189 \rightarrow 0.0269$), confirming that physics-informed kinematic encoding generalizes beyond idealized simulation to experimental detector conditions.

喷注识别图神经网络可解释性高能物理

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