Quantum chemistry-augmented neural networks for reactivity prediction: Performance, generalizability, and explainability
Name
5.0079574.pdf
Description
Published version
Size
6.18 MB
Format
Adobe PDF
Checksum (MD5)
1cc5b8e08c267a1064014f916043d3f8
Author(s)
Stuyver, Thijs
Coley, Connor W
Date Issued
2022
Journal
The Journal of Chemical Physics
Publisher
AIP Publishing
Citation
Stuyver, Thijs and Coley, Connor W. 2022. "Quantum chemistry-augmented neural networks for reactivity prediction: Performance, generalizability, and explainability." The Journal of Chemical Physics, 156 (8).
Version
Final published version
Abstract
There is a perceived dichotomy between structure-based and descriptor-based molecular representations used for predictive chemistry tasks. Here, we study the performance, generalizability, and explainability of the quantum mechanics-augmented graph neural network (ml-QM-GNN) architecture as applied to the prediction of regioselectivity (classification) and of activation energies (regression). In our hybrid QM-augmented model architecture, structure-based representations are first used to predict a set of atom- and bond-level reactivity descriptors derived from density functional theory calculations. These estimated reactivity descriptors are combined with the original structure-based representation to make the final reactivity prediction. We demonstrate that our model architecture leads to significant improvements over structure-based GNNs in not only overall accuracy but also in generalization to unseen compounds. Even when provided training sets of only a couple hundred labeled data points, the ml-QM-GNN outperforms other state-of-the-art structure-based architectures that have been applied to these tasks as well as descriptor-based (linear) regressions. As a primary contribution of this work, we demonstrate a bridge between data-driven predictions and conceptual frameworks commonly used to gain qualitative insights into reactivity phenomena, taking advantage of the fact that our models are grounded in (but not restricted to) QM descriptors. This effort results in a productive synergy between theory and data science, wherein QM-augmented models provide a data-driven confirmation of previous qualitative analyses, and these analyses in turn facilitate insights into the decision-making process occurring within ml-QM-GNNs.
MIT Department
Massachusetts Institute of Technology. Department of Mechanical Engineering
Terms of Use
Creative Commons Attribution 4.0 International license
Persistent DSpace Link
DOI of Published Version
https://doi.org/10.1063/5.0079574
