Annealed sampling of diffusion models in lattice field theory
Abstract
Machine learning methods have recently gained increasing interest as possible means of solving critical slowing down in lattice field theory, with particular emphasis on sampling methods using field transformations such as normalizing flows, annealing/out-of-equilibrium methods, and diffusion models. Annealing methods provide appealing polynomial scaling and exactness guarantees, while diffusion models use a simple and efficient score-matching training procedure. This work explores a method combining these benefits, first training an energy-based diffusion model using score matching, then sampling using either stochastic normalizing flows (SNFs) or flow-transformed replica exchange (T-REX). The choice of sampling method can be made after training the diffusion model; the theoretical properties of SNFs and T-REX and their relative trade-offs are discussed. Numerical results are presented for scalar field theory and the O(3) non-linear sigma model, using a class of local-footprint energy models with efficient gradient and divergence calculations. Sampling from the trained models follows predicted exactness and scaling properties, and significantly improves over sampling methods without a learned component.
Related
- Operator-Guided Model Reduction for Generative Sampling in Lattice Field Theory
- Generalizable Equivariant Diffusion Models for Non-Abelian Lattice Gauge Theory: Analytical Foundations and Numerical Scaling Tests
- Data Free Training of Diffusion Models for Lattice Field Theories
- Elastic deuteron-deuteron scattering in the 5S2 channel within nuclear lattice effective field theory
- Sampling SU(N) gauge theory on a 2D lattice from independent plaquettes via holonomies and corner reweighting
BibTeX
@misc{kanwar2026annealed,
title = {{Annealed sampling of diffusion models in lattice field theory}},
author = {Gurtej Kanwar and Octavio Vega},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.35792},
eprint = {2610.35792},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.35792},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Annealed sampling of diffusion models in lattice field theory
AU - Gurtej Kanwar
AU - Octavio Vega
PY - 2026
DA - 2026/10/26
PB - arXiv
DO - 10.48550/arxiv.2610.35792
UR - https://arxiv.org/abs/2610.35792
AB - Machine learning methods have recently gained increasing interest as possible means of solving critical slowing down in lattice field theory, with particular emphasis on sampling methods using field transformations such as normalizing flows, annealing/out-of-equilibrium methods, and diffusion models. Annealing methods provide appealing polynomial scaling and exactness guarantees, while diffusion models use a simple and efficient score-matching training procedure. This work explores a method combining these benefits, first training an energy-based diffusion model using score matching, then sampling using either stochastic normalizing flows (SNFs) or flow-transformed replica exchange (T-REX). The choice of sampling method can be made after training the diffusion model; the theoretical properties of SNFs and T-REX and their relative trade-offs are discussed. Numerical results are presented for scalar field theory and the O(3) non-linear sigma model, using a class of local-footprint energy models with efficient gradient and divergence calculations. Sampling from the trained models follows predicted exactness and scaling properties, and significantly improves over sampling methods without a learned component.
ER -