Training neural nets to learn reactive potential energy surfaces using interactive quantum chemistry in virtual reality

Abstract

Whilst the primary bottleneck to a number of computational workflows was not so long ago limited by processing power, the rise of machine learning technologies has resulted in a paradigm shift which places increasing value on issues related to data curation - i.e., data size, quality, bias, format, and coverage. Increasingly, data-related issues are equally as important as the algorithmic methods used to process and learn from the data. Here we introduce an open source GPU-accelerated neural network (NN) framework for learning reactive potential energy surfaces (PESs), and investigate the use of real-time interactive ab initio molecular dynamics in virtual reality (iMD-VR) as a new strategy for rapidly sampling geometries along reaction pathways which can be used to train NNs to learn reactive PESs. Focussing on hydrogen abstraction reactions of CN radical with isopentane, we compare the performance of NNs trained using iMD-VR data versus NNs trained using a more traditional method, namely molecular dynamics (MD) constrained to sample a predefined grid of points along hydrogen abstraction reaction coordinates. Both the NN trained using iMD-VR data and the NN trained using the constrained MD data reproduce important qualitative features of the reactive PESs, such as a low and early barrier to abstraction. Quantitatively, learning is sensitive to the training dataset. Our results show that user-sampled structures obtained with the quantum chemical iMD-VR machinery enable better sampling in the vicinity of the minimum energy path (MEP). As a result, the NN trained on the iMD-VR data does very well predicting energies in the vicinity of the MEP, but less well predicting energies for 'off-path' structures. The NN trained on the constrained MD data does better in predicting energies for 'off-path' structures, given that it included a number of such structures in its training set.

References in corpus (17)

• Scikit-learn: Machine Learning in Python
• Canonical sampling through velocity-rescaling
• TensorFlow: A system for large-scale machine learning
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Comparing molecules and solids across structural and alchemical space
• Machine Learning Unifies the Modelling of Materials and Molecules
• VAMPnets: Deep learning of molecular kinetics
• Sampling molecular conformations and dynamics in a multi-user virtual reality framework
• Remote optimization of an ultra-cold atoms experiment by experts and citizen scientists
• Semiempirical Molecular Orbital Models based on the Neglect of Diatomic Differential Overlap Approximation
• A parallel multistate framework for atomistic non-equilibrium reaction dynamics of solutes in strongly interacting organic solvents
• Real-time Quantum Chemistry
• Real-time feedback from iterative electronic structure calculations

Cited by in corpus (20)

• FCHL revisited: faster and more accurate quantum machine learning
• The Exploration of Chemical Reaction Networks
• Quantum Machine Learning for Chemistry and Physics
• Neural Network Potentials for Chemistry: Concepts, Applications and Prospects
• Interactive molecular dynamics in virtual reality from quantum chemistry to drug binding: An open-source multi-person framework
• Systematic Microsolvation Approach with a Cluster-Continuum Scheme and Conformational Sampling
• Universal QM/MM Approaches for General Nanoscale Applications
• Interactive molecular dynamics in virtual reality for accurate flexible protein-ligand docking
• Machine Learning Quantum Reaction Rate Constants
• Lifelong Machine Learning Potentials
• Expansive Quantum Mechanical Exploration of Chemical Reaction Paths
• Simulating solvation and acidity in complex mixtures with first-principles accuracy: the case of CH${}_3$SO${}_3$H and H${}_2$O${}_2$ in phenol
• Self-Parametrizing System-Focused Atomistic Models
• PhysNet Meets CHARMM: A Framework for Routine Machine Learning / Molecular Mechanics Simulations
• Training atomic neural networks using fragment-based data generated in virtual reality
• Mean-Field Density Matrix Decompositions
• Free energy along drug-protein binding pathways interactively sampled in virtual reality
• Learning Potentials of Quantum Systems using Deep Neural Networks
• Transformative Applications of Machine Learning for Chemical Reactions
• Learning to Make Chemical Predictions: the Interplay of Feature Representation, Data, and Machine Learning Algorithms