Conditional Wasserstein Generative Adversarial Networks for Fast Detector Simulation

18 May 2021, 15:52
13m
Short Talk Offline Computing Artificial Intelligence

Speaker

John Blue (Davidson College)

Description

Detector simulation in high energy physics experiments is a key yet computationally expensive step in the event simulation process. There has been much recent interest in using deep generative models as a faster alternative to the full Monte Carlo simulation process in situations in which the utmost accuracy is not necessary. In this work we investigate the use of conditional Wasserstein Generative Adversarial Networks to simulate both hadronization and the detector response to jets. Our model takes the $4$-momenta of jets formed from partons post-showering and pre-hadronization as inputs and predicts the $4$-momenta of the corresponding reconstructed jet. Our model is trained on fully simulated $t\overline{t}$ events using the publicly available GEANT-based simulation of the CMS Collaboration. We demonstrate that the model produces accurate conditional reconstructed jet transverse momentum ($p_T$) distributions over a wide range of $p_T$ for the input parton jet. Our model takes only a fraction of the time necessary for conventional detector simulation methods, running on a CPU in less than a millisecond per event.

Primary authors

John Blue (Davidson College) Sitong An (CERN, Carnegie Mellon University (US)) Dr Davide Di Croce (University of Alabama) Mrs Darya Dyachkova (Minerva Schools at KGI) Prof. Sergei Gleyzer (University of Alabama) Ali Hariri (American University of Beirut (LB)) Braden Kronheim Prof. Michelle Kuchera (Davidson College) Prof. Harrison Prosper (Florida State University) Dr Ramanujan Raghuram (Davidson College) Dr Emanuele Usai (Brown University)

Presentation materials

Proceedings

Paper