Transformer-Based Neural Quantum Digital Twins for Many-Body Quantum Simulation and Optimal Annealing Schedule Design
Jianlong Lu, Hanqiu Peng, Ying Chen
TL;DR
The paper tackles the challenge of designing high-fidelity, instance-specific annealing schedules for quantum annealers by reconstructing the full spectral evolution along the anneal path. It introduces Tx-NQDT, a graph-aware Transformer-based digital twin that predicts ground- and first-excited-state energies and coupling matrix elements along s, using Brauer deflation for excited states and transfer learning to propagate across the path. These spectral predictions feed a first-order adiabatic perturbation theory (FOAPT)–based control functional to generate adaptive, hardware-constrained schedules that slow near small gaps. End-to-end validation on D-Wave hardware shows substantial performance gains over linear schedules across multiple problem sizes and difficulty levels, confirming the practicality of spectrum-aware, data-driven control in noisy, real-world devices. The work highlights a scalable, graph-aware neural surrogate for adiabatic dynamics that can guide efficient quantum annealing and motivate future extensions to broader Hamiltonians and uncertainty-aware control.
Abstract
We introduce Transformer-based Neural Quantum Digital Twins (Tx-NQDTs) to simulate full adiabatic dynamics of many-body quantum systems, including ground and low-lying excited states, at low computational cost. Tx-NQDTs employ a graph-informed Transformer neural network trained to predict spectral properties (energy levels and gap locations) needed for annealing schedule design. We integrate these predictions with an adaptive annealing schedule design based on first-order adiabatic perturbation theory (FOAPT), which slows the evolution near predicted small gaps to maintain adiabaticity. Experiments on a D-Wave quantum annealer (N = 10, 15, 20 qubits, 12 control segments) show that Tx-NQDT-informed schedules significantly improve success probabilities despite hardware noise and calibration drift. The optimized schedules achieve success probabilities 2.2-11.7 percentage points higher than the default linear schedule, outperforming the D-Wave baseline in 44 of 60 cases. These results demonstrate a practical, data-driven route to improved quantum annealing performance on real hardware.
