arXiv:2602.01410cs.LGcs.AR2026-02中稿 · ASPLOS 2026

提出自适应混合精度框架SNIP,实现亚字节训练下高效保质的LLM训练。

SNIP: An Adaptive Mixed Precision Framework for Subbyte Large Language Model Training

  • 根据前向损失与反向权重偏差动态调整各层精度
  • 在1B至70B模型上降低80%计算量且保持质量
  • 适合追求极致推理效率的LLM训练场景

高效训练大语言模型(LLM)同时保持模型质量面临重大挑战,尤其在支持亚字节精度的先进GPU上。现有混合精度训练方法或对所有GEMM操作采用统一精度,或依赖难以泛化的启发式策略,导致收敛不佳和训练不稳定。为此,本文提出SNIP,一种面向亚字节精度的细粒度自适应混合精度训练框架。SNIP定期收集激活值、梯度和优化器状态的统计信息,评估量化引起的精度损失。定义两个关键指标:前向传递中的损失发散(由量化导致的训练损失增加)和反向传递中的权重发散(衡量梯度误差传播对模型更新的影响)。这些指标引导求解整数线性规划(ILP)问题,系统优化各层精度以最小化整体质量损失并满足效率目标。在1B、3B、7B和70B Llama类模型上的实验表明,SNIP持续优于现有基线,在不同模型规模和训练阶段均实现高达80%的FLOPs减少,同时保持模型质量,计算开销极低。

原文摘要 · Abstract (English)

Training large language models (LLMs) efficiently while preserving model quality poses significant challenges, particularly with subbyte precision supported by state-of-the-art GPUs. Current mixed-precision training approaches either apply uniform precision to all GEMM operations or rely on heuristic-based methods that fail to generalize during training, leading to suboptimal convergence and instability. To address these challenges, this paper introduces SNIP, a fine-grained adaptive mixed-precision training framework for LLM pretraining that supports subbyte precision. SNIP periodically collects statistics on activations, gradients, and optimizer states to assess the precision loss impact on model quality. We define two key metrics: loss divergence in the forward pass, caused by quantization-induced increases in training loss, and weight divergence in the backward pass, which measures error propagation through gradients affecting model updates. These metrics guide an Integer Linear Programming (ILP) problem that systematically optimizes layerwise precision to minimize overall quality loss while meeting efficiency targets. Experiments on 1B, 3B, 7B and 70B Llama-like models demonstrate that SNIP consistently outperforms existing baselines, reducing FLOPs by up to 80% while preserving model quality across different model sizes and training phases with minimal computational overhead.

大模型训练混合精度亚字节优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。