亚马逊提出ALoDLM:按token难度自适应分配循环深度

amazon · hf · 2026-10-06

ALoDLM:自适应循环的扩散语言模型

扩散语言模型(DLM)能并行预测多个 token 实现快速生成,但质量始终落后于同规模自回归(AR)模型。论文将这一差距归因于计算难度错配:部分未知 token 很容易预测,另一些则需要更多计算,而现有 DLM 在每一步去噪中对所有未知位置施加统一计算深度。

ALoDLM 用 token 级自适应隐式循环替代统一计算:每步去噪中迭代精炼隐表示,按 token 难度分配算力——准备就绪的 token 提交为离散上下文,未定的 token 保留隐状态继续循环精炼。训练上将 token 级计算调度建模为隐变量,推导出条件负证据下界(NELBO)联合学习预测与算力分配。

结果:在 1.7B 和 8B 两个规模训练,11 项基准上平均分全面超越所有受评 DLM 及对应 AR 基线,同时保留快速并行解码,在优化推理引擎下取得优异的质量-效率权衡。

原文链接 →

「模型」频道最新

更多「模型」频道 AI 资讯 →