行业新闻

阿里千问提出 TRACE:FP4 Rollout 强化学习训练提速最高 5.4 倍且不崩

TRACE 让训练侧对齐 Rollout 的真实 FP4 量化结果,解决了低精度 Rollout 导致 RL 训练崩溃的问题,同时保住加速收益。

强化学习里最耗算力的环节是 Rollout,即让模型大量生成轨迹采样。阿里千问与俄亥俄州立大学提出 TRACE:Rollout 用 FP4(4 位浮点量化)加速,但训练侧舍入时参考 Rollout 实际量化结果,避免两边数值落到不同码点。四个 MoE 模型上性能回到 BF16 水平,128K 输出解码吞吐最高提升 5.4 倍。

正文摘录

![](https://image.jiqizhixin.com/uploads/article/coverimage/045104c9-3b46-4b93-a24a-c11004b56605/085(2).jpg) ![图片](https://mmbiz.qpic.cn/mmbizjpg/5L8bhP5dIqFg0WfYlyqqeepLD4C3TibGpjf6FOH7EVuKjibcnDe74qjAU9XhgHylI3WghS7ricbx1ckkNhJHaQ2AnGqG5ZmOdwHJQLl99bxYHU/640?wxfmt=webp&from=appmsg&tp=webp&wxfrom=5&wxlazy=1imgIndex=0) 大模型强化学习越来越「吃算力」,最昂贵的环节之一正是 Rollout。把 Rollout 从 BF16 压到 FP4,理论上可以显著降低计算和显存开销,但问题也随之而来 —— 训练端与 Rollout 端即使面对非常接近的数值,也可能因为 FP4 的离散量化边界而落到不同码点,最终把微小的数值误差放大成策略偏差导致训练崩溃。 阿里千问联合俄亥俄州立大学的研究团队提出了 TRACE。 它的核心思路并不是继续分别提高训练端和 Rollout 端的量化精度,而是直接减少两条低精度执行路径之间的差异:用 Rollout 侧真实的 FP4 量化结果,反过来指导训练侧的舍入方向。

阅读原文(jiqizhixin.com)→

行业新闻机器之心2026-10-23原文

相关内容