混合精度训练

英文:Mixed Precision Training 分类:architecture

基本释义

训练中混合使用多种数值精度(FP32 主权重+FP16/BF16 计算)的技术,在不损失精度的前提下提升速度并降低显存,是大模型训练的标配。

技术要点

实现:前向/反向用低精度(BF16/FP16),优化器状态与主权重保留 FP32;配套损失缩放(FP16 防下溢)与梯度裁剪;BF16 无需缩放且范围大,成为 LLM 主流;FP8 进一步提速但需更精细管理。

应用场景

用于大模型预训练与微调、分布式训练(降低通信)与算力受限集群;混合精度是"单位显存训练更大模型"的第一手段。

返回AI百科目录 · AI最前沿首页

相关词条:MoE 路由 · 多头注意力 · 分组查询注意力 · 激活重计算 · BF16