Supervised Learning and the Finite-Temperature String Method for Computing Committor Functions and Reaction Rates

by   Muhammad R. Hasyim, et al.

A central object in the computational studies of rare events is the committor function. Though costly to compute, the committor function encodes complete mechanistic information of the processes involving rare events, including reaction rates and transition-state ensembles. Under the framework of transition path theory (TPT), recent work [1] proposes an algorithm where a feedback loop couples a neural network that models the committor function with importance sampling, mainly umbrella sampling, which collects data needed for adaptive training. In this work, we show additional modifications are needed to improve the accuracy of the algorithm. The first modification adds elements of supervised learning, which allows the neural network to improve its prediction by fitting to sample-mean estimates of committor values obtained from short molecular dynamics trajectories. The second modification replaces the committor-based umbrella sampling with the finite-temperature string (FTS) method, which enables homogeneous sampling in regions where transition pathways are located. We test our modifications on low-dimensional systems with non-convex potential energy where reference solutions can be found via analytical or the finite element methods, and show how combining supervised learning and the FTS method yields accurate computation of committor functions and reaction rates. We also provide an error analysis for algorithms that use the FTS method, using which reaction rates can be accurately estimated during training with a small number of samples.



There are no comments yet.


page 19

page 33


Multitask machine learning of collective variables for enhanced sampling of rare events

Computing accurate reaction rates is a central challenge in computationa...

A Note on Learning Rare Events in Molecular Dynamics using LSTM and Transformer

Recurrent neural networks for language models like long short-term memor...

Computing Committor Functions for the Study of Rare Events Using Deep Learning

The committor function is a central object of study in understanding tra...

Learning with rare data: Using active importance sampling to optimize objectives dominated by rare events

Deep neural networks, when optimized with sufficient data, provide accur...

Low-cost prediction of molecular and transition state partition functions via machine learning

We have generated an open-source dataset of over 30000 organic chemistry...

A reinforcement learning approach to rare trajectory sampling

Very often when studying non-equilibrium systems one is interested in an...

Learning Continuous Cost-to-Go Functions for Non-holonomic Systems

This paper presents a supervised learning method to generate continuous ...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.