跳到正文
arXiv · cs.LG· Yair Schiff, Omer Belhasin, Roy Uziel, Guanghan Wang, Marianne Arriola, Gilad Turok, Ran Zilberstein, Michael Elad, Volodymyr Kuleshov·· 1 天前

让掩码扩散模型从错误中学习:ProSeCo 自校正框架

Learn from Your Mistakes: Self-Correcting Masked Diffusion Models

AI 导读

研究者提出 Progressive Self-Correction(ProSeCo)训练与采样方法,让掩码扩散模型(MDM)同时执行去掩码与纠错,可迭代修正包括已生成 token 在内的整段序列。该方法在多项条件与无条件任务上实现更优的质量-效率权衡,采样速度最高提升约 4 倍且仍优于最佳准确率基线,并可通过推理时计算扩展在基准上再获约 1.2 倍提升。

来源:arXiv · cs.LG · arxiv.org