Mixture-of-Experts Graph Transformers for Interpretable Particle Collision Detection
On the map: Mixture-of-experts → MoE for time series and graphs
Abstract (arXiv)
The Large Hadron Collider at CERN produces immense volumes of complex data from high-energy particle collisions, demanding sophisticated analytical techniques for effective interpretation. Neural Networks, including Graph Neural Networks, have shown promise in tasks such as event classification and object identification by representing collisions as graphs. However, while Graph Neural Networks excel in predictive accuracy, their "black box" nature often limits their interpretability, making it difficult to trust their decision-making processes. In this paper, we propose a novel approach that combines a Graph Transformer model with Mixture-of-Expert layers to achieve high predictive performance while embedding interpretability into the architecture. By leveraging attention maps and expert specialization, the model offers insights into its internal decision-making, linking predictions to physics-informed features. We evaluate the model on simulated events from the ATLAS experiment, focusing on distinguishing rare Supersymmetric signal events from Standard Model background. Our results highlight that the model achieves competitive classification accuracy while providing interpretable outputs that align with known physics, demonstrating its potential as a robust and transparent tool for high-energy physics data analysis. This approach underscores the importance of explainability in machine learning methods applied to high energy physics, offering a path toward greater trust in AI-driven discoveries.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in MoE for time series and graphs, one of 12 directions in Mixture-of-experts. Builds on 25 of its 44 references that are in the map. First posted on arXiv 6 Jan 2025.
Builds on
25 of 44 references.
References · 25
- 2017Attention Is All You Needcited by 59,932
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2017A Unified Approach to Interpreting Model Predictionscited by 6,488
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,120
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,601
- 2019Explainable Artificial Intelligence (XAI): Concepts, Taxonomies, Opportunities and Challenges toward Responsible AIcited by 1,543
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2018Sanity Checks for Saliency Mapscited by 1,149
- 2018A Survey Of Methods For Explaining Black Box Modelscited by 970
- 2019GNNExplainer: Generating Explanations for Graph Neural Networkscited by 735
- 2020A Generalization of Transformer Networks to Graphscited by 653
13 more
- 2020Estimating Training Data Influence by Tracing Gradient Descentcited by 565
- 2018Explaining Explanations: An Overview of Interpretability of Machine Learningcited by 508
- 2017Interpretation of Neural Networks is Fragilecited by 443
- 2020Parameterized Explainer for Graph Neural Networkcited by 423
- 2019On the Relationship between Self-Attention and Convolutional Layerscited by 299
- 2013Learning Factored Representations in a Deep Mixture of Expertscited by 275
- 2022Multimodal Contrastive Learning with LIMoE: the Language-Image Mixture of Expertscited by 202
- 2022Sign and Basis Invariant Networks for Spectral Graph Representation Learningcited by 176
- 2021Graph Learning: A Surveycited by 140
- 2021Do Feature Attribution Methods Correctly Attribute Features?cited by 98
- 2020Explainable AI for ML jet taggers using expert variables and layerwise relevance propagationcited by 14
- 2021Explaining machine-learned particle-flow reconstructioncited by 12
- 2024Enhancing High-Energy Particle Physics Collision Analysis through Graph Data Attribution Techniquescited by 1
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Scalable Multi-Task Learning for Particle Collision Event Reconstruction with Heterogeneous Graph Neural Networks
- 2026Comprehensive reconstruction of collider events with hypergraph representation learning and graph-conditioned diffusion
- 2026Conditional Capacity and Routing in Mixture-of-Experts Particle Transformers
- 2026Application of a Mixture of Experts-based Foundation Model to the GlueX DIRC Detector
- 2023End-To-End Latent Variational Diffusion Models for Inverse Problems in High Energy Physics
- 2025Agents of Discovery
- 2024GINN-KAN: Interpretability pipelining with applications in Physics Informed Neural Networks
- 2024Generative Diffusion Models for Fast Simulations of Particle Collisions at CERN