GraphVAMPNet, using graph neural networks and variational approach to markov processes for dynamical modeling of biomolecules

Abstract

Finding low dimensional representation of data from long-timescale trajectories of biomolecular processes such as protein-folding or ligand-receptor binding is of fundamental importance and kinetic models such as Markov modeling have proven useful in describing the kinetics of these systems. Recently, an unsupervised machine learning technique called VAMPNet was introduced to learn the low dimensional representation and linear dynamical model in an end-to-end manner. VAMPNet is based on variational approach to Markov processes (VAMP) and relies on neural networks to learn the coarse-grained dynamics. In this contribution, we combine VAMPNet and graph neural networks to generate an end-to-end framework to efficiently learn high-level dynamics and metastable states from the long-timescale molecular dynamics trajectories. This method bears the advantages of graph representation learning and uses graph message passing operations to generate an embedding for each datapoint which is used in the VAMPNet to generate a coarse-grained representation. This type of molecular representation results in a higher resolution and more interpretable Markov model than the standard VAMPNet enabling a more detailed kinetic study of the biomolecular processes. Our GraphVAMPNet approach is also enhanced with an attention mechanism to find the important residues for classification into different metastable states.

References in corpus (6)

• Semi-Supervised Classification with Graph Convolutional Networks
• SchNet: A continuous-filter convolutional neural network for modeling quantum interactions
• Variational cross-validation of slow dynamical modes in molecular kinetics
• Nonlinear Discovery of Slow Molecular Modes using State-Free Reversible VAMPnets
• Graph Dynamical Networks for Unsupervised Learning of Atomic Scale Dynamics in Materials
• Deep learning Markov and Koopman models with physical constraints

Cited by in corpus (4)

• Advances of Machine Learning in Materials Science: Ideas and Techniques
• Driving and characterizing nucleation of urea and glycine polymorphs in water
• Descriptors-free Collective Variables From Geometric Graph Neural Networks
• Optimization of Non-Equilibrium Self-Assembly Protocols Using Markov State Models