← Search

Leo Zhang

5 accepted papers

2026

Accelerated Parallel Tempering via Neural Transports

ICLR 2026poster

Markov Chain Monte Carlo (MCMC) algorithms are essential tools in computational statistics for sampling from unnormalised probability distributions, but can be fragile when targeting high-dimensional, multimodal, or complex target distributions. Parallel Tempering (PT) enhances MCMC's sample efficie…

Cited by 0SourceScholar
2026

CREPE: Controlling diffusion with REPlica Exchange

ICLR 2026poster

Inference-time control of diffusion models aims to steer model outputs to satisfy new constraints without retraining. Previous approaches have mostly relied on heuristic guidance or have been coupled with Sequential Monte Carlo (SMC) for bias correction. In this paper, we propose a flexible alternat…

Cited by 0SourcecodeScholar
2026

SigmaDock: Untwisting Molecular Docking with Fragment-Based SE(3) Diffusion

ICLR 2026poster

Determining the binding pose of a ligand to a protein, known as molecular docking, is a fundamental task in drug discovery. Generative approaches promise faster, improved, and more diverse pose sampling than physics-based methods, but are often hindered by chemically implausible outputs, poor genera…

Cited by 0SourcecodeScholar
2025

SymDiff: Equivariant Diffusion via Stochastic Symmetrisation

ICLR 2025poster

We propose SymDiff, a method for constructing equivariant diffusion models using the framework of stochastic symmetrisation. SymDiff resembles a learned data augmentation that is deployed at sampling time, and is lightweight, computationally efficient, and easy to implement on top of arbitrary off-t…

Cited by 0SourcePDFScholar
2024

Metric Flow Matching for Smooth Interpolations on the Data Manifold

NeurIPS 2024poster

Matching objectives underpin the success of modern generative models and rely on constructing conditional paths that transform a source distribution into a target distribution. Despite being a fundamental building block, conditional paths have been designed principally under the assumption of $\text…