arXiv:2412.04677

Zephyr quantum-assisted hierarchical Calo4pQVAE for particle-calorimeter interactions

Lu, Ian, Hao Wei Jia, Gonzalez, Sebastian, Deniz Sogutlu, J. Quetzalcóatl Toledo-Marín, Sehmimul Hoque, Abhishek Singh, C. Gay, Roger G. Melko, Paquet, Eric, Geoffrey Fox, M. Swiatlowski, Wojciech T. Fedorko

Published Dec 6, 2024• 0 citations

arXiv HTML PDF

Abstract

With the approach of the High Luminosity Large Hadron Collider (HL-LHC) era set to begin particle collisions by the end of this decade, it is evident that the computational demands of traditional collision simulation methods are becoming increasingly unsustainable. Existing approaches, which rely heavily on first-principles Monte Carlo simulations for modeling event showers in calorimeters, are projected to require millions of CPU-years annually -- far exceeding current computational capacities. This bottleneck presents an exciting opportunity for advancements in computational physics by integrating deep generative models with quantum simulations. We propose a quantum-assisted hierarchical deep generative surrogate founded on a variational autoencoder (VAE) in combination with an energy conditioned restricted Boltzmann machine (RBM) embedded in the model's latent space as a prior. By mapping the topology of D-Wave's Zephyr quantum annealer (QA) into the nodes and couplings of a 4-partite RBM, we leverage quantum simulation to accelerate our shower generation times significantly. To evaluate our framework, we use Dataset 2 of the CaloChallenge 2022. Through the integration of classical computation and quantum simulation, this hybrid framework paves way for utilizing large-scale quantum simulations as priors in deep generative models.

DOI: 10.48550/arxiv.2412.04677

Related papers

Conditioned quantum-assisted deep generative surrogate for particle-calorimeter interactions

J. Quetzalcóatl Toledo-Marín, Sebastian Gonzalez, Hao Jia +9 more

Particle collisions at accelerators like the Large Hadron Collider (LHC), recorded by experiments such as ATLAS and CMS, enable precise standard model measurements and searches for new phenomena. Simulating these collisions significantly influences experiment design and analysis but incurs immense computational costs, projected at millions of CPU-years annually during the high luminosity LHC (HL-LHC) phase. Currently, simulating a single event with Geant4 consumes around 1000 CPU seconds, with calorimeter simulations especially demanding. To address this, we propose a conditioned quantum-assisted generative model, integrating a conditioned variational autoencoder (VAE) and a conditioned restricted Boltzmann machine (RBM). Our RBM architecture is tailored for D-Wave's Pegasus-structured advantage quantum annealer for sampling, leveraging the flux bias for conditioning. This approach combines classical RBMs as universal approximators for discrete distributions with quantum annealing's speed and scalability. We also introduce an adaptive method for efficiently estimating effective inverse temperature, and validate our framework on Dataset 2 of CaloChallenge.

arXiv:2410.22870npj Quantum InformationJul 7, 20253 citations

Quantum-Assisted Generative AI for Simulation of the Calorimeter Response

Wojciech T. Fedorko, J. Quetzalcóatl Toledo-Marín, Geoffrey Fox +7 more

As CERN approaches the launch of the High Luminosity Large Hadron Collider (HL-LHC) by the decade’s end, the computational demands of traditional simulations have become untenably high. Projections show millions of CPU-years required to create simulated datasets - with a substantial fraction of CPU time devoted to calorimetric simulations. This presents unique opportunities for breakthroughs in computational physics. We show how Quantumassisted Generative AI can be used for the purpose of creating synthetic, realistically scaled calorimetry dataset. The model is constructed by combining D-Wave’s Quantum Annealer processor with a Deep Learning architecture, increasing the timing performance with respect to first principles simulations and Deep Learning models alone, while maintaining current state-of-the-art data quality

Oct 7, 20250 citations

CaloQVAE for electromagnetic shower synthetic data generation in the ATLAS calorimeter

Wojciech T. Fedorko

