Informed Correctors for Discrete Diffusion Models

This repo contains the code to reproduce results from Informed Correctors for Discrete Diffusion Models (Zhao et al., 2025). This repo is currently still work in progress and will be reorganized soon.

📄 Preprint: https://arxiv.org/abs/2407.21243

Contents

• imagenet/ — based on VDM
• text8/ — based on MD4

Citing this work

@article{zhao2024informed,
  title={Informed correctors for discrete diffusion models},
  author={Zhao, Yixiu and Shi, Jiaxin and Chen, Feng and Druckmann, Shaul and Mackey, Lester and Linderman, Scott},
  journal={arXiv preprint arXiv:2407.21243},
  year={2025}
}