MLPF: Efficient machine-learned particle-flow reconstruction using graph neural networks
Not on the map. Why?
Abstract (arXiv)
In general-purpose particle detectors, the particle-flow algorithm may be used to reconstruct a comprehensive particle-level view of the event by combining information from the calorimeters and the trackers, significantly improving the detector resolution for jets and the missing transverse momentum. In view of the planned high-luminosity upgrade of the CERN Large Hadron Collider (LHC), it is necessary to revisit existing reconstruction algorithms and ensure that both the physics and computational performance are sufficient in an environment with many simultaneous proton-proton interactions (pileup). Machine learning may offer a prospect for computationally efficient event reconstruction that is well-suited to heterogeneous computing platforms, while significantly improving the reconstruction quality over rule-based algorithms for granular detectors. We introduce MLPF, a novel, end-to-end trainable, machine-learned particle-flow algorithm based on parallelizable, computationally efficient, and scalable graph neural networks optimized using a multi-task objective on simulated events. We report the physics and computational performance of the MLPF algorithm on a Monte Carlo dataset of top quark-antiquark pairs produced in proton-proton collisions in conditions similar to those expected for the high-luminosity LHC. The MLPF algorithm improves the physics response with respect to a rule-based benchmark algorithm and demonstrates computationally scalable particle-flow reconstruction in a high-pileup environment.
Key facts. Cited by 36 papers in the map (data through 14 Oct 2026). Builds on 28 of its 81 references that are in the map. First posted on arXiv 21 Jan 2021.
Builds on
28 of 81 references.
References · 28
- 2017Attention Is All You Needcited by 60,361
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,248
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2018Dynamic Graph CNN for Learning on Point Cloudscited by 2,223
- 2020Reformer: The Efficient Transformercited by 1,668
- 2019Simplifying Graph Convolutional Networkscited by 1,477
- 2018Relational inductive biases, deep learning, and graph networkscited by 1,360
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,070
- 2016Interaction Networks for Learning about Objects, Relations and Physicscited by 509
16 more
- 2018Fast inference of deep neural networks in FPGAs for particle physicscited by 133
- 2018Energy Flow Networks: Deep Sets for Particle Jetscited by 84
- 2019ParticleNet: Jet Tagging via Particle Cloudscited by 64
- 2019Learning representations of irregular particle-detector geometry with distance-weighted graph networkscited by 43
- 2019Calorimetry with Deep Learning: Particle Simulation and Reconstruction for Collider Physicscited by 42
- 2020Distance-Weighted Graph Neural Networks on FPGAs for Real-Time Particle Reconstruction in High Energy Physicscited by 34
- 2019How to GAN away Detector Effectscited by 28
- 2020Object condensation: one-stage grid-free multi-object reconstruction in physics detectors, graph and image datacited by 24
- 2020Accelerated Charged Particle Tracking with Graph Neural Networks on FPGAscited by 23
- 2020Jet Flavour Classification Using DeepJetcited by 16
- 2020Towards a Computer Vision Particle Flowcited by 15
- 2020Track Seeding and Labelling with Embedded-space Graph Neural Networkscited by 12
- 2020Graph Generative Adversarial Networks for Sparse Data Generation in High Energy Physicscited by 11
- 2020FPGAs-as-a-Service Toolkit (FaaST)cited by 10
- 2020GPU coprocessors as a service for deep learning inference in high energy physicscited by 9
- 2020Graph Highway Networkscited by 2
Built on by
Citing papers per year: 2021: 8; 2022: 11; 2023: 6; 2024: 4; 2025: 3; 2026: 4.
36 papers.
Sort:
- 2021A Living Review of Machine Learning for Particle Physicscited by 78
- 2021Applications and Techniques for Fast Machine Learning in Sciencecited by 32
- 2021Performance of a Geometric Deep Learning Pipeline for HL-LHC Particle Trackingcited by 30
- 2021Hybrid Quantum Classical Graph Neural Networks for Particle Track Reconstructioncited by 20
- 2024Finetuning Foundation Models for Joint Analysis Optimizationcited by 18
- 2024Learning Long Range Dependencies on Graphs via Random Walkscited by 18
- 2021Graph Neural Networks for Charged Particle Tracking on FPGAscited by 17
- 2022Machine Learning for Particle Flow Reconstruction at CMScited by 14
- 2021The Tracking Machine Learning challenge : Throughput phasecited by 12
- 2021Explaining machine-learned particle-flow reconstructioncited by 12
- 2022Graph Neural Networks in Particle Physics: Implementations, Innovations, and Challengescited by 12
- 2022Snowmass 2021 Computational Frontier CompF03 Topical Group Report: Machine Learningcited by 10
24 more
- 2024A Language Model for Particle Trackingcited by 10
- 2025Fine-tuning machine-learned particle-flow reconstruction for new detector geometries in future colliderscited by 9
- 2022Explainable AI for High Energy Physicscited by 7
- 2023Progress towards an improved particle flow algorithm at CMS with machine learningcited by 7
- 2022End-to-end multi-particle reconstruction in high occupancy imaging calorimeters with graph neural networkscited by 6
- 2021Shared Data and Algorithms for Deep Learning in Fundamental Physicscited by 4
- 2022Do graph neural networks learn traditional jet substructure?cited by 4
- 2023Improved particle-flow event reconstruction with scalable neural networks for current and future particle detectorscited by 4
- 2022Hyperparameter optimization of data-driven AI models on HPC systemscited by 3
- 2022Bias and Priors in Machine Learning Calibrations for High Energy Physicscited by 3
- 2022Data Science and Machine Learning in Educationcited by 3
- 2022Learning Tree Structures from Leaves For Particle Decay Reconstructioncited by 3
- 2023Graph Structure from Point Clouds: Geometric Attention is All You Needcited by 2
- 2025Double Descent and Overparameterization in Particle Physics Datacited by 2
- 2026End-to-end event reconstruction for precision physics at future colliderscited by 2
- 2026Prompting Particle Physics: Tokenized Multi-modal Foundation Models for Combinatorially Many Taskscited by 2
- 2023Hyperparameter optimization, quantum-assisted model performance prediction, and benchmarking of AI-based High Energy Physics workloads using HPCcited by 1
- 2023Low Latency Edge Classification GNN for Particle Trajectory Tracking on FPGAscited by 1
- 2023Model Performance Prediction for Hyperparameter Optimization of Deep Learning Models Using High Performance Computing and Quantum Annealingcited by 1
- 2025Machine-learning based particle-flow algorithm in CMScited by 1
- 2026Machine-learned particle flow as a foundation model for collider physicscited by 1
- 2022Deep learning techniques for energy clustering in the CMS ECALcited by 0
- 2024Denoising Graph Super-Resolution towards Improved Collider Event Reconstructioncited by 0
- 2026HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstructioncited by 0
- 2026Prompting Particle Physics: Tokenized Multi-modal Foundation Models for Combinatorially Many Tasks25 Sep 2026
- 2026HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction18 Jun 2026
- 2026Machine-learned particle flow as a foundation model for collider physics12 Jun 2026
- 2026End-to-end event reconstruction for precision physics at future colliders4 Mar 2026
- 2025Double Descent and Overparameterization in Particle Physics Data1 Sep 2025
- 2025Machine-learning based particle-flow algorithm in CMS28 Aug 2025
- 2025Fine-tuning machine-learned particle-flow reconstruction for new detector geometries in future colliders28 Feb 2025
- 2024Denoising Graph Super-Resolution towards Improved Collider Event Reconstruction24 Sep 2024
- 2024Learning Long Range Dependencies on Graphs via Random Walks5 Jun 2024
- 2024A Language Model for Particle Tracking14 Feb 2024
- 2024Finetuning Foundation Models for Joint Analysis Optimization24 Jan 2024
- 2023Model Performance Prediction for Hyperparameter Optimization of Deep Learning Models Using High Performance Computing and Quantum Annealing29 Nov 2023
24 more
- 2023Improved particle-flow event reconstruction with scalable neural networks for current and future particle detectors13 Sep 2023
- 2023Graph Structure from Point Clouds: Geometric Attention is All You Need31 Jul 2023
- 2023Low Latency Edge Classification GNN for Particle Trajectory Tracking on FPGAs20 Jun 2023
- 2023Progress towards an improved particle flow algorithm at CMS with machine learning30 Mar 2023
- 2023Hyperparameter optimization, quantum-assisted model performance prediction, and benchmarking of AI-based High Energy Physics workloads using HPC27 Mar 2023
- 2022Do graph neural networks learn traditional jet substructure?17 Nov 2022
- 2022Snowmass 2021 Computational Frontier CompF03 Topical Group Report: Machine Learning15 Sep 2022
- 2022Learning Tree Structures from Leaves For Particle Decay Reconstruction31 Aug 2022
- 2022Data Science and Machine Learning in Education19 Jul 2022
- 2022Explainable AI for High Energy Physics14 Jun 2022
- 2022Bias and Priors in Machine Learning Calibrations for High Energy Physics10 May 2022
- 2022Deep learning techniques for energy clustering in the CMS ECAL21 Apr 2022
- 2022End-to-end multi-particle reconstruction in high occupancy imaging calorimeters with graph neural networks4 Apr 2022
- 2022Graph Neural Networks in Particle Physics: Implementations, Innovations, and Challenges23 Mar 2022
- 2022Hyperparameter optimization of data-driven AI models on HPC systems2 Mar 2022
- 2022Machine Learning for Particle Flow Reconstruction at CMS1 Mar 2022
- 2021Graph Neural Networks for Charged Particle Tracking on FPGAs3 Dec 2021
- 2021Explaining machine-learned particle-flow reconstruction24 Nov 2021
- 2021Applications and Techniques for Fast Machine Learning in Science25 Oct 2021
- 2021Hybrid Quantum Classical Graph Neural Networks for Particle Track Reconstruction26 Sep 2021
- 2021Shared Data and Algorithms for Deep Learning in Fundamental Physics1 Jul 2021
- 2021The Tracking Machine Learning challenge : Throughput phase3 May 2021
- 2021Performance of a Geometric Deep Learning Pipeline for HL-LHC Particle Tracking11 Mar 2021
- 2021A Living Review of Machine Learning for Particle Physics2 Feb 2021
- 2021A Living Review of Machine Learning for Particle Physicscited by 78
- 2021Applications and Techniques for Fast Machine Learning in Sciencecited by 32
- 2021Performance of a Geometric Deep Learning Pipeline for HL-LHC Particle Trackingcited by 30
- 2021Hybrid Quantum Classical Graph Neural Networks for Particle Track Reconstructioncited by 20
- 2024Finetuning Foundation Models for Joint Analysis Optimizationcited by 18
- 2024Learning Long Range Dependencies on Graphs via Random Walkscited by 18
- 2021Graph Neural Networks for Charged Particle Tracking on FPGAscited by 17
- 2022Machine Learning for Particle Flow Reconstruction at CMScited by 14
- 2021The Tracking Machine Learning challenge : Throughput phasecited by 12
- 2021Explaining machine-learned particle-flow reconstructioncited by 12
- 2022Graph Neural Networks in Particle Physics: Implementations, Innovations, and Challengescited by 12
- 2022Snowmass 2021 Computational Frontier CompF03 Topical Group Report: Machine Learningcited by 10
12 most cited of 36 All 36 (JSON)
- 2026Prompting Particle Physics: Tokenized Multi-modal Foundation Models for Combinatorially Many Tasks25 Sep 2026
- 2026HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction18 Jun 2026
- 2026Machine-learned particle flow as a foundation model for collider physics12 Jun 2026
- 2026End-to-end event reconstruction for precision physics at future colliders4 Mar 2026
- 2025Double Descent and Overparameterization in Particle Physics Data1 Sep 2025
- 2025Machine-learning based particle-flow algorithm in CMS28 Aug 2025
- 2025Fine-tuning machine-learned particle-flow reconstruction for new detector geometries in future colliders28 Feb 2025
- 2024Denoising Graph Super-Resolution towards Improved Collider Event Reconstruction24 Sep 2024
- 2024Learning Long Range Dependencies on Graphs via Random Walks5 Jun 2024
- 2024A Language Model for Particle Tracking14 Feb 2024
- 2024Finetuning Foundation Models for Joint Analysis Optimization24 Jan 2024
- 2023Model Performance Prediction for Hyperparameter Optimization of Deep Learning Models Using High Performance Computing and Quantum Annealing29 Nov 2023
12 newest of 36 All 36 (JSON)