Fast surrogate modelling of EIT in atomic quantum systems using LSTM neural networks

Abstract

Simulations of optical quantum systems are essential for the development of quantum technologies. However, these simulations are often computationally intensive, especially when repeated evaluations are required for data fitting, parameter estimation, or real-time feedback. To address this challenge, we develop a Long Short-Term Memory neural network capable of replicating the output of these simulations with high accuracy and significantly reduced computational cost. Once trained, the surrogate model produces spectra in milliseconds, providing a speed-up of 5000x relative to traditional numerical solvers. We focus on applying this technique to Doppler-broadened Electromagnetically Induced Transparency in a ladder-type scheme for Rydberg-based sensing, achieving near-unity agreement with the physics solver for resonant and off-resonant regimes. We demonstrate the effectiveness of the LSTM model on this representative optical quantum system, establishing it as a surrogate tool capable of supporting real-time signal processing and feedback-based optimisation.