Composing diffusion priors with explicit physical context via generative Gibbs sampling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Weizhou, Weare, Jonathan, Dinner, Aaron R.
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910209349255168
author Wang, Weizhou
Weare, Jonathan
Dinner, Aaron R.
author_facet Wang, Weizhou
Weare, Jonathan
Dinner, Aaron R.
contents Pretrained diffusion models provide powerful learned priors, but in scientific sampling the target distribution often depends on physical context that is not fully represented by one generative model. We introduce Generative Gibbs for Physics-Aware Sampling (GG-PA), a training-free framework that formulates the composition of learned partial priors and explicit physical context as inference over a joint target distribution in an augmented state space. We derive a Gibbs sampler for this joint target, show that it is asymptotically exact as the diffusion time approaches zero, and prove that in settings with quadratic interactions it remains exact at finite diffusion times. We further introduce replica exchange over diffusion time to accelerate mixing. Experiments on a double-well system, a $ϕ^4$ lattice model, and atomistic peptide systems show that GG-PA recovers context-induced distribution shifts and emergent collective behavior in interacting systems using partial priors without retraining. These results demonstrate GG-PA as a practical approach for combining pretrained generative priors with explicit physical context.
format Preprint
id arxiv_https___arxiv_org_abs_2605_10642
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Composing diffusion priors with explicit physical context via generative Gibbs sampling
Wang, Weizhou
Weare, Jonathan
Dinner, Aaron R.
Machine Learning
Statistical Mechanics
Pretrained diffusion models provide powerful learned priors, but in scientific sampling the target distribution often depends on physical context that is not fully represented by one generative model. We introduce Generative Gibbs for Physics-Aware Sampling (GG-PA), a training-free framework that formulates the composition of learned partial priors and explicit physical context as inference over a joint target distribution in an augmented state space. We derive a Gibbs sampler for this joint target, show that it is asymptotically exact as the diffusion time approaches zero, and prove that in settings with quadratic interactions it remains exact at finite diffusion times. We further introduce replica exchange over diffusion time to accelerate mixing. Experiments on a double-well system, a $ϕ^4$ lattice model, and atomistic peptide systems show that GG-PA recovers context-induced distribution shifts and emergent collective behavior in interacting systems using partial priors without retraining. These results demonstrate GG-PA as a practical approach for combining pretrained generative priors with explicit physical context.
title Composing diffusion priors with explicit physical context via generative Gibbs sampling
topic Machine Learning
Statistical Mechanics
url https://arxiv.org/abs/2605.10642