arXiv:2603.05468

Kraus Constrained Sequence Learning For Quantum Trajectories from Continuous Measurement

Priyanshi Singh, Krishna Bhatia

Published Mar 5, 2026• 0 citations

Abstract

Real-time reconstruction of conditional quantum states from continuous measurement records is a fundamental requirement for quantum feedback control, yet standard stochastic master equation (SME) solvers require exact model specification, known system parameters, and are sensitive to parameter mismatch. While neural sequence models can fit these stochastic dynamics, the unconstrained predictors can violate physicality such as positivity or trace constraints, leading to unstable rollouts and unphysical estimates. We propose a Kraus-structured output layer that converts the hidden representation of a generic sequence backbone into a completely positive trace preserving (CPTP) quantum operation, yielding physically valid state updates by construction. We instantiate this layer across diverse backbones, RNN, GRU, LSTM, TCN, ESN and Mamba; including Neural ODE as a comparative baseline, on stochastic trajectories characterized by parameter drift. Our evaluation reveals distinct trade-offs between gating mechanisms, linear recurrence, and global attention. Across all models, Kraus-LSTM achieves the strongest results, improving state estimation quality by 7% over its unconstrained counterpart while guaranteeing physically valid predictions in non-stationary regimes.

DOI: 10.48550/arxiv.2603.05468

Related papers

Continuous Quantum Feedback Control via Kraus-Parameterized Belief Reinforcement Learning

Priyanshi Singh, Krishna Bhatia

Quantum feedback control requires acting on noisy continuous measurement records without direct access to the underlying quantum state. We propose Kraus-Parameterized Belief Reinforcement Learning, a pipeline in which a recurrent encoder, constrained to the Stiefel manifold, produces density-matrix estimates that are guaranteed positive-semidefinite and trace-normalized by construction, embedding quantum state geometry directly into the learning loop. A Proximal Policy Optimization (PPO) actor then maps these physically valid belief states to continuous control actions. On a simulated continuously monitored qubit, the resulting policy achieves stable feedback control, maintaining a measurement-conditioned belief fidelity of approximately 0.77-0.80 and exhibiting substantially lower return variance than a parameter-matched LSTM-history baseline across both nominal and out-of-distribution conditions. Although gains in raw target fidelity are modest, the geometric constraint guarantees a physically valid, interpretable belief representation and yields markedly more stable control under measurement inefficiency and abrupt dynamics switches. These results indicate that physics-informed neural memory is a practical inductive bias for reliable quantum feedback control.

quant-pharXiv:2608.15715Aug 18, 20260 citations

Eliminating Secular Density Drift and Positivity Violation in Open Quantum Neural Simulators via Idempotent Quantum State Manifold Projections

A. Emre Cetin

Simulating open quantum multi-body systems governed by the Gorini-Kossakowski-Sudarshan-Lindblad (GKSL) master equation is essential for quantum optimal control, superconducting qubit characterization, and quantum error mitigation. However, when deploying Physics-Informed Neural Networks (PINNs) or Neural Ordinary Differential Equations (Neural ODEs) as accelerated surrogates for open quantum dynamics, long-horizon autoregressive rollouts suffer from secular drift: (1) trace conservation breakdown (|Tr(ρ) − 1| > 0), violating total probability conservation; (2) loss of positive semi-definiteness (λmin(ρ) < 0), producing unphysical negative quantum measurement probabilities; and (3) distortion of purity and decoherence envelopes. While static metric projection of noisy density matrices onto the state spectrahedron is established in quantum tomography (Smolin et al., PRL 2012; Michelot 1986), its rigorous dynamical integration into autoregressive scientific machine learning (SciML) time-stepping loops remains unexplored. In this paper, we formulate an exact, non-dissipative dynamical framework: Idempotent Quantum State Manifold Projections (ΠS2 = ΠS) integrated via operator splitting into neural surrogate rollouts. We prove that ΠS is firmly non-expansive, preserving the high-order convergence rate of the underlying differential integrator while unconditionally eradicating secular trace runaway and negative probabilities without damping physical coherences. On a canonical benchmark of a driven-dissipative superconducting qubit under spontaneous relaxation and dephasing (4,000 integration steps, t = 20.0, replicated over 10 independent random seeds): Unconstrained neural rollouts suffer severe trace drift (|Tr(ρ) − 1| = (3.91 ± 1.25) × 10−3) and negative eigenvalues (λmin = (−1.21 ± 0.82) × 10−4); A standard Cholesky regularization baseline (ρ = L L† / Tr(L L†) with adaptive diagonal jitter εI) requires artificial diagonal entropy injection that dilutes low-entropy states; IdemSolver ΠS maintains exact unit trace ((8.44 ± 0.89) × 10−16), strictly non-negative eigenvalues (λmin ≥ 0), pristine Rabi coherence, exact steady-state purity (0.5244 ± 0.0005 matching the exact physical Lindblad baseline 0.5244), and machine-zero idempotency residual ((1.10 ± 0.08) × 10−15); and Achieves a step latency of 122.70 ± 4.70 μs (>8.1 kHz update rate), establishing a verified algebraic safeguard for real-time quantum control and SciML digital twins.

