A comparative study of different machine learning methods for dissipative quantum dynamics

Abstract

It has been recently shown that supervised machine learning (ML) algorithms can accurately and efficiently predict the long-time populations dynamics of dissipative quantum systems given only short-time population dynamics. In the present article we benchmaked 22 ML models on their ability to predict long-time dynamics of a two-level quantum system linearly coupled to harmonic bath. The models include uni- and bidirectional recurrent, convolutional, and fully-connected feed-forward artificial neural networks (ANNs) and kernel ridge regression (KRR) with linear and most commonly used nonlinear kernels. Our results suggest that KRR with nonlinear kernels can serve as inexpensive yet accurate way to simulate long-time dynamics in cases where the constant length of input trajectories is appropriate. Convolutional Gated Recurrent Unit model is found to be the most efficient ANN model.

22 pages, 6 figures

References in corpus (4)

• Deep Learning in Neural Networks: An Overview
• Perspective: Numerically "exact" approach to open quantum dynamics: The hierarchical equations of motion (HEOM)
• Initial System-Environment Correlations via the Transfer Tensor Method
• One-shot trajectory learning of open quantum systems dynamics

Cited by in corpus (11)

• MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflows
• Four-Dimensional-Spacetime Atomistic Artificial Intelligence Models
• QD3SET-1: A Database with Quantum Dissipative Dynamics Data Sets
• Quantum Next Generation Reservoir Computing: An Efficient Quantum Algorithm for Forecasting Quantum Dynamics
• Kernel-based quantum regressor models learn non-Markovianity
• Machine learning meets $\mathfrak{su}(n)$ Lie algebra: Enhancing quantum dynamics learning with exact trace conservation
• Machine learning for predicting control landscape maps of quantum molecular dynamics: Laser-induced three-dimensional alignment of asymmetric top molecules
• Multi-timescale time encoding for CNN prediction of Fenna-Matthews-Olson energy-transfer dynamics
• Compact representation and long-time extrapolation of real-time data for quantum systems using the ESPRIT algorithm
• Toward Quantum-Aware Machine Learning: Improved Prediction of Quantum Dissipative Dynamics via Complex Valued Neural Networks
• Theory of Moment Propagation for Quantum Dynamics in Single-Particle Description