MIT 单作者论文:Soft-MTP 修正多 token 预测的噪声梯度,零成本提升推理与投机解码

Massachusetts-Institute-of-Technology · hf · 2026-10-30

MIT 的 Jayden Teoh 发布单作者预训练论文「Multi-Token Prediction Done Right」。指出多 token 预测(MTP)中存在一条噪声梯度残留:NTP 损失的负梯度会经共享 LM head 泄入 MTP 头,干扰训练。

Soft-MTP 用一个直通(straight-through)KL 恢复技巧移除该噪声梯度——h0[:, k:] += hk.detach() - hk,数值上 h0 不变、NTP 损失与梯度保持不变,同时回收 KL 梯度。改进在困惑度、投机解码与推理能力上均有提升,且不增加额外成本。

原文链接 →

「模型」频道最新

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