DOE PAGES title logo U.S. Department of Energy
Office of Scientific and Technical Information

Title: From Latent Dynamics to Meaningful Representations

Journal Article · · Journal of Chemical Theory and Computation

While representation learning has been central to the rise of machine learning and artificial intelligence, a key problem remains in making the learnt representations meaningful. For this the typical approach is to regularize the learned representation through prior probability distributions. However such priors are usually unavailable or are ad hoc. To deal with this, recent efforts have shifted towards leveraging the insights from physical principles to guide the learning process. In this spirit, we propose a purely dynamics-constrained representation learning framework. Instead of relying on predefined probabilities, we restrict the latent representation to follow overdamped Langevin dynamics with a learnable transition density — a prior driven by statistical mechanics. We show this is a more natural constraint for representation learning in stochastic dynamical systems, with the crucial ability to uniquely identify the ground truth representation. We validate our framework for different systems including a real-world fluorescent DNA movie dataset. Here, we show that our algorithm can uniquely identify orthogonal, isometric and meaningful latent representations.

Research Organization:
University of Maryland, College Park, MD (United States)
Sponsoring Organization:
USDOE Office of Science (SC), Basic Energy Sciences (BES)
Grant/Contract Number:
SC0021009
OSTI ID:
2352717
Journal Information:
Journal of Chemical Theory and Computation, Journal Name: Journal of Chemical Theory and Computation Journal Issue: 9 Vol. 20; ISSN 1549-9618
Publisher:
American Chemical SocietyCopyright Statement
Country of Publication:
United States
Language:
English

References (25)

Enhanced Sampling with Machine Learning journal February 2024
Past–future information bottleneck for sampling molecular reaction coordinate simultaneously with thermodynamics and kinetics journal August 2019
Brownian Motion, Dynamical Friction, and Stellar Dynamics journal July 1949
Identifying Latent Stochastic Differential Equations journal January 2022
On the distance between A and B in molecular configuration space journal April 2020
Position-Dependent Diffusion from Biased Simulations and Markov State Model Analysis journal March 2021
Geometry-enhanced molecular representation learning for property prediction journal February 2022
Controlled Amplification of DNA Brownian Motion Using Electrokinetic Noise journal November 2020
Non-linear independent component analysis with diffusion maps journal September 2008
Prediction of pharmacological activities from chemical structures with graph convolutional neural networks journal January 2021
An Introduction to Variational Autoencoders journal January 2019
Machine learning approaches for analyzing and enhancing molecular dynamics simulations journal April 2020
State predictive information bottleneck journal April 2021
A Commentary on the Unsupervised Learning of Disentangled Representations journal April 2020
Learning Force Fields from Stochastic Trajectories journal April 2020
A graph-convolutional neural network model for the prediction of chemical reactivity journal January 2019
Normalizing Flows: An Introduction and Review of Current Methods journal January 2020
Optimal estimates of self-diffusion coefficients from molecular dynamics simulations journal July 2020
COATI: Multimodal Contrastive Pretraining for Representing and Traversing Chemical Space journal February 2024
Markov models of molecular kinetics: Generation and validation journal May 2011
Position-dependent diffusion coefficients and free energies from Bayesian analysis of equilibrium and replica molecular dynamics simulations journal January 2005
Physics-constrained deep learning for high-dimensional surrogate modeling and uncertainty quantification without labeled data journal October 2019
Well-Tempered Metadynamics: A Smoothly Converging and Tunable Free-Energy Method journal January 2008
Representation Learning: A Review and New Perspectives journal August 2013
Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations journal February 2019