Search papers, labs, and topics across Lattice.
This paper introduces a Kraus-structured output layer for neural sequence models to ensure physically valid quantum state updates when learning from continuous measurement records. The Kraus layer converts hidden representations into completely positive trace preserving (CPTP) quantum operations, addressing the issue of unphysical predictions from unconstrained models. Experiments across diverse sequence backbones, including RNN, GRU, LSTM, TCN, ESN, Mamba, and Neural ODE, demonstrate that Kraus-LSTM achieves the best state estimation, improving quality by 7% over unconstrained LSTM while guaranteeing physicality in non-stationary regimes.
Guaranteeing physically valid quantum state predictions is now possible, as a Kraus-structured layer improves state estimation by 7% over unconstrained models.
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.
