Towards Robust Variational Quantum Simulation of Lindblad Dynamics via Stochastic Magnus Expansion

In this paper, we introduce a novel and general framework for the variational quantum simulation of Lindblad equations. Building on the close relationship between the unraveled Lindblad dynamics, stochastic Magnus integrators, and variational quantum simulation, we propose a high-order scheme for solving the quantum state diffusion equation using exponential integrators. This formulation facilitates the simulation of wavefunction trajectories within the established framework of variational quantum algorithms for time evolution. Our algorithm significantly enhances robustness in two key aspects: the stability of the simulation with large time steps, and the reduction in the number of quantum trajectories required to accurately simulate the Lindblad dynamics in terms of the ensemble average. We demonstrate the effectiveness of our algorithm through numerical examples in both classical and quantum implementations, including the transverse-field Ising model (TFIM) with damping, the Fenna-Matthews-Olson (FMO) complex, and the radical pair model (RPM). The simulation accuracy can be systematically improved, and the algorithm remains reliable even in highly oscillatory regimes. These methods are expected to be applicable to a broader class of open quantum systems beyond the specific models considered in this study.