arXiv Machine Learning

ENCORE: Exact Non-equilibrium COntrol with Replica Exchange for Diffusion Generation

arXiv Machine Learning
Sep 17

METALICA: METAdynamics and repLICA exchange for enhanced diffusion sampling

METALICA is a new method that combines Metadynamics with Replica Exchange on pretrained diffusion models to improve sampling of rare conformational states in proteins. It builds a bias potential along a chosen Collective Variable, pushes new samples away from previously visited ones, and reweights them to recover the unbiased distribution. By running one replica per diffusion level and allowing inter‑replica communication, METALICA can generate long Markov chains that uncover rare events, as demonstrated on a bimodal test system and on protein unfolding where it reveals a second free‑energy minimum that other methods miss.

By Alireza Omidi, Jiajun He, J\"org Gsponer, Saifuddin Syed
arXiv Machine Learning
Jun 10

Rare Event Analysis via Stochastic Optimal Control

arXiv:2604. 13213v2 Announce Type: replace-cross Abstract: Rare events such as conformational changes in biomolecules, phase transitions, and chemical reactions are central to the behavior of many physical systems, yet they are extremely difficult to study computationally because unbiased simulations seldom produce them.

By Yuanqi Du, Jiajun He, Dinghuai Zhang, Eric Vanden-Eijnden, Carles Domingo-Enrich
arXiv Statistics ML
4d ago

Error-Corrected Inference-Time Scaling for Imperfect Diffusion Models

The paper introduces the Energy-based Feynman-Kac Corrector (EBFKC), a method that improves inference-time scaling for diffusion models that are imperfect due to data and training limitations. EBFKC derives Feynman-Kac dynamics to exactly track a prescribed path in continuous time, then approximates these dynamics with sequential Monte Carlo and variance‑controlling guidance. Experiments on Gaussian mixtures, particle systems, and molecular simulations demonstrate that EBFKC closely matches target distributions and free‑energy profiles, outperforming standard inference‑time scaling baselines that still exhibit significant sampling errors.

By Zuokai Wen, Louis Grenioux, Weinan E, Jiequn Han
arXiv Machine Learning
3d ago

Specificity-Aware Diffusion Steering via Variance-Reduced Sequential Monte Carlo

The paper introduces a method for specificity‑aware diffusion steering that suppresses undesired samples while preserving desired ones. By formulating the problem as a target‑design task, it derives a time‑dependent target distribution based on overlap between positive and negative reference distributions, and samples from it using a variance‑reduced Sequential Monte Carlo (SMC) sampler. Experiments on synthetic, class‑contrastive, text‑to‑image, and peptide‑MHC tasks demonstrate reduced mode shift, improved sampling stability, and better suppression of undesired regions compared to negative‑guidance baselines.

By Luran Wang, Linrui Ma, Hannes St\"ark, Regina Barzilay