Semi-supervised Graph Neural Networks for Pileup Noise Removal

Abstract

The high instantaneous luminosity of the CERN Large Hadron Collider leads to multiple proton-proton interactions in the same or nearby bunch crossings (pileup). Advanced pileup mitigation algorithms are designed to remove this noise from pileup particles and improve the performance of crucial physics observables. This study implements a semi-supervised graph neural network for particle-level pileup noise removal, by identifying individual particles produced from pileup. The graph neural network is firstly trained on charged particles with known labels, which can be obtained from detector measurements on data or simulation, and then inferred on neutral particles for which such labels are missing. This semi-supervised approach does not depend on the ground truth information from simulation and thus allows us to perform training directly on experimental data. The performance of this approach is found to be consistently better than widely-used domain algorithms and comparable to the fully-supervised training using simulation truth information. The study serves as the first attempt at applying semi-supervised learning techniques to pileup mitigation, and opens up a new direction of fully data-driven machine learning pileup mitigation studies.

References in corpus (35)

• Community detection in graphs
• Observation of a new particle in the search for the Standard Model Higgs boson with the ATLAS detector at the LHC
• Observation of a new boson at a mass of 125 GeV with the CMS experiment at the LHC
• The anti-k_t jet clustering algorithm
• An Introduction to PYTHIA 8.2
• DELPHES 3, A modular framework for fast simulation of a generic collider experiment
• Community detection algorithms: a comparative analysis
• Particle-flow reconstruction and global event description with the CMS detector
• Description and performance of track and primary-vertex reconstruction with the CMS tracker
• ParticleNet: Jet Tagging via Particle Clouds
• Pileup Per Particle Identification
• Performance of missing transverse momentum reconstruction with the ATLAS detector using proton-proton collisions at $\sqrt{s}$ = 13 TeV
• Jet energy scale and resolution measured in proton-proton collisions at $\sqrt{s}=13$ TeV with the ATLAS detector
• Interleaved Parton Showers and Tuning Prospects
• Precision luminosity measurement in proton-proton collisions at $\sqrt{s}=$ 13 TeV in 2015 and 2016 at CMS
• Luminosity Determination in pp Collisions at sqrt(s)=7 TeV Using the ATLAS Detector at the LHC
• Performance of missing transverse momentum reconstruction in proton-proton collisions at $\sqrt{s}=$ 13 TeV using the CMS detector
• Energy Flow Networks: Deep Sets for Particle Jets
• Graph Neural Networks in Particle Physics
• Pileup mitigation at CMS in 13 TeV data
• Identification of heavy, energetic, hadronically decaying particles using machine-learning techniques
• Performance of the ATLAS Track Reconstruction Algorithms in Dense Environments in LHC Run 2
• Learning representations of irregular particle-detector geometry with distance-weighted graph networks
• Pileup Mitigation with Machine Learning (PUMML)
• SoftKiller, a particle-level pileup removal method
• ABCNet: An attention-based method for particle tagging
• Particle-level pileup subtraction for jets and jet shapes
• Graph Neural Networks for Particle Reconstruction in High Energy Physics detectors
• MLPF: Efficient machine-learned particle-flow reconstruction using graph neural networks
• Jet Cleansing: Pileup Removal at High Luminosity
• Identification and rejection of pile-up jets at high pseudorapidity with the ATLAS detector
• Quantifying the performance of jet definitions for kinematic reconstruction at the LHC
• Optimizing Generalized PageRank Methods for Seed-Expansion Community Detection
• Pile-Up Mitigation using Attention
• On the use of charged-track information to subtract neutral pileup

Cited by in corpus (5)

• Optimal transport for a novel event description at hadron colliders
• Learnable cut flow for high energy physics
• Variational inference for pile-up removal at hadron colliders with diffusion models
• Trials Factor for Semi-Supervised NN Classifiers in Searches for Narrow Resonances at the LHC
• Distilling particle knowledge for fast reconstruction at high-energy physics experiments