Machine learning of double-valued nonadiabatic coupling vectors around conical intersections

A simple approach to rotationally invariant machine learning of a vector quantity

Unlike with the energy, which is a scalar property, machine learning (ML) prediction of vector or tensor properties poses the additional challenge of achieving proper invariance (covariance) with respect to molecular rotation. For the energy gradients needed in molecular dynamics (MD), this symmetry is automatically fulfilled when taking analytic derivative of the energy, which is a scalar invariant (using properly invariant molecular descriptors). However, if the properties cannot be obtained by differentiation, other appropriate methods should be applied to retain the covariance. Several approaches have been suggested to properly treat this issue. For nonadiabatic couplings and polarizabilities, for example, it was possible to construct virtual quantities from which the above tensorial properties are obtained by differentiation and thus guarantee the covariance. Another possible solution is to build the rotational equivariance into the design of a neural network employed in the model. Here, we propose a simpler alternative technique, which does not require construction of auxiliary properties or application of special equivariant ML techniques. We suggest a three-step approach, using the molecular tensor of inertia. In the first step, the molecule is rotated using the eigenvectors of this tensor to its principal axes. In the second step, the ML procedure predicts the vector property relative to this orientation, based on a training set where all vector properties were in this same coordinate system. As the third step, it remains to transform the ML estimate of the vector property back to the original orientation. This rotate-predict-rotate (RPR) procedure should thus guarantee proper covariance of a vector property and is trivially extensible also to tensors such as polarizability. The RPR procedure has an advantage that the accurate models can be trained very fast for thousands of molecular configurations, which might be beneficial where many training sets are required (e.g., in active learning). We have implemented the RPR technique, using the MLatom and Newton-X programs for ML and MD, and performed its assessment on the dipole moment along MD trajectories of 1,2-dichloroethane.

Making Peace with Random Phases: Ab Initio Conical Intersection Quantum Dynamics in Random Gauges

Ab initio modeling of conical intersection wave packet dynamics is crucial for various photochemical, photophysical, and biological processes. However, adiabatic electronic states obtained from electronic structure computations involve random phases, or more generally, random gauge fixings, which cannot be directly used in the modeling of nonadiabatic wave packet simulations. Here we develop a random-gauge local diabatic representation that allows an exact modeling of conical intersection dynamics directly using the adiabatic electronic states with phases randomly assigned during the electronic structure computations. Its utility is demonstrated by an exact ab initio modeling of the two-dimensional Shin-Metiu model with and without an external magnetic field. Our results provide a simple approach to integrating the electronic structure computations into nonadiabatic quantum dynamics, thus paving the way for ab initio modeling of conical intersection dynamics.

dxtb-An efficient and fully differentiable framework for extended tight-binding

Automatic differentiation (AD) emerged as an integral part of machine learning, accelerating model development by enabling gradient-based optimization without explicit analytical derivatives. Recently, the benefits of AD and computing arbitrary-order derivatives with respect to any variable were also recognized in the field of quantum chemistry. In this work, we present dxtb-an open-source, fully differentiable framework for semiempirical extended tight-binding (xTB) methods. Developed entirely in Python and leveraging PyTorch for array operations, dxtb facilitates extensibility and rapid prototyping while maintaining computational efficiency. Through comprehensive code vectorization and optimization, we essentially reach the speed of compiled xTB programs for high-throughput calculations of small molecules. The excellent performance also scales to large systems, and batch operability yields additional benefits for execution on parallel hardware. In particular, energy evaluations are on par with existing programs, whereas the speed of automatically differentiated nuclear derivatives is only 2 to 5 times slower compared to their analytical counterparts. We showcase the utility of AD in dxtb by calculating various molecular and spectroscopic properties, highlighting its capacity to enhance and simplify such evaluations. Furthermore, the framework streamlines optimization tasks and offers seamless integration of semiempirical quantum chemistry in machine learning, paving the way for physics-inspired end-to-end differentiable models. Ultimately, dxtb aims to further advance the capabilities of semiempirical methods, providing an extensible foundation for future developments and hybrid machine learning applications. The framework is accessible at https://github.com/grimme-lab/dxtb.

Construction of Highly Accurate Machine Learning Potential Energy Surfaces for Excited-State Dynamics Simulations Based on Low-Level Data Sets

Machine learning is capable of effectively predicting the potential energies of molecules in the presence of high-quality data sets. Its application in the construction of ground- and excited-state potential energy surfaces is attractive to accelerate nonadiabatic molecular dynamics simulations of photochemical reactions. Because of the huge computational cost of excited-state electronic structure calculations, the construction of a high-quality data set becomes a bottleneck. In the present work, we first built two data sets. One was obtained from surface hopping dynamics simulations at the semiempirical OM2/MRCI level. Another was extracted from the dynamics trajectories at the CASSCF level, which was reported previously. The ground- and excited-state potential energy surfaces of ethylene-bridged azobenzene at the CASSCF computational level were constructed based on the former low-level data set. Although non-neural network machine learning methods can achieve good or modest performance during the training process, only neural network models provide reliable predictions on the latter external test data set. The BPNN and SchNet combined with the Δ-ML scheme and the force term in the loss functions are recommended for dynamics simulations. Then, we performed excited-state dynamics simulations of the photoisomerization of ethylene-bridged azobenzene on machine learning potential energy surfaces. Compared with the lifetimes of the first excited state (S1) estimated at different computational levels, our results on the E isomer are in good agreement with the high-level estimation. However, the overestimation of the Z isomer is unimproved. It suggests that smaller errors during the training process do not necessarily translate to more accurate predictions on high-level potential energies or better performance on nonadiabatic dynamics simulations, at least in the present case.

MLatom Software Ecosystem for Surface Hopping Dynamics in Python with Quantum Mechanical and Machine Learning Methods