As CERN approaches the High-Luminosity LHC (HL-LHC) run, the computational cost of first-principles detector simulation has become a critical bottleneck, with projections indicating millions of CPU-years required to generate the necessary Monte Carlo datasets — a substantial share of which is consumed by calorimeter shower simulation. Building on our earlier proof-of-concept demonstrating Quantum-assisted Generative AI for synthetic calorimetry data generation, we present what is, to our knowledge, the first full-scale demonstration of this approach applied to ATLAS detector simulated data. Our model couples a Restricted Boltzmann Machine (RBM), mapped onto the logical qubit topology of D-Wave's quantum annealing processor, with a deep generative architecture to produce realistically scaled calorimeter showers. We extend our earlier framework with several techniques critical to scaling the approach to production-grade ATLAS geometry and granularity: advanced prior conditioning strategies, Spin Reversal Transforms to mitigate intrinsic qubit biases, orbit averaging to reduce sampling noise, and QPU-aware weight normalization to stabilize the sampling process. We show how these advances yield quantum processor generated calorimeter datasets that maintain high fidelity in physics features relative to first-principles GEANT4 simulation while promising substantial improvements in generation throughput over both traditional simulation and purely classical deep learning generative models. We discuss the implications of these results for scaling quantum-assisted generative modeling to full ATLAS fast-simulation pipelines ahead of the HL-LHC era.

Sep 25, 20260 citations

Quantum-Assisted Generative AI for Simulation of the Calorimeter Response

Wojciech T. Fedorko, J. Quetzalcóatl Toledo-Marín, Geoffrey Fox +7 more

EPJ Web of ConferencesJan 1, 20250 citations

A Path Towards Quantum Advantage in Training Deep Generative Models with Quantum Annealers

Walter Vinci, Lorenzo Buffoni, Sadeghi, Hossein +3 more

The development of quantum-classical hybrid (QCH) algorithms is critical to achieve state-of-the-art computational models. A QCH variational autoencoder (QVAE) was introduced in Ref. [1] by some of the authors of this paper. QVAE consists of a classical auto-encoding structure realized by traditional deep neural networks to perform inference to, and generation from, a discrete latent space. The latent generative process is formalized as thermal sampling from either a quantum or classical Boltzmann machine (QBM or BM). This setup allows quantum-assisted training of deep generative models by physically simulating the generative process with quantum annealers. In this paper, we have successfully employed D-Wave quantum annealers as Boltzmann samplers to perform quantum-assisted, end-to-end training of QVAE. The hybrid structure of QVAE allows us to deploy current-generation quantum annealers in QCH generative models to achieve competitive performance on datasets such as MNIST. The results presented in this paper suggest that commercially available quantum annealers can be deployed, in conjunction with well-crafted classical deep neutral networks, to achieve competitive results in unsupervised and semisupervised tasks on large-scale datasets. We also provide evidence that our setup is able to exploit large latent-space (Q)BMs, which develop slowly mixing modes. This expressive latent space results in slow and inefficient classical sampling, and paves the way to achieve quantum advantage with quantum annealing in realistic sampling applications.

arXiv:1912.02119Dec 4, 20199 citations

A path towards quantum advantage in training deep generative models with quantum annealers

Walter Vinci, Lorenzo Buffoni, Hossein Sadeghi +3 more

The development of quantum-classical hybrid (QCH) algorithms is critical to achieve state-of-the-art computational models. A QCH variational autoencoder (QVAE) was introduced in reference [ 1 ] by some of the authors of this paper. QVAE consists of a classical auto-encoding structure realized by traditional deep neural networks to perform inference to and generation from, a discrete latent space. The latent generative process is formalized as thermal sampling from a quantum Boltzmann machine (QBM). This setup allows quantum-assisted training of deep generative models by physically simulating the generative process with quantum annealers. In this paper, we have successfully employed D-Wave quantum annealers as Boltzmann samplers to perform quantum-assisted, end-to-end training of QVAE. The hybrid structure of QVAE allows us to deploy current-generation quantum annealers in QCH generative models to achieve competitive performance on datasets such as MNIST. The results presented in this paper suggest that commercially available quantum annealers can be deployed, in conjunction with well-crafted classical deep neutral networks, to achieve competitive results in unsupervised and semisupervised tasks on large-scale datasets. We also provide evidence that our setup is able to exploit large latent-space QBMs, which develop slowly mixing modes. This expressive latent space results in slow and inefficient classical sampling and paves the way to achieve quantum advantage with quantum annealing in realistic sampling applications.

Machine Learning Science and TechnologyJul 2, 202054 citations