Learn from Your Mistakes: Self-Correcting Masked Diffusion Models
从错误中学习:自纠正掩码扩散模型
机构 * Cornell(康奈尔大学) ; NVIDIA(英伟达)
AI总结 提出ProSeCo框架,通过训练模型同时进行掩码去除和错误纠正,在生成过程中迭代修正已解码标记,提升样本质量并实现更快的采样速度。
Comments Code to reproduce our experiments is available here: https://github.com/kuleshov-group/proseco