Understanding and Accelerating the Training of Masked Diffusion Language Models
理解并加速掩码扩散语言模型的训练
机构 * KAIST(韩国科学技术院) ; Sony AI(索尼人工智能) ; University of Tokyo(东京大学) ; Sony Group Corporation(索尼集团)
AI总结 本文研究了掩码扩散模型训练缓慢的原因,提出bell-shaped时间采样策略,显著提升训练效率和模型性能。
Comments Preprint