Pileup Mitigation with Flow Matching
Expand AI Research Group - University of Puerto Rico at Mayagüez
This project combines two ideas: machine-learning-based pileup mitigation and flow-based fast simulation. Pileup refers to additional unwanted proton-proton collisions that occur in the same beam crossing as the hard-scattering event of interest. To address this, we study pileup-cleaning methods such as PUMML , which uses convolutional neural networks to recover the leading-vertex neutral energy inside jets.
At the same time, we explore FlowSim , an end-to-end fast-simulation approach based on Flow Matching and Continuous Normalizing Flows. Instead of following the full conventional simulation chain — event generation, GEANT4-based detector simulation, digitization, reconstruction, and final analysis ntuples — the goal is to learn a direct mapping from generator-level information to realistic analysis-level observables.
Our direction is to make this end-to-end simulation more realistic by including pileup in the generated events and studying how well different mitigation strategies recover clean physics observables.