Learning Spatiotemporally Correlated Noise in Multi-Qubit Systems with Neural Networks

Noisy quantum processors require precise and coherent quantum control to effectively execute quantum algorithms. Key to achieving precise control is the characterization of a quantum device to elucidate important details of environmentally-induced and systematic error sources. Quantum noise spectroscopy (QNS) is one such protocol used to characterize spatiotemporally correlated noise. QNS strives to provide estimates of noise statistical properties by using the quantum system as a dynamical probe. Despite its success, QNS can be challenging as it requires model selection prior to protocol implementation and suffers from scalability issues as system size grows. Here, we leverage machine learning to overcome model selection and assist in learning complex noise environments. In particular, we utilize physics-informed neural networks based on the “gray box” model to learn the dynamics of a two-qubit system subject to spatiotemporally correlated noise. Using training data sets based on dynamical decoupling, we show that the model is capable of learning the underlying power spectral densities for each qubit and the strength of parasitic qubit-qubit interactions. Furthermore, we show that the model can predict state-dependent behavior for multi-qubit observables. The results of this study further display the potential utility of machine learning to assist in the domain of quantum characterization and control.