High-throughput Quantum Theory of Atoms in Molecules (QTAIM) for Geometric Deep Learning of Molecular and Reaction Properties

Abstract

We present a package, Generator, for geometric molecular property prediction based on topological features of quantum mechanical electron density. Generator computes Quantum Theory of Atoms in Molecules (QTAIM) features, at density functional theory (DFT) level, for sets of molecules or reac- tions in a high-throughput manner, and compiles features into a single data structure for processing, analysis, and geometric machine learning. An accompanying graph neural network package can be used for property prediction and allows users to readily use computed features for learning tasks. To test the efficacy of electron density-based data for machine learning, we benchmark several datasets including QM8, QM9, LIBE, Tox21, and a Green 2022 Reaction dataset. This wide dataset diversity underscores the flexibility of QTAIM descriptors and our package. In addition, we made our code high-throughput methods compatible with new versions of BondNet and Chemprop architectures to allow for both reaction and molecular property prediction out-of-the-box. To motivate the use of QTAIM features for varied prediction tasks we also perform extensive benchmarking of our new mod- els to existing benchmark models as well as to our own models without QTAIM features. We show that almost universally, QTAIM features improve model performance on our algorithms, ChemProp, and BondNet. We also determine that QTAIM can aid in generalizing model performance to out-of- domain (OOD) datasets and improve learning at smaller data regimes. Combined, we hope that this framework could enable QTAIM-enhanced structure-to-property predictions - especially in domains with less data, including experimental or reaction-level datasets with complex underlying chemistries