Effectiveness of denoising diffusion probabilistic models for fast and high-fidelity whole-event simulation in high-energy heavy-ion experiments

Abstract

Artificial intelligence (AI) generative models, such as generative adversarial networks (GANs), variational auto-encoders, and normalizing flows, have been widely used and studied as efficient alternatives for traditional scientific simulations. However, they have several drawbacks, including training instability and inability to cover the entire data distribution, especially for regions where data are rare. This is particularly challenging for whole-event, full-detector simulations in high-energy heavy-ion experiments, such as sPHENIX at the Relativistic Heavy Ion Collider and Large Hadron Collider experiments, where thousands of particles are produced per event and interact with the detector. This work investigates the effectiveness of Denoising Diffusion Probabilistic Models (DDPMs) as an AI-based generative surrogate model for the sPHENIX experiment that includes the heavy-ion event generation and response of the entire calorimeter stack. DDPM performance in sPHENIX simulation data is compared with a popular rival, GANs. Results show that both DDPMs and GANs can reproduce the data distribution where the examples are abundant (low-to-medium calorimeter energies). Nonetheless, DDPMs significantly outperform GANs, especially in high-energy regions where data are rare. Additionally, DDPMs exhibit superior stability compared to GANs. The results are consistent between both central and peripheral centrality heavy-ion collision events. Moreover, DDPMs offer a substantial speedup of approximately a factor of 100 compared to the traditional Geant4 simulation method.

11 pages, 7 figures

References in corpus (13)

• An Introduction to Variational Autoencoders
• CaloGAN: Simulating 3D High Energy Particle Showers in Multi-Layer Electromagnetic Calorimeters with Generative Adversarial Networks
• Learning Particle Physics by Example: Location-Aware Generative Adversarial Networks for Physics Synthesis
• Accelerating Science with Generative Adversarial Networks: An Application to 3D Particle Showers in Multi-Layer Calorimeters
• Precise simulation of electromagnetic calorimeter showers using a Wasserstein Generative Adversarial Network
• Calorimetry with Deep Learning: Particle Simulation and Reconstruction for Collider Physics
• Fast Point Cloud Generation with Diffusion Models in High Energy Physics
• PC-JeDi: Diffusion for Particle Cloud Generation in High Energy Physics
• CaloScore v2: Single-shot Calorimeter Shower Simulation with Diffusion Models
• PC-Droid: Faster diffusion and improved quality for particle cloud generation
• Design and Beam Test Results for the sPHENIX Electromagnetic and Hadronic Calorimeter Prototypes
• Predictions for the sPHENIX physics program
• Physics Validation of Novel Convolutional 2D Architectures for Speeding Up High Energy Physics Simulations

Cited by in corpus (2)

• Ultra fast, event-by-event heavy-ion simulations for next generation experiments
• Toward a foundation model for heavy-ion collision experiments based on point-cloud diffusion