Generative Modeling & Sampling

Generative modeling connects sampling, inference, and stochastic control through methods for learning and transforming probability distributions. These connections bring together machine learning, applied mathematics, and statistical physics, with applications in scientific simulation and discovery. Dr. Hasyim develops methods at this intersection to explore molecular distributions, evaluate generative models, and compute reaction pathways and rates. He also participated in the MSRNE Generative Modeling & Sampling Workshop, which brought these communities together around shared mathematical and computational questions.

Probing Co-Folding Models with Path Sampling

Research in progress (October 2026). Figures from Dr. Hasyim’s presentation; manuscript in preparation.

Co-folding models such as Boltz-2 generate structures from distributions learned from structural databases. To reveal what these models learn, Dr. Hasyim developed transition path sampling of their denoising trajectories, combined with adaptive biasing and replica exchange. Reweighting estimates the original model’s probabilities, allowing evaluations of rare conformations, plausible flexibility, and memorization beyond a few successful predictions. The resulting learned probability landscape, \(-\ln p_\theta(s)+C\), describes model output rather than thermodynamic free energy. The figures use \(F/k_{\mathrm B}T\) for this dimensionless quantity, with its minimum shifted to zero; the populations are those of the model.  [1]  [2]  [3]

For MAP2K1 (MEK1), the method reveals a chemically plausible alternative FZC pose, accounting for about 7.6% of the model population, in which a terminal ring rotates by 61 degrees while the rest of the ligand remains nearly fixed. Moreover, EF-Tu sampling reveals partly open and open, GDP-like structures despite the model’s strong preference for a compact conformation, with no nucleotide specified in the input. Across eight Boltz-2 targets, a Pareto comparison exposes the trade-off between exploring rare regions and converging their probability estimates. Together, these results motivate evaluations that examine both the structures a model can generate and the distribution it assigns to them.

MAP2K1 with bound FZC, a two-basin probability map, and ligand structures showing the terminal ring rotated by 61 degrees.

MAP2K1–FZC. The alternative pose changes the terminal ring’s orientation and hydrogen-bond partners. The landscape and percentages describe learned model probabilities.

EF-Tu probability map and three structures showing the compact GTP-like form, a partly open beta-hairpin-containing form, and the open GDP-like form.

EF-Tu. Independent sampling reveals compact, partly open, and open conformations. The partly open structure has a refolded effector region; biased estimates of the rare-state weights have yet to converge.

Three panels comparing observed distribution coverage, error against an independent-sampling reference, and the Pareto trade-off between coverage gain and relative accuracy.

Exploration and convergence across eight Boltz-2 targets. The Pareto frontier compares coverage gains with relative accuracy at a fixed sampling budget. Error is measured against a reference distribution from the same model.

Computing Committor Functions with Neural Networks

The discovery of transition pathways governing kinetic processes at the microscopic level is of fundamental interest in chemistry. Common to these problems is the presence of high-energy barriers, which make the pathway discovery difficult due to the need to sample rare events.

The committor function \(q(\mathbf{x})\) in the context of a 2D state space. Dashed lines are contours of isocommittor values and gray lines, which are perpendicular to isocommittor contours, denote possible reactive pathways. Given \(q(\mathbf{x})\), such pathways along with the rate in which a system arrives to the reactant state can be computed.

The committor function \(q(\mathbf{x})\), the probability that a trajectory starting from some initial configuration enters the product state before the reactant state, encodes the complete mechanistic details of these pathways, including the reaction rates and transition-state ensembles. Computing the committor function by traditional means is prohibitively expensive, requiring hundreds of trajectories per configuration to yield sufficient estimates along with the need for rare configurations along the transition pathway.

Dr. Hasyim’s work  [4] extends the previous approach of Grant Rotskoff and Eric Vanden-Eijnden that combines importance sampling, in which high probability regions of the transition pathway are targeted, and machine learning, in which an artificial neural network approximates the committor function.

An example of combining the finite-temperature string (FTS) method with a neural network. The FTS method collects data adaptively along the transition pathways. Those data are then fed into the neural network, which learns from the data through stochastic gradient descent of an optimization objective. The output of the neural network is a committor value given a configuration \(\mathbf{x}\).

Dr. Hasyim and collaborators improved their method by adding two elements: (1) supervised learning, in which sample-mean estimates of the committor function obtained via short simulation trajectories are used to fit the neural network, and (2) the finite-temperature string method, a path-finding algorithm that enables homogeneous sampling across the transition pathway.

The research team then demonstrated these modifications on low-dimensional systems first, showing that they yield accurate estimates of the committor function and reaction rates. Additionally, an error analysis for their algorithm was developed from which the reaction rates can be accurately estimated via a small number of samples.

This work is done in collaboration with Clay H. Batton and under the supervision of Prof. Kranthi Mandadapu. Code may be found on Dr. Hasyim’s Github. Paper can be found in Ref.  [4]

References

[1]
[2]
P. G. Bolhuis, D. Chandler, C. Dellago, and P. L. Geissler, Transition path sampling: Throwing ropes over rough mountain passes, in the dark, Annual Review of Physical Chemistry 53, 291 (2002).
[3]
M. Invernizzi and M. Parrinello, Rethinking metadynamics: From bias potentials to probability distributions, The Journal of Physical Chemistry Letters 11, 2731 (2020).
[4]
Hasyim, M. R., C. H. Batton, and K. K. Mandadapu, Supervised learning and the finite-temperature string method for computing committor functions and reaction rates, The Journal of Chemical Physics 157, 184111 (2022).