29 January 2024 to 2 February 2024
CERN
Europe/Zurich timezone

Conditional Set-to-Set Generation for Fast Simulation using Diffusion and Graph-to-Graph Translation

2 Feb 2024, 09:40
20m
222/R-001 (CERN)

222/R-001

CERN

200
Show room on map
Contributed talk 3 ML for simulation and surrogate model : Application of Machine Learning to simulation or other cases where it is deemed to replace an existing complex model Contributed Talks

Speaker

Dmitrii Kobylianskii (Weizmann Institute of Science (IL))

Description

Simulating particle physics data is a crucial yet computationally expensive aspect of analyzing data at the LHC. Typically, in fast simulation methods, we rely on a surrogate calorimeter model with a subsequent reconstruction algorithm to generate a set of reconstructed objects. This work demonstrates the potential to generate these reconstructed objects in one shot, effectively replacing both the calorimeter simulation and reconstruction steps. Our primary goal in this set-to-set generation is to accurately replicate the detector's resolution and the properties of the reconstructed objects.
Building on the success of our previous slot-attention-based model, we introduce two innovative approaches to improve this task and evaluate their performance using a realistic dataset. This dataset incorporates a realistic detector simulation and a machine learning-based reconstruction algorithm.
In the first approach, we enhance the slot-attention mechanism with a state-of-the-art graph diffusion model. This entails starting with a noisy graph and progressively eliminating noise conditioned on the truth particle set, ultimately generating the reconstructed objects.
The second approach involves graph refinement, directly converting the set of truth particles into the set of reconstructed objects. These approaches outperform our previous baseline regarding both accuracy and resolution of predicted particle properties.

Authors

Dmitrii Kobylianskii (Weizmann Institute of Science (IL)) Nathalie Soybelman (Weizmann Institute of Science (IL)) Nilotpal Kakati (Weizmann Institute of Science (IL))

Co-authors

Eilam Gross (Weizmann Institute of Science (IL)) Etienne Dreyer (Weizmann Institute of Science (IL))

Presentation materials