SetCommit:免训练协调并行 token 提交,加速扩散语言模型解码

论文AI 评分 72/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

论文提出 SetCommit,一种免训练方法,用于扩散语言模型少步解码中的协调并行 token 提交。该方法不再按置信度独立挑选 token,而是选出一个提交集合,在单个 token 的可靠性与局部联合提交风险之间取得平衡,从而为后续去噪步骤保留条件化机会;同时进行预算中性的局部替换,当新上下文支持更可靠的选项时,替换掉此前较脆弱的提交。方法无需重训练或修改架构,兼容现有扩散解码策略。在 LLaDA-8B 与 Dream-7B 上,于 GSM8K、MATH-500、MBPP、HumanEval 测试,32 次模型评估下准确率/pass@1 最高提升 7.4 个百分点;在常见质量目标下,几何平均增量加速为 1.44 至 1.74 倍。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

少步解码的提交难题

扩散语言模型能够并行生成多个 token 位置,但在少步解码场景下,并行提交变得困难。材料指出,单独看似乎可靠的 token,放在一起可能彼此不一致;而过早提交又会阻断后续依赖上下文的修正机会。SetCommit 针对的正是这一矛盾:如何在并行提交时既保证速度,又不牺牲后续去噪步骤的调整空间。

协调提交与局部替换

SetCommit 的做法是不再按置信度逐个独立挑选 token,而是选出一个提交集合,使单个 token 的可靠性与局部联合提交风险之间取得平衡,从而为后续去噪保留条件化机会。此外,方法执行预算中性的局部替换:当新揭示的上下文支持更可靠的替代项时,替换掉此前较脆弱的提交。整个过程不需要重训练,也不需要修改模型架构,并且可以与现有扩散解码策略配合使用。

实验设置与结果

实验在 LLaDA-8B 和 Dream-7B 两个模型上进行,评测集包括 GSM8K、MATH-500、MBPP 和 HumanEval。在 32 次模型评估的条件下,准确率/pass@1 相比原生解码最高提升 7.4 个百分点。在常见质量目标下,几何平均增量加速落在 1.44 至 1.74 倍之间。论文代码已公开。

为什么值得看

免训练、不改架构即可提升扩散语言模型少步解码的质量与速度,对推理效率优化有直接参考价值。

论文

信源1 家

  1. [1]arXiv cs.AI一手信源SetCommit: Coordinated Token Commitments for Few-Step Diffusion Language Model Decoding

关键事实

  • SetCommit 是免训练的协调并行 token 提交方法,无需重训练或架构修改[1]

  • 方法在 token 可靠性与局部联合提交风险之间做平衡,并执行预算中性的局部替换[1]

  • 在 LLaDA-8B 与 Dream-7B 上测试 GSM8K、MATH-500、MBPP、HumanEval[1]

  • 32 次模型评估下准确率/pass@1 最高提升 7.4 个百分点[1]

  • 常见质量目标下几何平均增量加速为 1.44 至 1.74 倍[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。