Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2022 Apr 12;18(4):2341-2353.
doi: 10.1021/acs.jctc.1c00143. Epub 2022 Mar 11.

Multitask Machine Learning of Collective Variables for Enhanced Sampling of Rare Events

Lixin Sun  1 , Jonathan Vandermause  1 , Simon Batzner  1 , Yu Xie  1 , David Clark  1 , Wei Chen  1 , Boris Kozinsky  1
Affiliations

Multitask Machine Learning of Collective Variables for Enhanced Sampling of Rare Events

Lixin Sun et al. J Chem Theory Comput. .

Abstract

Computing accurate reaction rates is a central challenge in computational chemistry and biology because of the high cost of free energy estimation with unbiased molecular dynamics. In this work, a data-driven machine learning algorithm is devised to learn collective variables with a multitask neural network, where a common upstream part reduces the high dimensionality of atomic configurations to a low dimensional latent space and separate downstream parts map the latent space to predictions of basin class labels and potential energies. The resulting latent space is shown to be an effective low-dimensional representation, capturing the reaction progress and guiding effective umbrella sampling to obtain accurate free energy landscapes. This approach is successfully applied to model systems including a 5D Müller Brown model, a 5D three-well model, the alanine dipeptide in vacuum, and an Au(110) surface reconstruction unit reaction. It enables automated dimensionality reduction for energy controlled reactions in complex systems, offers a unified and data-efficient framework that can be trained with limited data, and outperforms single-task learning approaches, including autoencoders.

PubMed Disclaimer

Similar articles

See all similar articles

Cited by

See all "Cited by" articles