English

Discrete Diffusion with Sample-Efficient Estimators for Conditionals

Machine Learning 2026-03-02 v2 Machine Learning

Abstract

We study a discrete denoising diffusion framework that integrates a sample-efficient estimator of single-site conditionals with round-robin noising and denoising dynamics for generative modeling over discrete state spaces. Rather than approximating a discrete analog of a score function, our formulation treats single-site conditional probabilities as the fundamental objects that parameterize the reverse diffusion process. We employ a sample-efficient method known as Neural Interaction Screening Estimator (NeurISE) to estimate these conditionals in the diffusion dynamics. Controlled experiments on synthetic Ising models, MNIST, and scientific data sets produced by a D-Wave quantum annealer, synthetic Potts model and one-dimensional quantum systems demonstrate the proposed approach. On the binary data sets, these experiments demonstrate that the proposed approach outperforms popular existing methods including ratio-based approaches, achieving improved performance in total variation, cross-correlations, and kernel density estimation metrics.

Keywords

Cite

@article{arxiv.2602.20293,
  title  = {Discrete Diffusion with Sample-Efficient Estimators for Conditionals},
  author = {Karthik Elamvazhuthi and Abhijith Jayakumar and Andrey Y. Lokhov},
  journal= {arXiv preprint arXiv:2602.20293},
  year   = {2026}
}
R2 v1 2026-07-01T10:48:42.869Z