11–15 Mar 2024
Charles B. Wang Center, Stony Brook University
US/Eastern timezone

"Accelerating Particle Physics Simulations with Machine Learning using Normalizing Flows and Flow Matching"

13 Mar 2024, 17:50
20m
Theatre ( Charles B. Wang Center, Stony Brook University )

Theatre

Charles B. Wang Center, Stony Brook University

100 Circle Rd, Stony Brook, NY 11794
Oral Track 1: Computing Technology for Physics Research Track 1: Computing Technology for Physics Research

Speaker

Francesco Vaselli (Scuola Normale Superiore & INFN Pisa (IT))

Description

The simulation of high-energy physics collision events is a key element for data analysis at present and future particle accelerators. The comparison of simulation predictions to data allows us to look for rare deviations that can be due to new phenomena not previously observed. We show that novel machine learning algorithms, specifically Normalizing Flows and Flow Matching, can be effectively used to perform accurate simulations with several orders of magnitude of speed-up compared to traditional approaches. The classical simulation chain starts from a physics process of interest, computes energy deposits of particles and electronics response; and finally employs the same reconstruction algorithms used for data. Eventually, the data is reduced to some high-level analysis format. Instead, we propose an end-to-end approach, simulating the final data format directly from physical generator inputs, skipping any intermediate steps. We use particle jets simulation as a benchmark for comparing both discrete and continuous Normalizing Flows models. The models are validated across a variety of metrics to select the best ones. We discuss the scaling of performance with the increase in training data, as well as the generalization power of these models on physical processes different from the training one. We investigate sampling multiple times from the same inputs, a procedure we name oversampling, and we show that it can effectively reduce the statistical uncertainties of a sample. This class of ML algorithms is found to be highly expressive and useful for the task of HEP simulation. Their speed and accuracy, coupled with the stability of the training procedure, make them a compelling tool for the needs of current and future experiments.

Significance

Application of novel machine learning algorithms and training routines to the end-to-end simulation problem in high energy physics. Demonstrated major improvements in simulation speed while retaining a high level of accuracy and fidelity.

Primary author

Francesco Vaselli (Scuola Normale Superiore & INFN Pisa (IT))

Co-authors

Andrea Rizzi (Universita & INFN Pisa (IT)) Filippo Cattafesta (Universita & INFN Pisa (IT)) Patrick Asenov (Universita & INFN Pisa (IT))

Presentation materials