English

Debiasing Guidance for Discrete Diffusion with Sequential Monte Carlo

Machine Learning 2025-09-03 v3

Abstract

Discrete diffusion models are a class of generative models that produce samples from an approximated data distribution within a discrete state space. Often, there is a need to target specific regions of the data distribution. Current guidance methods aim to sample from a distribution with mass proportional to p0(x0)p(ζx0)αp_0(x_0) p(\zeta|x_0)^\alpha but fail to achieve this in practice. We introduce a Sequential Monte Carlo algorithm that generates unbiasedly from this target distribution, utilising the learnt unconditional and guided process. We validate our approach on low-dimensional distributions, controlled images and text generations. For text generation, our method provides strong control while maintaining low perplexity compared to guidance-based approaches.

Keywords

Cite

@article{arxiv.2502.06079,
  title  = {Debiasing Guidance for Discrete Diffusion with Sequential Monte Carlo},
  author = {Cheuk Kit Lee and Paul Jeha and Jes Frellsen and Pietro Lio and Michael Samuel Albergo and Francisco Vargas},
  journal= {arXiv preprint arXiv:2502.06079},
  year   = {2025}
}

Comments

29 pages, 14 figures

R2 v1 2026-06-28T21:38:00.230Z