Geometric learning of the conformational dynamics of molecules using dynamic graph neural networks

Abstract

We apply a temporal edge prediction model for weighted dynamic graphs to predict time-dependent changes in molecular structure. Each molecule is represented as a complete graph in which each atom is a vertex and all vertex pairs are connected by an edge weighted by the Euclidean distance between atom pairs. We ingest a sequence of complete molecular graphs into a dynamic graph neural network (GNN) to predict the graph at the next time step. Our dynamic GNN predicts atom-to-atom distances with a mean absolute error of 0.017 Å, which is considered ``chemically accurate'' for molecular simulations. We also explored the transferability of a trained network to new molecular systems and found that finetuning with less than 10% of the total trajectory provides a mean absolute error of the same order of magnitude as that when training from scratch on the full molecular trajectory.

arXiv comment11 pages, 4 figures

References in corpus (13)

• Adam: A Method for Stochastic Optimization
• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Molecular Graph Convolutions: Moving Beyond Fingerprints
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Foundations and modelling of dynamic networks using Dynamic Graph Neural Networks: A survey
• Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
• Graph Convolutional Policy Network for Goal-Directed Molecular Graph Generation
• A Two-Step Graph Convolutional Decoder for Molecule Generation
• DEFactor: Differentiable Edge Factorization-based Probabilistic Graph Generation
• Decoding Molecular Graph Embeddings with Reinforcement Learning
• GraphDF: A Discrete Flow Model for Molecular Graph Generation
• Tiered Graph Autoencoders with PyTorch Geometric for Molecular Graphs