跳到正文
arXiv · AI· Han Wang, Zhitao Wang, Natalia Frumkin, Xuanwu Yin, Alireza Khodamoradi, Dong Li, Emad Barsoum·· 5 小时前AI 评分34

ALTO 实现 GPT-OSS-20B 在 MXFP4 下高效训练,差距收窄至 5%

Training an MoE LLM in MXFP4: How We Closed the Gap to BF16

AI 导读

开源配方 ALTO 让 GPT-OSS-20B 在 MLPerf Small MoE 基准下以 MXFP4 格式训练,验证损失仅比 BF16 基线高 0.007,差距缩减约 50%。 该方案结合随机 Hadamard 变换与简化权重去振荡,使 FP4 相对 BF16 的收敛开销从 20.0% 降至 5.0%,在 AMD MI300 上完成模拟验证。

来源:arXiv · AI · arxiv.org