1–4 Nov 2022
Rutgers University
US/Eastern timezone

Fast calorimeter simulation with VQVAE

2 Nov 2022, 16:50
20m
Multipurpose Room (aka Livingston Hall) (Rutgers University)

Multipurpose Room (aka Livingston Hall)

Rutgers University

Livingston Student Center

Speaker

Chase Owen Shimmin (Yale University (US))

Description

Simulation of calorimeter response is important for modern high energy physics experiments. With the increasingly large and high granularity design of calorimeters, the computational cost of conventional MC-based simulation of each particle-material interaction is becoming a major bottleneck. We propose a new generative model based on a Vector-Quantized Variational Autoencoder (VQ-VAE) to generate the calorimeter response. This model achieved a speedup of more than 5x10^4 times over GEANT4 on the CaloGAN dataset and the comparable performance of energy deposition and shower shape as existing ML-models such as CaloGAN and CaloFlow, with substantially fewer parameters and factor of 2 more speedup. We also demonstrate that the VQVAE approach can be adapted to a variety of encoder/decoder architectures, ranging from fully-connected to convolutional networks. The former is more suited to smaller, or irregular geometries, while the latter can perform well on very high granularity datasets with regular structure.

Primary authors

Qibin Liu (Tsung-Dao Lee Institute (CN) & Shanghai Jiao Tong University (CN)) Chase Owen Shimmin (Yale University (US)) Xiulong Liu (University of Washington, Seattle) Prof. Eli Shlizerman (University of Washington, Seattle) Shih-Chieh Hsu (University of Washington Seattle (US))

Presentation materials