回到最新动态
研究论文 / THE SIGNAL

扩散语言模型学习型解掩码策略的受控研究

What Makes a Good Unmasking Policy? A Controlled Study of Learned Unmasking in Diffusion Language Models

重要性 5.5 AI 辅助整理

论文用轨迹收缩目标训练解掩码策略,在 LLaDA-8B 上改善准确率—NFE 前沿。

资讯摘要

论文对掩码扩散语言模型的学习型解掩码策略做受控研究,分别考察教师标注哪些预测可提交、策略在何种状态上模仿、策略观察什么输入,以及需降低多少函数评估次数(NFE)才能抵消策略开销。作者发现,以最终输出为标签在低 NFE 下过于宽松,隐藏状态单独使用较弱但可补充置信度;约 10% 的 NFE 降幅即可抵消最多 7.5M 参数策略的开销。据此作者提出轨迹收缩目标,仅接受留在参考轨迹上的提交,并结合置信度与隐藏状态训练策略。

要点速读

  1. 论文将监督式解掩码策略视为通过教师蒸馏解码过程,并分别检验标注、模仿状态、输入与 NFE 开销等设计选择。
  2. 作者报告:以匹配最终输出为标签在低 NFE 下过于宽松;隐藏状态单独较弱,但能与置信度互补;约 10% 的 NFE 降幅可抵消最多 7.5M 参数策略的开销。
  3. 作者用轨迹收缩目标训练 856K 参数策略,在 LLaDA-8B-Instruct 上取得最佳准确率—NFE 前沿与 GSM8K 最高吞吐,在 MATH-500、HumanEval 中等目标下具竞争力,并较大程度迁移到 Dream-7B-Instruct。

中文摘要由 AI 根据公开来源生成,可能存在理解偏差;具体能力、数字、开放时间等信息请以原文为准。
采集于 2026/10/24 08:00:37(北京时间)。

阅读 Daily Papers 原文arxiv.org · 在新窗口打开