Improving Sampling for Masked Diffusion Models via Information Gain
通过信息增益改进掩码扩散模型的采样
机构 * Department of Computer Science and Technology, Tsinghua University(清华大学计算机科学与技术系) ; Massachusetts Institute of Technology(麻省理工学院) ; Shanghai Jiao Tong University(上海交通大学) ; Beihang University(北航) ; College of AI, Tsinghua University(清华大学人工智能学院)
专题命中 扩散模型 :diffusion(title,abstract);image generation(abstract)
AI总结 提出一种无需训练的Info-Gain采样器,利用掩码扩散模型的双向结构平衡即时不确定性与剩余掩码位置的信息增益,在推理、编码、创意写作和图像生成任务中优于现有采样器。
Comments https://github.com/yks23/Information-Gain-Sampler Accepted by ICML2026 Accepted by ICML2026