OrbGNN: A Wave function-based Machine Learning Interelectronic Representation

Abstract

Machine learning interatomic potentials (MLIPs) have become emerging tools in molecular modeling and computational chemistry. By learning high-dimensional potential energy surfaces from quantum chemical data, MLIPs enable accurate and efficient predictions of structural, thermodynamic, and dynamical properties. However, such models have limitations in predictions of electronic properties and the effects of static electron correlation due to their lack of electronic structure information. This work presents OrbGNN, an electronic structure graph architecture analogous to molecular graph and MLIP frameworks, where pair-orbital interactions constitute the graph representation, while orbital entanglement encodes the connectivity between them. By embedding information derived from orbital correlation metrics directly into the graph topology, OrbGNN provides a compact representation of a molecule s orbital landscape and electron correlation patterns. Analysis of the behavior of the feature space in an orbital graph are shown to demonstrate model robustness. The model is evaluated for the dissociation of nitrogen and for a larger dataset of diatomic molecules. Finally, the OrbGNN model is applied to a set of octahedral iron(II) complexes to predict spin-state energy gaps.