Accurate machine learning of rate coefficients for state-to-state transitions in molecular collisions

Abstract

We present an algorithm that combines quantum scattering calculations with probabilistic machine-learning models to predict quantum dynamics rate coefficients for a large number of state-to-state transitions in molecule–molecule collisions much faster than with direct solutions of the Schrödinger equation. By utilizing the predictive power of Gaussian process regression with kernels, optimized to make accurate predictions outside of the input parameter space, the present strategy reduces the computational cost by about 75%, with an accuracy within 5%. Our method uses temperature dependences of rate coefficients for transitions from the isolated states of initial rotational angular momentum j, determined via explicit calculations, to predict the temperature dependences of rate coefficients for other values of j. The approach, demonstrated here for rovibrational transitions of SiO due to thermal collisions with H2, uses different prediction models and is thus adaptive to various time and accuracy requirements. The procedure outlined in this work can be used to extend multiple inelastic molecular collision databases without exponentially large computational resources required for conventional rigorous quantum dynamics calculations.