CaloClouds II: Ultra-Fast Geometry-Independent Highly-Granular Calorimeter Simulation

Buhmann, Erik; Krüger, Katja; Kasieczka, Gregor; Gaede, Frank; McKeown, Peter; Korcari, William; Korol, Anatolii

Preprint

PUBDB-2023-05463

CaloClouds II: Ultra-Fast Geometry-Independent Highly-Granular Calorimeter Simulation

Buhmann, E. (Corresponding author)Extern* ; Gaede, F.DESY* ; Kasieczka, G.Extern* ; Korol, A.Extern*DESY* ; Korcari, W.Extern* ; Krüger, K.DESY* ; McKeown, P.DESY*

2023

[10.3204/PUBDB-2023-05463]

This record in other databases:

Please use a persistent id in citations: doi:10.3204/PUBDB-2023-05463

Report No.: DESY-23-130; arXiv:2309.05704

Abstract: Fast simulation of the energy depositions in high-granular detectors is needed for future collider experiments with ever increasing luminosities. Generative machine learning (ML) models have been shown to speed up and augment the traditional simulation chain in physics analysis. However, the majority of previous efforts were limited to models relying on fixed, regular detector readout geometries. A major advancement is the recently introduced CaloClouds model, a geometry-independent diffusion model, which generates calorimeter showers as point clouds for the electromagnetic calorimeter of the envisioned International Large Detector (ILD). In this work, we introduce CaloClouds II which features a number of key improvements. This includes continuous time score-based modelling, which allows for a 25 step sampling with comparable fidelity to CaloClouds while yielding a $6\times$ speed-up over Geant4 on a single CPU ($5\times$ over CaloClouds). We further distill the diffusion model into a consistency model allowing for accurate sampling in a single step and resulting in a $46\times$ ($37\times$) speed-up. This constitutes the first application of consistency distillation for the generation of calorimeter showers.

Keyword(s): diffusion, model ; calorimeter, electromagnetic ; showers ; geometry ; readout ; machine learning ; ILD detector ; cloud