Optimal control of open-quantum-system dynamics predicted by long short-term memory

Abstract

The realization of high-fidelity quantum control is crucial for quantum information processing, particularly in noisy environments where control strategies must simultaneously achieve precise manipulation and effective noise suppression. Conventional optimal control designs typically require numerical calculations of the system dynamics. Recent studies have demonstrated that long short-term memory neural networks (LSTM) can accurately predict the time evolution of open quantum systems. Based on LSTM predicted dynamics, we propose an optimal control framework for rapid and efficient optimal control design in open quantum systems. As illustrative examples, we apply the proposed framework to design optimal control for adiabatic speedup in a two-level system and for quantum state transfer in a spin chain, both under non-Markovian environments. For adiabatic speedup, our optimization procedure involves two steps: driving trajectory optimization and zero-area pulse optimization. Fidelity improvements for both steps have been obtained, demonstrating the effectiveness of the scheme. Furthermore, this effectiveness is validated for quantum state transfer in a spin chain, which is a high dimensional control problem. Our optimal control design scheme utilizes predicted dynamics to generate optimized controls, offering broad application potential in quantum computing, communication, and sensing.