We present an open-source MLatom@XACS software ecosystem for on-the-fly surface hopping nonadiabatic dynamics based on the Landau-Zener-Belyaev-Lebedev algorithm. The dynamics can be performed via Python API with a wide range of quantum mechanical (QM) and machine learning (ML) methods, including ab initio QM (CASSCF and ADC(2)), semiempirical QM methods (e.g., AM1, PM3, OMx, and ODMx), and many types of ML potentials (e.g., KREG, ANI, and MACE). Combinations of QM and ML methods can also be used. While the user can build their own combinations, we provide AIQM1, which is based on Δ-learning and can be used out-of-the-box. We showcase how AIQM1 reproduces the isomerization quantum yield of trans-azobenzene at a low cost. We provide example scripts that, in dozens of lines, enable the user to obtain the final population plots by simply providing the initial geometry of a molecule. Thus, those scripts perform geometry optimization, normal mode calculations, initial condition sampling, parallel trajectories propagation, population analysis, and final result plotting. Given the capabilities of MLatom to be used for training different ML models, this ecosystem can be seamlessly integrated into the protocols building ML models for nonadiabatic dynamics. In the future, a deeper and more efficient integration of MLatom with Newton-X will enable a vast range of functionalities for surface hopping dynamics, such as fewest-switches surface hopping, to facilitate similar workflows via the Python API.

A Local Diabatisation Method for Two-State Adiabatic Conical Intersections

A methodology to locally characterize conical intersections (CIs) between two adiabatic electronic states for which no nonadiabatic coupling (NAC) vectors are available is presented. Based on the Hessian and gradient at the CI, the branching space coordinates are identified. The potential energy surface around the CI in the branching space is expressed in the diabatic representation, from which the NAC vectors can be calculated in a wave-function-free, energy-based approach. To demonstrate the universality of the developed methodology, the minimum-energy CI (MECI) between the first (S1) and second (S2) singlet excited states of formamide is investigated at the state-averaged complete active space self-consistent field (SA-CASSCF) and extended multistate complete active space second-order perturbation theory (XMS-CASPT2) levels of theory. In addition, the asymmetrical MECI between the ground state (S0) and S1 of cyclopropanone is evaluated using SA-CASSCF, as well as (ME)CIs between the S1 and S2 states of benzene using SA-CASSCF and time-dependent density functional theory (TDDFT). Finally, a CI between the S1 and S2 excited states of thiophene was analyzed using TDDFT.

Machine Learning Framework for Modeling Exciton Polaritons in Molecular Materials

A light-matter hybrid quasiparticle, called a polariton, is formed when molecules are strongly coupled to an optical cavity. Recent experiments have shown that polariton chemistry can manipulate chemical reactions. Polariton chemistry is a collective phenomenon, and its effects increase with the number of molecules in a cavity. However, simulating an ensemble of molecules in the excited state coupled to a cavity mode is theoretically and computationally challenging. Recent advances in machine learning (ML) techniques have shown promising capabilities in modeling ground-state chemical systems. This work presents a general protocol to predict excited-state properties, such as energies, transition dipoles, and nonadiabatic coupling vectors with the hierarchically interacting particle neural network. ML predictions are then applied to compute the potential energy surfaces and electronic spectra of a prototype azomethane molecule in the collective coupling scenario. These computational tools provide a much-needed framework to model and understand many molecules' emerging excited-state polariton chemistry.

Nonadiabatic Derivative Couplings Calculated Using Information of Potential Energy Surfaces without Wavefunctions: Ab Initio and Machine Learning Implementations

In this work, we implemented an approximate algorithm for calculating nonadiabatic coupling matrix elements (NACMEs) of a polyatomic system with ab initio methods and machine learning (ML) models. Utilizing this algorithm, one can calculate NACMEs using only the information of potential energy surfaces (PESs), i.e., energies, and gradients as well as Hessian matrix elements. We used a realistic system, namely CH₂NH, to compare NACMEs calculated by this approximate PES-based algorithm and the accurate wavefunction-based algorithm. Our results show that this approximate PES-based algorithm can give very accurate results comparable to the wavefunction-based algorithm except at energetically degenerate points, i.e., conical intersections. We also tested a machine learning (ML)-trained model with this approximate PES-based algorithm, which also supplied similarly accurate NACMEs but more efficiently. The advantage of this PES-based algorithm is its significant potential to combine with electronic structure methods that do not implement wavefunction-based algorithms, low-scaling energy-based fragment methods, etc., and in particular efficient ML models, to compute NACMEs. The present work could encourage further research on nonadiabatic processes of large systems simulated by ab initio nonadiabatic dynamics simulation methods in which NACMEs are always required.

A mapping approach to surface hopping

We present a nonadiabatic classical-trajectory approach that offers the best of both worlds between fewest-switches surface hopping (FSSH) and quasiclassical mapping dynamics. This mapping approach to surface hopping (MASH) propagates the nuclei on the active adiabatic potential-energy surface, such as in FSSH. However, unlike in FSSH, transitions between active surfaces are deterministic and occur when the electronic mapping variables evolve between specified regions of the electronic phase space. This guarantees internal consistency between the active surface and the electronic degrees of freedom throughout the dynamics. MASH is rigorously derivable from exact quantum mechanics as a limit of the quantum-classical Liouville equation (QCLE), leading to a unique prescription for momentum rescaling and frustrated hops. Hence, a quantum-jump procedure can, in principle, be used to systematically converge the accuracy of the results to that of the QCLE. This jump procedure also provides a rigorous framework for deriving approximate decoherence corrections similar to those proposed for FSSH. We apply MASH to simulate the nonadiabatic dynamics in various model systems and show that it consistently produces more accurate results than FSSH at a comparable computational cost.