CoeffNet : predicting activation barriers through a chemically-interpretable, equivariant and physically constrained graph neural network

Sudarshan Vijay,Maxwell C. Venetos,Evan Walter Clark Spotte-Smith,Aaron D. Kaplan,Mingjian Wen,Kristin A. Persson
DOI: https://doi.org/10.1039/d3sc04411d
IF: 8.4
2024-01-23
Chemical Science
Abstract:Activation barriers of elementary reactions are essential to predict molecular reaction mechanisms and kinetics. However, computing these energy barriers by identifying transition states with electronic structure methods ( e.g. , density functional theory) can be time-consuming and computationally expensive. In this work, we introduce CoeffNet , an equivariant graph neural network that predicts activation barriers using coefficients of any frontier molecular orbital (such as the highest occupied molecular orbital) of reactant and product complexes as graph node features. We show that using coefficients as features offer several advantages, such as chemical interpretability and physical constraints on the network's behaviour and numerical range. Model outputs are either activation barriers or coefficients of the chosen molecular orbital of the transition state; the latter quantity allows us to interpret the results of the neural network through chemical intuition. We test CoeffNet on a dataset of S N 2 reactions as a proof-of-concept and show that the activation barriers are predicted with a mean absolute error of less than 0.025 eV. The highest occupied molecular orbital of the transition state is visualized and the distribution of the orbital densities of the transition states is described for a few prototype S N 2 reactions.
chemistry, multidisciplinary
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the challenges faced when calculating the activation barriers of reactions. Specifically, it is crucial to predict molecular reaction mechanisms and kinetics by identifying transition states, but calculating these energy barriers using electronic structure methods (such as density functional theory, DFT) is usually time - consuming and computationally expensive. To solve this problem, the author introduced a new model - **CoeffNet**, which is an equivariant graph neural network. It uses the coefficients of any frontier molecular orbitals (such as the highest occupied molecular orbital, HOMO) of reactant and product complexes as graph node features to predict the activation barriers. This method not only improves the prediction accuracy but also provides chemical interpretability and imposes physical constraints to ensure the rationality of the model behavior and numerical range. #### Main objectives: 1. **Improve computational efficiency**: Compared with traditional DFT calculations, CoeffNet can predict activation barriers more quickly. 2. **Provide chemical interpretability**: By using molecular orbital coefficients as features, the model's output can be interpreted based on chemical intuition. 3. **Impose physical constraints**: Ensure that the model learns within a fixed numerical range, thereby improving the reliability of prediction. #### Specific applications: - **SN2 reaction**: As a proof of concept, the author tested the performance of CoeffNet on a set of SN2 reactions, and the results showed that the mean absolute error was less than 0.025 eV. - **Visualization and analysis**: The model can also predict and visualize the highest occupied molecular orbital (HOMO) of the transition state and describe its orbital density distribution. Through these improvements, CoeffNet can not only accurately predict activation barriers but also provide valuable chemical explanations for chemists, thereby guiding further research and experimental design.