Quantum State Generation with Structure-Preserving Diffusion Model (2404.06336v2)
Abstract: This article considers the generative modeling of the (mixed) states of quantum systems, and an approach based on denoising diffusion model is proposed. The key contribution is an algorithmic innovation that respects the physical nature of quantum states. More precisely, the commonly used density matrix representation of mixed-state has to be complex-valued Hermitian, positive semi-definite, and trace one. Generic diffusion models, or other generative methods, may not be able to generate data that strictly satisfy these structural constraints, even if all training data do. To develop a machine learning algorithm that has physics hard-wired in, we leverage mirror diffusion and borrow the physical notion of von Neumann entropy to design a new map, for enabling strict structure-preserving generation. Both unconditional generation and conditional generation via classifier-free guidance are experimentally demonstrated efficacious, the latter enabling the design of new quantum states when generated on unseen labels.
- Anderson, B. D. Reverse-time diffusion equation models. Stochastic Processes and their Applications, 12(3):313–326, 1982.
- Structured denoising diffusion models in discrete state-spaces. Advances in Neural Information Processing Systems, 34:17981–17993, 2021.
- Geometry of quantum states: an introduction to quantum entanglement. Cambridge university press, 2017.
- Align your latents: High-resolution video synthesis with latent diffusion models. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 22563–22575, 2023.
- Finite-time analysis of projected langevin monte carlo. Advances in Neural Information Processing Systems, 28, 2015.
- Reconstructing quantum states with generative models. Nature Machine Intelligence, 1(3):155–161, 2019.
- Quantum wasserstein generative adversarial networks. Advances in Neural Information Processing Systems, 32, 2019.
- Quantum generative diffusion model. arXiv preprint arXiv:2401.07039, 2024.
- Deep momentum multi-marginal schr\\\backslash\” odinger bridge. NeurIPS, 2023.
- Fast mcmc sampling algorithms on polytopes. The Journal of Machine Learning Research, 19(1):2146–2231, 2018.
- Quantum generative adversarial networks. Physical Review A, 98(1):012324, 2018.
- Max-sliced wasserstein distance and its use for gans. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 10648–10656, 2019.
- Diffusion models beat gans on image synthesis. Advances in neural information processing systems, 34:8780–8794, 2021.
- Dieleman, S. Guidance: a cheat code for diffusion models, 2022. URL https://benanne.github.io/2022/05/26/guidance.html.
- Accurate transition state generation with an object-aware equivariant elementary reaction diffusion model. Nature Computational Science, 2023.
- Fisher, D. S. Random transverse field ising spin chains. Physical review letters, 69(3):534, 1992.
- Fisher, D. S. Critical behavior of random transverse-field ising spin chains. Physical review b, 51(10):6411, 1995.
- Diffusion models for constrained domains. TMLR, 2023a.
- Metropolis sampling for constrained diffusion models. NeurIPS, 2023b.
- Preserve your own correlation: A noise prior for video diffusion models. In Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 22930–22941, 2023.
- A kernel two-sample test. The Journal of Machine Learning Research, 13(1):723–773, 2012.
- Time reversal of diffusions. The Annals of Probability, pp. 1188–1205, 1986.
- Heaven, W. D. Horse riding astronaut, 2022. URL https://www.technologyreview.com/2022/04/06/1049061/dalle-openai-gpt3-ai-agi-multimodal-image-generation/.
- Classifier-free diffusion guidance. arXiv preprint arXiv:2207.12598, 2022.
- Denoising diffusion probabilistic models. Advances in neural information processing systems, 33:6840–6851, 2020.
- Cascaded diffusion models for high fidelity image generation. The Journal of Machine Learning Research, 23(1):2249–2281, 2022.
- Equivariant diffusion for molecule generation in 3d. In International conference on machine learning, pp. 8867–8887. PMLR, 2022.
- Predicting many properties of a quantum system from very few measurements. Nature Physics, 16(10):1050–1057, 2020.
- Provably efficient machine learning for quantum many-body problems. Science, 377(6613):eabk3333, 2022.
- Learning energy-based representations of quantum many-body states. arXiv preprint arXiv:2304.04058, 2023.
- Highly accurate protein structure prediction with alphafold. Nature, 596(7873):583–589, 2021.
- Random walks on polytopes and an affine interior point method for linear programming. In Proceedings of the forty-first annual ACM symposium on Theory of computing, pp. 561–570, 2009.
- Quantum variational autoencoder. Quantum Science and Technology, 4(1):014001, 2018.
- Generalized sliced wasserstein distances. Advances in neural information processing systems, 32, 2019.
- Convergence of kinetic langevin monte carlo on lie groups. arXiv:2403.12012, 2024.
- Kovács, I. A. Quantum multicritical point in the two-and three-dimensional random transverse-field ising model. Physical Review Research, 4(1):013072, 2022.
- The mirror langevin algorithm converges with vanishing bias. In International Conference on Algorithmic Learning Theory, pp. 718–742. PMLR, 2022.
- Mirror diffusion models for constrained and watermarked generation. NeurIPS, 2023.
- Reflected diffusion models. ICML, 2023.
- Discrete diffusion language modeling by estimating the ratios of the data distribution. NeurIPS, 2023.
- Lovász, L. Hit-and-run mixes fast. Mathematical programming, 86:443–461, 1999.
- Random walks in a convex body and an improved volume algorithm. Random structures & algorithms, 4(4):359–412, 1993.
- Diffusion probabilistic models for 3d point cloud generation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 2837–2845, 2021.
- Residual diffusion modeling for km-scale atmospheric downscaling. 2024.
- Measurement-altered ising quantum criticality. Physical Review X, 13(4):041042, 2023.
- Quantum computation and quantum information. Cambridge university press, 2010.
- Rockafellar, R. Convex analysis. Princeton landmarks in mathematics and physics. Princeton University Press, 2015.
- High-resolution image synthesis with latent diffusion models. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 10684–10695, 2022.
- Deep unsupervised learning using nonequilibrium thermodynamics. ICML, 2015.
- Score-based generative modeling through stochastic differential equations. ICLR, 2021.
- Qugan: A quantum state fidelity based generative adversarial network. In 2021 IEEE International Conference on Quantum Computing and Engineering (QCE), pp. 71–81. IEEE, 2021.
- Variational optimization on lie groups, with examples of leading (generalized) eigenvalue problems. In International Conference on Artificial Intelligence and Statistics, pp. 4269–4280. PMLR, 2020.
- Latent space purification via neural density operators. Physical review letters, 120(24):240503, 2018.
- Neural-network quantum state tomography. Nature Physics, 14(5):447–450, 2018.
- Positive-definite parametrization of mixed quantum states with deep neural networks. arXiv preprint arXiv:2206.13488, 2022.
- Vincent, P. A connection between score matching and denoising autoencoders. Neural computation, 23(7):1661–1674, 2011.
- Broadly applicable and accurate protein design by integrating structure prediction networks and diffusion generative models. BioRxiv, pp. 2022–12, 2022.
- Watson, J. L. et al. De novo design of protein structure and function with rfdiffusion. Nature, 2023.
- Lion: Latent point diffusion models for 3d shape generation. arXiv preprint arXiv:2210.06978, 2022.
- Generative quantum machine learning via denoising diffusion probabilistic models. Physical Review Letters, 132(10):100602, 2024.
- Wasserstein control of mirror langevin monte carlo. In Conference on Learning Theory, pp. 3814–3841. PMLR, 2020.