Advancing Molecular Machine Learning Representations with Stereoelectronics-Infused Molecular Graphs

Abstract

Molecular representation is a critical element in our understanding of the physical world and the foundation for modern molecular machine learning. Previous molecular machine learning models have employed strings, fingerprints, global features, and simple molecular graphs that are inherently information-sparse representations. However, as the complexity of prediction tasks increases, the molecular representation needs to encode higher fidelity information. This work introduces a novel approach to infusing quantum-chemical-rich information into molecular graphs via stereoelectronic effects, enhancing expressivity and interpretability. Learning to predict the stereoelectronics-infused representation with a tailored double graph neural network workflow enables its application to any downstream molecular machine learning task without expensive quantum chemical calculations. We show that the explicit addition of stereoelectronic information significantly improves the performance of message-passing 2D machine learning models for molecular property prediction. We show that the learned representations trained on small molecules can accurately extrapolate to much larger molecular structures, yielding chemical insight into orbital interactions for previously intractable systems, such as entire proteins, opening new avenues of molecular design. Finally, we have developed a web application (simg.cheme.cmu.edu) where users can rapidly explore stereoelectronic information for their own molecular systems.

arXiv comment23 pages, 6 figures

References in corpus (16)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• ChemRL-GEM: Geometry Enhanced Molecular Representation Learning for Property Prediction
• Applying machine learning techniques to predict the properties of energetic materials
• Principal Neighbourhood Aggregation for Graph Nets
• Bayesian Neural Networks: An Introduction and Survey
• A mathematical and computational review of Hartree-Fock SCF methods in Quantum Chemistry
• DeepGCNs: Can GCNs Go as Deep as CNNs?
• A Note on Over-Smoothing for Graph Neural Networks
• Uncertainty in Gradient Boosting via Ensembles
• SPA${}^{\mathrm{H}}$M: the Spectrum of Approximated Hamiltonian Matrices representations
• Smooth, exact rotational symmetrization for deep learning on point clouds
• Simple GNN Regularisation for 3D Molecular Property Prediction & Beyond
• Bridging deep learning force fields and electronic structures with a physics-informed approach