Diabatic neural network potentials for accurate vibronic quantum dynamics-The test case of planar NO₃

A recently developed scheme to produce high-dimensional coupled diabatic potential energy surfaces (PESs) based on artificial neural networks (ANNs) [D. M. G. Williams and W. Eisfeld, J. Chem. Phys. 149, 204106 (2019)] is tested for its viability for quantum dynamics applications. The method, capable of reproducing high-quality ab initio data with excellent accuracy, utilizes simple coupling matrices to produce a basic low-order diabatic potential matrix as an underlying backbone for the model. This crude model is then refined by making its expansion coefficients geometry-dependent by the output neurons of the ANN. This structure, strongly guided by a straightforward physical picture behind nonadiabatic coupling, combines structural simplicity with high accuracy, reproducing ab initio data without introducing unphysical artifacts to the surface, even for systems with complicated electronic structure. The properties of diabatic potentials obtained by this method are tested thoroughly in the present study. Vibrational/vibronic eigenstates are computed on the X̃ and à states of NO₃, a notoriously difficult Jahn-Teller system featuring strong nonadiabatic couplings and complex spectra. The method is investigated in terms of how consistently it produces dynamics results for PESs of similar (fitting) quality and how the results depend on the ANN size and ANN topography. A central aspect of this work is to understand the convergence properties of the new method in order to evaluate its predictive power. A previously developed, high-quality model utilizing a purely (high-order) polynomial ansatz is used as a reference to showcase improvements of the overall quality which can be obtained by the new method.