Constructing Diabatic Potential Energy Matrices with Neural Networks Based on Adiabatic Energies and Physical Considerations: Toward Quantum Dynamic Accuracy

A permutation invariant polynomial-neural network (PIP-NN) approach for constructing the global diabatic potential energy matrices (PEMs) of the coupled states of molecules is proposed. Specifically, the diabatization scheme is based merely on the adiabatic energy data of the system, which is ideally a most convenient way due to not requiring additional ab initio calculations for the data of the derivative coupling or any other physical properties of the molecule. Considering the permutation and coupling characteristics of the system, particularly in the presence of conical intersections, some vital treatments for the off-diagonal terms in diabatic PEM are essentially needed. Taking the photodissociation of H2O((X) over bar/(B) over bar)/NH3((X) over bar/(A) over bar) and nonadiabatic reaction Na(3p) + H-2 -> NaH(Sigma(+)) + H for example, this PIP-NN method is shown to build up the global diabatic PEMs effectively and accurately. The root-mean-square errors of the adiabatic potential energies in the fitting for three different systems are all small (<10 meV). Further quantum dynamic calculations show that the absorption spectra and product branching ratios in both H2O(<(X)over bar>/(B) over bar) and NH3((X) over bar/(A) over bar) nonadiabatic photodissociation are well reproduced on the new diabatic PEMs, and the nonadiabatic reaction probability of Na(3p) + H-2 -> NaH(Sigma(+)) + H obtained on the new diabatic PEMs of the 1(2)A(1) and 1(2)B(2) states is in reasonably good agreement with previous theoretical result as well, validating this new PIP-NN method.