Interpretable representation learning of quantum data enabled by probabilistic variational autoencoders

Abstract

Interpretable machine learning is rapidly becoming a crucial tool for scientific discovery. Among existing approaches, variational autoencoders (VAEs) have shown promise in extracting the hidden physical features of some input data, with no supervision nor prior knowledge of the system at study. Yet, the ability of VAEs to create meaningful, interpretable representations relies on their accurate approximation of the underlying probability distribution of their input. When dealing with quantum data, VAEs must hence account for its intrinsic randomness and complex correlations. While VAEs have been previously applied to quantum data, they have often neglected its probabilistic nature, hindering the extraction of meaningful physical descriptors. Here, we demonstrate that two key modifications enable VAEs to learn physically meaningful latent representations: a decoder capable of faithfully reproduce quantum states and a probabilistic loss tailored to this task. Using benchmark quantum spin models, we identify regimes where standard methods fail while the representations learned by our approach remain meaningful and interpretable. Applied to experimental data from Rydberg atom arrays, the model autonomously uncovers the phase structure without access to prior labels, Hamiltonian details, or knowledge of relevant order parameters, highlighting its potential as an unsupervised and interpretable tool for the study of quantum systems.

Main text 10 pages, total document 16 pages, 10 figures

References in corpus (23)

• Solving the Quantum Many-Body Problem with Artificial Neural Networks
• Machine learning phases of matter
• Quantum Phases of Matter on a 256-Atom Programmable Quantum Simulator
• Observation of Rydberg blockade between two atoms
• Observation of entanglement propagation in a quantum many-body system
• Unsupervised learning of phase transitions: from principal component analysis to variational autoencoders
• Discovering physical concepts with neural networks
• Spread of correlations in long-range interacting quantum systems
• Long-range Ising and Kitaev Models: Phases, Correlations and Edge Modes
• NetKet 3: Machine Learning Toolbox for Many-Body Quantum Systems
• Kernel methods for interpretable machine learning of order parameters
• Identifying Quantum Phase Transitions with Adversarial Neural Networks
• Correlator Convolutional Neural Networks: An Interpretable Architecture for Image-like Quantum Matter Data
• Machine learning discovery of new phases in programmable quantum simulator snapshots
• Deep Learning on the 2-Dimensional Ising Model to Extract the Crossover Region with a Variational Autoencoder
• Dynamical structure factors of dynamical quantum simulators
• Quantum circuit synthesis with diffusion models
• Operationally meaningful representations of physical systems in neural networks
• Mapping out phase diagrams with generative classifiers
• Learning minimal representations of stochastic processes with variational autoencoders
• Explainable Representation Learning of Small Quantum States
• Unsupervised Learning of Rydberg Atom Array Phase Diagram with Siamese Neural Networks
• Predicting Topological Entanglement Entropy in a Rydberg analog simulator