Differentiable Surrogate for Detector Simulation and Design with Diffusion Models
arXiv:2601.07859 · doi:10.1088/2632-2153/ae5c56
Abstract
In this work, we present a conditional denoising-diffusion surrogate for electromagnetic calorimeter showers that is trained to generate high-fidelity energy-deposition maps conditioned on key detector and beam parameters. The model employs efficient inference using Denoising Diffusion Implicit Model sampling and is pre-trained on GEANT4 simulations before being adapted to a new calorimeter geometry through Low-Rank Adaptation, requiring only a small post-training dataset. We evaluate physically meaningful observables, including total deposited energy, energy-weighted radius, and shower dispersion, obtaining relative root mean square error values below 2% for representative high-energy cases. This is in line with state-of-the-art calorimeter surrogates which report comparable fidelity on high-level observables. Furthermore, we compare gradients of a reconstruction-based utility function with respect to design parameters between the surrogate and finite-difference references. The diffusion surrogate reproduces the qualitative structure and directional trends of the true utility landscape, providing usable sensitivities for gradient-based optimization. These results show that diffusion-based surrogates can accelerate simulation-driven detector design while enabling differentiable, gradient-informed analysis.
43 pages, 21 figures
References in corpus (9)
- Denoising Diffusion Probabilistic Models
- CaloGAN: Simulating 3D High Energy Particle Showers in Multi-Layer Electromagnetic Calorimeters with Generative Adversarial Networks
- Score-based Generative Models for Calorimeter Shower Simulation
- CaloFlow: Fast and Accurate Generation of Calorimeter Showers with Normalizing Flows
- Detector and Physics Performance at a Muon Collider
- CaloClouds: Fast Geometry-Independent Highly-Granular Calorimeter Simulation
- CaloDREAM -- Detector Response Emulation via Attentive flow Matching
- Progress in End-to-End Optimization of Detectors for Fundamental Physics with Differentiable Programming
- Hadron Identification Prospects With Granular Calorimeters