跳到正文
HuggingFace Daily Papers·· 1 天前AI 评分43

TRACE:面向 MoE 语言模型 FP4 强化学习的 rollout 引导量化感知训练

TRACE: Rollout-Guided Quantization-Aware Training for FP4 Reinforcement Learning of MoE Language Models

AI 导读

TRACE 是面向 MoE 语言模型 RL 训练的 FP4 量化框架,用 rollout 侧量化结果引导训练侧 FP4 舍入决策,直接缩小训练与 rollout 两条量化执行路径的差异。它支持 FP4 权重/激活与 FP4 KV-cache rollout,在四个大规模 MoE 语言模型的推理、编码和长程 RL 任务上性能与 BF16 rollout 相当,rollout 提速最高 5.4 倍。

来源:HuggingFace Daily Papers · arxiv.org