当前位置：首页 > news >正文

MIT：LLM自适应量化策略

news 2026/7/4 22:51:05

📖标题：Four Over Six: More Accurate NVFP4 Quantization with Adaptive Block Scaling
🌐来源：arXiv, 2512.02010

🌟摘要

随着大型语言模型的增长，NVFP4 等低精度数值格式因其提供的速度和内存优势而变得越来越流行。然而，为了用NVFP4加速计算，前向传递中的所有矩阵乘法操作数-权重和激活，后向传递中的权重、激活和梯度必须量化为NVFP4，通常会导致训练过程中的发散和推理过程中的性能下降。为了解决这个问题，在这项工作中，我们介绍了Four Over Six ，这是对 NVFP4 量化算法的修改，该算法评估每个块的两个潜在比例因子。与整数格式不同，FP4 等浮点格式在每个块的接近最大值上具有最大的量化误差，我们发现这主要负责下游性能下降。我们发现，对于某些块，缩放到较小的 FP4 值会使可表示值的分布更加均匀，提高了接近最大值的表示。重要的是，4/6 可以在 NVIDIA Blackwell GPU 上实现，这使得在使用 NVFP4 训练 LLM 时可以使用是可行的。在 Transformer 和混合模型架构的预训练实验中，我们发现 4/6 在某些情况下可以防止分歧，与使用当前最先进的 NVFP4 训练配方训练的模型相比，将训练损失显着更接近 BF16。我们还发现 4/6 可以很容易地合并到许多不同的训练后量化方法中，并且通常可以提高下游精度。我们希望这激发了未来使用 NVFP4 训练模型和部署模型的工作。项目在https://github.com/mit-han-lab/fouroversix

🛎️文章简介

🔸研究问题：如何通过自适应块缩放来提高NVFP4量化的精度？
🔸主要贡献：论文提出了一种新的量化方法Four Over Six，通过自适应缩放实现了NVFP4量化的准确性提升，并减少了计算开销。

📝重点思路

🔸引入了Four Over Six方法，该方法在处理量化时允许块使用不同的缩放值（4或6），以更准确地表示几乎最大的值。
🔸在预训练和后训练量化过程中，采用了基于均方误差（MSE）的缩放选择规则，比较不同缩放条件下的量化效果，以选择最优缩放策略。
🔸利用NVIDIA Blackwell GPU的PTX指令高效实现Four Over Six，确保在保持高性能的同时，量化过程的开销在合理范围内。

🔎分析总结

🔸Four Over Six显著减少了大型值的量化误差，提高了多种模型架构的预训练性能，解决了目前NVFP4训练过程中出现的发散问题。
🔸在后训练量化中，Four Over Six与现有方法（如GPTQ、AWQ、SmoothQuant）结合使用，可广泛提升不同任务的模型性能，尤其在Word Perplexity指标上显示出较大改善。
🔸引入Four Over Six的模型在各类任务上表现出更接近于高精度模型的效果，尤其在处理具有极大值的块时，展现出更佳的量化精度。