A Physics-Informed Neural Network Scheme for Shortcuts to Adiabaticity in Three-Level Non-Hermitian Quantum Systems

Abstract

We propose a physics-informed neural network (PINN) scheme for designing shortcuts to adiabaticity in a three-level non-Hermitian quantum system. The PINN is used to solve an inverse control problem in which the state amplitudes and the auxiliary driving field are learned simultaneously from the Schrödinger residual, the initial and target population constraints and a probability conservation constraint on the control pulse. The learned compensation field counteracts the loss of the intermediate state and enables high-fidelity population inversion in an open-system setting. Importantly, the imposed probability conservation is treated as an auxiliary constraint along the learned trajectory rather than as an intrinsic property of the non-Hermitian Hamiltonian. Under this constrained evolution, the Hamiltonian expectation value evaluated on the obtained state remains real within numerical accuracy. Numerical simulations and independent propagation with the fitted control field verify the population inversion and exhibit strong generalization capability over a range of coupling strengths and dissipation rates, as verified by retraining the network independently for each parameter set. The results demonstrate that PINNs provide a flexible inverse design tool for non-Hermitian shortcut to adiabaticity protocols when the governing dynamics and physical constraints are explicitly incorporated into the loss function.