Iterative Exact Discrete Guidance for Energy-Based Sampling
Not on the map. Why?
Abstract (arXiv)
Sampling from unnormalized distributions over large discrete state spaces becomes difficult when a multimodal target is far from a tractable reference. We introduce Iterative Exact Discrete Guidance (IEDG), a population-exact, trajectory-wise guidance framework for unnormalized discrete targets. Rather than learn the full reference-to-target correction in one step, IEDG introduces a global Boltzmann tilt along an annealing trajectory. Each stage learns a stage-local posterior correction for an incremental Boltzmann tilt of the current source, while the resulting corrections are accumulated relative to a fixed analytic posterior. At the population optimum, exact stage posteriors recover the correct reverse dynamics, whose exact simulation reproduces the target distribution. IEDG chooses stage increments by relative effective sample size (rESS), which controls R'enyi-2 displacement and locally adapts the step size to the thermodynamic geometry of the annealing path. Our stagewise total-variation analysis shows that limited overlap amplifies Bregman fitting error by 1/√rESS, while posterior, simulation, and truncation errors enter additively. IEDG improves all distribution-level errors over the neural baselines on ordered, exactly enumerated Ising 4×4, while substantially reducing one-shot errors on Ising/Potts 16×16 across thermodynamic regimes and attaining the best neural-sampler result on several reported local-statistic and phase-coverage metrics. On Max-Cut, its best-of-512 and average-sample ratios exceed all the baselines. Code and artifacts are available at https://github.com/StillFantasy123/iterative-exact-discrete-guidance.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 39 references that are in the map. First posted on arXiv 29 Sep 2026.
Builds on
28 of 39 references.
References · 28
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
- 2024Discrete Flow Matchingcited by 317
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
- 2021GFlowNet Foundationscited by 290
- 2022Score-based Continuous-time Discrete Diffusion Modelscited by 169
- 2024Simple Guidance Mechanisms for Discrete Diffusion Modelscited by 155
- 2024Unlocking Guidance for Discrete State-Space Diffusion and Flow Modelscited by 129
- 2022Flow Annealed Importance Sampling Bootstrapcited by 127
- 2024Iterated Denoising Energy Matching for Sampling from Boltzmann Densitiescited by 116
16 more
- 2018Solving Statistical Mechanics Using Variational Autoregressive Networkscited by 100
- 2021Oops I Took A Gradient: Scalable Sampling for Discrete Distributionscited by 92
- 2021Annealed Flow Transport Monte Carlocited by 90
- 2022Continual Repeated Annealed Flow Transport Monte Carlocited by 66
- 2024A Diffusion Model Framework for Unsupervised Neural Combinatorial Optimizationcited by 59
- 2022A Langevin-like Sampler for Discrete Distributionscited by 44
- 2025LEAPS: A discrete neural sampler via locally equivariant networkscited by 31
- 2025Scalable Discrete Diffusion Samplers: Combinatorial Optimization and Statistical Physicscited by 29
- 2025MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal Controlcited by 21
- 2025Proximal Diffusion Neural Samplercited by 16
- 2025Temperature-Annealed Boltzmann Generatorscited by 15
- 2025Discrete Neural Flow Samplers with Locally Equivariant Transformercited by 13
- 2025Progressive Tempering Sampler with Diffusioncited by 13
- 2025What Exactly Does Guidance Do in Masked Discrete Diffusion Modelscited by 9
- 2025Discrete Guidance Matching: Exact Guidance for Discrete Flow Matchingcited by 6
- 2026Discrete diffusion samplers and bridges: Off-policy algorithms and applications in latent spacescited by 2
Built on by
Not cited yet.