Explainable AI for Jet Tagging: A Comparative Study of GNNExplainer, GNNShap, and GradCAM for Jet Tagging in the Lund Jet Plane
Not on the map. Why?
Abstract (arXiv)
Graph neural networks such as ParticleNet and transformer based networks on point clouds such as ParticleTransformer achieve state-of-the-art performance on jet tagging benchmarks at the Large Hadron Collider, yet the physical reasoning behind their predictions remains opaque. We present different methods, i.e. perturbation-based (GNNExplainer), Shapley-value-based (GNNShap), and gradient-based (GRADCam); adapted to operate on LundNet's Lund-plane graph representation. Leveraging the fact that each node in the Lund plane corresponds to a physically meaningful parton splitting, we construct Monte Carlo truth explanation masks and introduce a physics-informed evaluation framework that goes beyond standard fidelity metrics. We perform the analysis in three transverse-momentum bins (pT ∈ [500,700], [800,1000], and the inclusive region [500,1000] GeV), revealing how explanation quality and focus shift between non-perturbative and perturbative regimes. We further quantify the correlation between explainer-assigned node importance and classical jet substructure observables -- N-subjettiness ratios τ21 and τ32 and the energy correlation functions -- establishing the degree to which the model has learned known QCD features. We find that overall the weight assigned by explainability methods has a correlation with analytic observables, with expected shift across different phase space regimes, indicating that a trained neural network indeed learns some aspects of jet-substructure moments. Our open-source implementation enables reproducible explainability studies for graph-based jet taggers.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Builds on 15 of its 54 references that are in the map. First posted on arXiv 28 Apr 2026.
Builds on
15 of 54 references.
References · 15
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,223
- 2019GNNExplainer: Generating Explanations for Graph Neural Networkscited by 737
- 2020Explainability in Graph Neural Networks: A Taxonomic Surveycited by 384
- 2022Evaluating Explainability for Graph Neural Networkscited by 87
- 2019ParticleNet: Jet Tagging via Particle Cloudscited by 64
- 2015Jet-Images -- Deep Learning Editioncited by 41
- 2023Explainable Equivariant Neural Networks for Particle Physics: PELICANcited by 24
- 2020Jet tagging in the Lund plane with graph networkscited by 22
- 2016Deep learning in color: towards automated quark/gluon jet discriminationcited by 21
- 2025Interpretable Machine Learning in Physics: A Reviewcited by 14
- 2022Graph Neural Networks in Particle Physics: Implementations, Innovations, and Challengescited by 12
- 2024GNNShap: Scalable and Accurate GNN Explanation using Shapley Valuescited by 12
Built on by
Citing papers per year: 2026: 3.
3 papers.