Quantum Filtering and Stabilization of Dissipative Quantum Systems via Augmented Neural Ordinary Differential Equations

Abstract: Modeling open quantum dynamics without full knowledge of the system Hamiltonian or noise model is a key challenge in quantum control and quantum state estimation. We introduce an Augmented Quantum Neural Ordinary Differential Equation (AQNODE) framework that learns quantum trajectories and dissipation parameters directly from partial continuous measurement data. By embedding the system into a latent space evolved via neural ODEs, AQNODE captures both observable and hidden non-Markovian dynamics with temporal smoothness and physical consistency. Our approach integrates weak measurement data to reconstruct qubit states and time-dependent decoherence rates, enabling accurate state prediction and parameter inference without explicit physical equations. Furthermore, we incorporate AQNODE-based feedback control techniques, including proportional-derivative and time-varying linear-quadratic regulator (LQR) strategies, to steer the quantum system toward target states in real time. Extensive numerical simulations demonstrate AQNODE's ability to generalize across system configurations, achieve low prediction errors, and perform robust quantum filtering and control. These results establish AQNODE as a scalable, differentiable, and experimentally compatible framework for real-time modeling and control of dissipative quantum systems.