EPiC-GAN: Equivariant point cloud generation for particle jets
Erik Buhmann, Gregor Kasieczka, Jesse Thaler
SciPost Phys. 15, 130 (2023) · published 3 October 2023
- doi: 10.21468/SciPostPhys.15.4.130
- Submissions/Reports
Abstract
With the vast data-collecting capabilities of current and future high-energy collider experiments, there is an increasing demand for computationally efficient simulations. Generative machine learning models enable fast event generation, yet so far these approaches are largely constrained to fixed data structures and rigid detector geometries. In this paper, we introduce EPiC-GAN - equivariant point cloud generative adversarial network - which can produce point clouds of variable multiplicity. This flexible framework is based on deep sets and is well suited for simulating sprays of particles called jets. The generator and discriminator utilize multiple EPiC layers with an interpretable global latent vector. Crucially, the EPiC layers do not rely on pairwise information sharing between particles, which leads to a significant speed-up over graph- and transformer-based approaches with more complex relation diagrams. We demonstrate that EPiC-GAN scales well to large particle multiplicities and achieves high generation fidelity on benchmark jet generation tasks.
Cited by 26
Authors / Affiliations: mappings to Contributors and Organizations
See all Organizations.- 1 Erik Buhmann,
- 1 2 Gregor Kasieczka,
- 3 4 Jesse Thaler
- 1 Universität Hamburg / University of Hamburg [UH]
- 2 Center for Data and Computing in Natural Sciences [CDCS]
- 3 Massachusetts Institute of Technology [MIT]
- 4 The NSF AI Institute for Artificial Intelligence and Fundamental Interactions [IAIFI]