Normalizing Flow MD Accelerator for Boltzmann Sampling

Resolve Bottlenecks,
Find Innovative Solutions
Generate Solutions

Solution Overview

Problem

Molecular dynamics simulations face challenges in sampling from the Boltzmann distribution due to the presence of metastable states, which require long simulation times, making it infeasible to explore conformational changes on millisecond timescales, and existing enhanced sampling methods increase computational complexity and data requirements.

Innovation Solution

A deep learning-based MD accelerator using a normalizing flow model acts as a proposal distribution for a Markov-chain Monte Carlo method, targeting the Boltzmann distribution asymptotically unbiasedly, allowing for efficient sampling of molecular trajectories by making large steps in time and generalizing to molecules outside the training set.

Engineering Contradictions & Design Principles

VSEngineering Contradiction Analysis

1Measurement precision

If traditional molecular dynamics simulations are used to sample from the Boltzmann distribution, then accurate free-energy estimates can be obtained, but the sampling time becomes excessively long due to metastable states

Engineering Contradiction:
Improveaccuracy of free-energy estimatesVSAvoidsampling time
Core Design Contradiction:
Measurement precisionVSLoss of time

Solution Approach 1:

The patent introduces a machine learning model as an intermediary between traditional MD simulations and Boltzmann sampling. The model is trained on MD simulation data to learn the underlying energy landscape, then used to propose conformational moves that are subsequently accepted or rejected based on the Metropolis criterion. This intermediary enables efficient sampling while maintaining accuracy.

Inventive Principle:
Principle #24Intermediary (Mediator)

Solution Approach 2:

The patent performs preliminary training of the machine learning model using traditional MD simulation data before actual Boltzmann sampling. This preliminary action allows the model to learn the energy landscape characteristics and propose informed conformational moves, avoiding the need for long simulation times during the actual sampling process.

Inventive Principle:
Principle #10Preliminary action

2Productivity

If existing enhanced sampling methods are used to explore conformational changes, then sampling efficiency is improved, but computational complexity and data requirements increase

Engineering Contradiction:
Improvesampling efficiencyVSAvoidcomputational complexity
Core Design Contradiction:
ProductivityVSDevice complexity

Solution Approach 1:

The patent replaces the mechanical simulation approach of traditional MD with a machine learning-based proposal mechanism. Instead of relying on force field calculations and small time-step integration, the ML model directly proposes conformational moves based on learned patterns from training data, significantly reducing computational complexity per sampling step.

Inventive Principle:
Principle #28Mechanics substitution (Replace mechanical system)

Solution Approach 2:

The patent changes the parameter space by using a machine learning model that operates in a higher-dimensional feature space learned from training data. The model transforms the sampling problem from direct coordinate space manipulation to a learned representation space, enabling more efficient exploration of conformational space.

Inventive Principle:
Principle #35Parameter changes

3Measurement precision

If traditional MD simulations are used to make small time steps, then numerical accuracy is maintained, but the ability to reach millisecond timescales is lost

Engineering Contradiction:
Improvenumerical accuracyVSAvoidsimulation timescale
Core Design Contradiction:
Measurement precisionVSDuration of action of moving object

Solution Approach 1:

The patent moves the simulation from direct physical time integration to a learned timescale dimension. The ML model is trained to predict conformational changes over arbitrary time intervals, effectively adding a timescale dimension that allows jumping from femtosecond MD steps to millisecond-scale conformational changes without losing accuracy.

Inventive Principle:
Principle #17Another dimension (Dimensionality change)

Solution Approach 2:

The patent creates a learned copy of the energy landscape and dynamics from training data. This copy allows the system to simulate conformational changes at accelerated timescales while maintaining fidelity to the underlying physics, as the ML model reproduces the essential dynamics without requiring actual physical time evolution.

Inventive Principle:
Principle #26Copying

Data Source

PatentUS20240249800A1Trained machine learning model for forecasting molecular conformations
Publication Date: 2024.07.25 MICROSOFT TECHNOLOGY LICENSING LLC
  • US20240249800A1 patent drawing
  • US20240249800A1 patent drawing
  • US20240249800A1 patent drawing

AI summary

A computerized method for forecasting a future conformation of a molecular system based on a current conformation of the molecular system comprises (a) receiving the current conformation in a trained machine-learning model that has been previously trained to map a plurality of conformations received to a corresponding plurality of conformations proposed; (b) mapping the current conformation to a proposed conformation via the trained machine-learning model, wherein the proposed conformation is appended to a Markov chain; and (c) returning the proposed conformation as the future conformation.