arXiv:2412.21082 · Proceedings of the AAAI Symposium Series
Quantum Diffusion Model for Quark and Gluon Jet Generation
Mariia Baidachna, Rey Guadarrama, Gopal Ramesh Dahale, Tom Magorsch, I. Pedraza, Konstantin Tzvetanov Matchev, Katia Matcheva, Kyoungchul Kong, Sergei Gleyzer
Diffusion models have demonstrated remarkable success in image generation, but they are computationally intensive and time-consuming to train. In this paper, we introduce a novel diffusion model that benefits from quantum computing techniques in order to mitigate computational challenges and enhance generative performance within high energy physics data. The fully quantum diffusion model replaces Gaussian noise with random unitary matrices in the forward process and incorporates a variational quantum circuit within the U-Net in the denoising architecture. We run evaluations on the structurally complex quark and gluon jets dataset from the Large Hadron Collider. The results demonstrate that the fully quantum and hybrid models are competitive with a similar classical model for jet generation, highlighting the potential of using quantum techniques for machine learning problems.
Diffusion models have demonstrated remarkable success in image generation, but they are computationally intensive and time-consuming to train. In this paper, we introduce a novel diffusion model that benefits from quantum computing techniques in order to mitigate computational challenges and enhance generative performance within high energy physics data. The fully quantum diffusion model replaces Gaussian noise with random unitary matrices in the forward process and incorporates a variational quantum circuit within the U-Net in the denoising architecture. We run evaluations on the structurally complex quark and gluon jets dataset from the Large Hadron Collider. The results demonstrate that the fully quantum and hybrid models are competitive with a similar classical model for jet generation, highlighting the potential of using quantum techniques for machine learning problems.
Quantum generative models offer a promising new direction in machine learning by leveraging quantum circuits to enhance data generation capabilities. In this study, we propose a hybrid quantum-classical image generation framework that integrates variational quantum circuits into a diffusion-based model. To improve training dynamics and generation quality, we introduce two novel noise strategies: intrinsic quantum-generated noise and a tailored noise scheduling mechanism. Our method is built upon a lightweight U-Net architecture, with the quantum layer embedded in the bottleneck module to isolate its effect. We evaluate our model on MNIST and MedMNIST datasets to examine its feasibility and performance. Notably, our results reveal that under limited data conditions (fewer than 100 training images), the quantum-enhanced model generates images with higher perceptual quality and distributional similarity than its classical counterpart using the same architecture. While the quantum model shows advantages on grayscale data such as MNIST, its performance is more nuanced on complex, color-rich datasets like PathMNIST. These findings highlight both the potential and current limitations of quantum generative models and lay the groundwork for future developments in low-resource and biomedical image generation.
Generative Artificial Intelligence is one of the most relevant technologies of our century, enabling digital machines not only to analyze or classify data but also to generate new content such as images, text, and video. This technology relies on deep learning algorithms designed to capture complex relationships from finite data sets. In this context, Diffusion Models have recently emerged as state-of-the-art in deep generative modeling thanks to their ability to produce high-quality data. Their pipeline typically consists of a forward diffusion process, where data samples are progressively corrupted into pure (classical) noise, and a reverse denoising process, where deep neural networks learn to reverse the diffusion process and then generate new synthetic samples. In parallel, Quantum Machine Learning represents one of the most promising applications of quantum computing. However, currently available quantum processors, known as Noisy Intermediate-Scale Quantum (NISQ) devices, are composed of tens or hundreds of noisy qubits. Therefore, designing scalable and efficient practical quantum algorithms represents a significant challenge. This thesis proposes the first generalization of classical Denoising Diffusion Probabilistic Models to the quantum domain, introducing the novel generative paradigm of Quantum Diffusion Models. In particular, the three possible scenarios are formalized and implemented: (i) classical data corrupted with Gaussian noise and a reverse process implemented via a quantum neural network; (ii) quantum data corrupted with quantum noise and a denoising process through a classical deep neural network; and (iii) a fully quantum approach, where both forward and backward processes are implemented within a quantum framework. Subsequently, the impact of different stochastic dynamics in the forward process is analyzed using the formalism of Quantum Stochastic Walks. We observe that a specific hybrid quantum-classical dynamics produces more statistically robust models, capable of generating sets of images closely resembling real ones compared to purely classical dynamics. Furthermore, we propose a hybrid quantum-classical diffusion model based on Quantum Walks that exploits the intrinsic noise of real quantum processors as a useful resource for generating images. This model is implementable on current NISQ devices and scalable, as it allows the generation of images of any size, requiring q + 1 qubits for grayscale images with pixel intensities in the range [0, 2^q - 1]. Finally, an application in the context of medical imaging is presented, specifically the generation of 64 × 64 color images of blood cells from the BloodMNIST dataset and 200 × 200 brain MRI images from the BraTS2020 dataset.
Generative models have achieved remarkable success in data synthesis, though recent advances driven by increasing model scale have introduced challenges in computational cost and efficiency. Quantum machine learning offers a promising alternative, representing complex data distributions using compact, highly expressive models. Here, we propose QDiffusion-TS, the first quantum generative diffusion model for time series synthesis, and validate it on the IQM quantum processor. The framework extends a classical diffusion architecture by replacing feed-forward components within the denoising transformer with quantum neural networks, yielding a hybrid quantum transformer that reduces the number of trainable parameters in each replaced component by nearly three orders of magnitude. Evaluated on financial time series from Apple and Amazon, the model generates synthetic data that more accurately reproduces the real distributions, reducing Wasserstein distance by approximately 44% relative to its classical counterpart across both datasets. In a downstream forecasting task, augmentation with the generated data improves predictive performance by up to 71% in RMSE over a baseline trained solely on real data. These results show that quantum enhanced architectures can consistently match and frequently surpass classical performance with substantially fewer parameters, establishing a practical framework towards more efficient and scalable data-driven generative modelling.
Machine Learning algorithms are playing a fundamental role in solving High Energy Physics tasks. In particular, the classification of hadronic jets at the Large Hadron Collider is suited for such types of algorithms, and despite the great effort that has been put in place to tackle such a classification task, there is room for improvement. In this context, Quantum Machine Learning is a new methodology that takes advantage of the intrinsic properties of quantum computation (e.g. entanglement between qubits) to possibly improve the performance of a classification task. In this contribution, a new study of Quantum Machine Learning applied to jet identification is presented. Namely, a Variational Quantum Classifier is trained and evaluated on fully simulated data of the LHCb experiment, in order to identify jets containing a hadron formed by a b or bˉ quark at the moment of production. The jet identification performance of the quantum classifier is compared with a Deep Neural Network using the same input features.
In recent years, machine learning models like DALL-E, Craiyon, and Stable Diffusion have gained significant attention for their ability to generate high-resolution images from concise descriptions. Concurrently, quantum computing is showing promising advances, especially with quantum machine learning which capitalizes on quantum mechanics to meet the increasing computational requirements of traditional machine learning algorithms. This paper explores the integration of quantum machine learning and variational quantum circuits to augment the efficacy of diffusion-based image generation models. Specifically, we address two challenges of classical diffusion models: their low sampling speed and the extensive parameter requirements. We introduce two quantum diffusion models and benchmark their capabilities against their classical counterparts using MNIST digits, Fashion MNIST, and CIFAR-10. Our models surpass the classical models with similar parameter counts in terms of performance metrics FID, SSIM, and PSNR. Moreover, we introduce a consistency model unitary single sampling architecture that combines the diffusion procedure into a single step, enabling a fast one-step image generation.
Deep generative models are key-enabling technology to computer vision, text generation, and large language models. Denoising diffusion probabilistic models (DDPMs) have recently gained much attention due to their ability to generate diverse and high-quality samples in many computer vision tasks, as well as to incorporate flexible model architectures and a relatively simple training scheme. Quantum generative models, empowered by entanglement and superposition, have brought new insight to learning classical and quantum data. Inspired by the classical counterpart, we propose the quantum denoising diffusion probabilistic model (QuDDPM) to enable efficiently trainable generative learning of quantum data. QuDDPM adopts sufficient layers of circuits to guarantee expressivity, while it introduces multiple intermediate training tasks as interpolation between the target distribution and noise to avoid barren plateau and guarantee efficient training. We provide bounds on the learning error and demonstrate QuDDPM's capability in learning correlated quantum noise model, quantum many-body phases, and topological structure of quantum data. The results provide a paradigm for versatile and efficient quantum generative learning.