跳到正文
HF Daily Papers·· 3 天前精选AI 评分54

Soft-MTP 通过 KL 梯度恢复改进多 Token 预测

Multi-Token Prediction Done Right

AI 导读

Soft-MTP 通过 KL 梯度恢复移除多 Token 预测中的噪声残差,在不增加参数且无可测开销下改善困惑度、推测解码与推理表现。该方法支持 DeepSeek 和 Meta 风格的 MTP;在 167M 至 1.2B 参数规模上,相比标准 MTP 获得更低的 NTP 困惑度、更快的推测解码和更高的 zero-shot 准确率,1.2B 模型微调后的数学与代码推理也更强。

推荐理由

论文将标准 MTP 梯度拆为 KL 自蒸馏信号与噪声残差,并给出兼容现有训练内核的一行代码改法,为多 Token 训练提供低开销路径。

来源:HF Daily Papers · huggingface.co