Sep 19, 20260 citations

Quantum-Enhanced Recurrent Neural Networks via Variational Quantum Gating for Battery State of Health Prediction

Yin Xu, Qinglin Liu, Li Gao +1 more

Accurate state-of-health (SOH) estimation for lithium-ion batteries remains a challenging problem due to complex electrochemical degradation mechanisms and long-range temporal dependencies. In this work, we propose a quantum-enhanced recurrent framework, termed QLSTM, in which variational quantum circuits are directly embedded into the gating mechanisms of long short-term memory networks. By replacing classical affine transformations with parameterized unitary operations, the proposed model introduces structured nonlinear transformations into the recurrent state-transition process. Extensive experiments on multiple benchmark battery datasets demonstrate that QLSTM consistently outperforms classical sequence models in both predictive accuracy and robustness, achieving significant reductions in mean absolute error (MAE), with improvements on the order of 20% compared with classical LSTM baselines. Ablation studies further confirm that these improvements arise primarily from quantum-enhanced gating rather than input-level transformations. Additional analyses on qubit scaling and noise robustness reveal that model performance is governed by a balance between expressive capacity and trainability. These results provide empirical evidence that embedding quantum computational primitives within recurrent architectures offers a structurally grounded approach to improving sequence modeling capability. The proposed framework establishes a new design paradigm for integrating quantum operators into temporal learning models, with potential applications in complex dynamical system prediction tasks.

quant-pharXiv:2604.20438Apr 23, 20260 citations

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

Qamar, Shahid, Mushtaq Rana Imran, Bo Li +1 more

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.

arXiv:2509.07196Sep 8, 20250 citations

Learning quantum ground states in the space of measurement outcomes

Kartiek Agarwal

We investigate variational learning of quantum many-body ground states directly in measurement space using autoregressive neural networks. In particular, we represent quantum states via probability distributions of outcomes over a symmetric informationally complete positive operator-valued measure (SIC-POVM). The probability distribution is encoded in the parameters of an autoregressive neural-network-based on gated recurrent units (GRUs). Ground states are obtained by gradient descent that updates the probability distribution to minimize the energy with respect to a given Hamiltonian, while enforcing positivity constraints that ensure that the distribution of measurement outcomes correspond to a physical quantum state. We analyze the role of constraint enforcement (hierarchy of positivity conditions), variety of neural network architectures (multiple layers, dilation, and modifications of input data) in determining the success of this approach. We benchmark our approach on one-dimensional transverse-field Ising model and the Heisenberg model, along with gapping fields, for system sizes up to L=128, illustrating its efficacy across a wide variety of models.

quant-pharXiv:2605.28931May 29, 20260 citations

Using a Recurrent Neural Network to Reconstruct Quantum Dynamics of a Superconducting Qubit from Physical Observations

Emmanuel Flurin, Leigh S. Martin, Shay Hacohen-Gourgy +1 more

At its core, quantum mechanics is a theory developed to describe fundamental observations in the spectroscopy of solids and gases.Despite these practical roots, however, quantum theory is infamous for being highly counterintuitive, largely due to its intrinsically probabilistic nature.Neural networks have recently emerged as a powerful tool that can extract nontrivial correlations in vast datasets.These networks routinely outperform state-of-the-art techniques in language translation, medical diagnosis, and image recognition.It remains to be seen if neural networks can be trained to predict stochastic quantum evolution without a priori specifying the rules of quantum theory.Here, we demonstrate that a recurrent neural network can be trained in real time to infer the individual quantum trajectories associated with the evolution of a superconducting qubit under unitary evolution, decoherence, and continuous measurement from physical observations only.The network extracts the system Hamiltonian, measurement operators, and physical parameters.It is also able to perform tomography of an unknown initial state without any prior calibration.This method has the potential to greatly simplify and enhance tasks in quantum systems such as noise characterization, parameter estimation, feedback, and optimization of quantum control.

arXiv:1811.12420Physical Review XJan 9, 202088 citations