掘金 人工智能 9小时前
华为中科大联创大模型低比特量化算法,1‰数据实现昇腾无损压缩 7 倍
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

华为诺亚方舟实验室联合中科大提出CBQ(Cross-Block Quantization)方案,针对大模型压缩难题,仅用0.1%的训练数据实现7倍压缩率,同时保留99%的精度。该方案通过跨块依赖机制、自适应LoRA-Rounding技术和粗到细的预处理策略,有效解决了极低比特量化带来的性能下降问题。CBQ已应用于华为盘古大模型和多个开源LLM,并在昇腾芯片上实现高效部署,标志着大模型在国产算力上的普及加速。

🚀 CBQ 方案的核心在于**跨块依赖机制(CBD)**,它通过滑动窗口优化多个Transformer块的量化参数,捕捉模型内部的长距离依赖关系,提升量化性能。这种方法使得相邻块能协同量化,从而提高整体精度。

💡 **自适应LoRA-Rounding 技术**是CBQ的另一大亮点,该技术通过低秩矩阵学习量化权重的自适应补偿值,动态调整权重量化精度,更好地适应模型的内部结构和数据分布。与传统方法相比,LoRA-Rounding 减少了可学习参数,训练速度更快,内存消耗更低。

🔍 针对异常值问题,CBQ采用了**粗到细的预处理策略(CFP)**。该策略通过分阶段检测和处理权重和激活中的异常值,精确识别异常值位置,有效减少量化误差,确保模型在低比特场景下稳定运行。

【新智元导读】大模型巨无霸体量,让端侧部署望而却步?华为联手中科大提出 CBQ 新方案,仅用 0.1% 的训练数据实现 7 倍压缩率,保留 99% 精度。

大模型,如今堪称 AI 界的「吞金巨兽」。

从写诗到解题,从对话到编程,它们几乎无所不能,但动辄千亿甚至万亿参数的规模,让部署成本高得离谱。

以 FP16 精度部署的 DeepSeek-R1 671B 为例,推理时大概需要 1342GB 的显存,如果是 32GB 5090 显卡,需要整整 42 张!

为了降低成本,天才工程师们想出了**后训练量化(Post-training Quantization,PTQ)的方法,**它能够在有限的校准数据和计算资源下对模型进行高效压缩。

但是 PTQ 依然带来新的问题——在极低比特精度(如 W2A16、W4A4)时往往会出现明显的性能下降,规模是降了,但是不好用了!

就在这关键时刻**,华为诺亚方舟实验室联合中科大****亮出了「杀手锏」——CBQ(Cross-Block Quantization),一种基于跨块重建的后训练量化方案。**

论文地址:openreview.net/pdf?id=eW4y…

相比量化感知训练(QAT)所需数据量,CBQ 仅用 0.1% 的训练数据,一键压缩大模型至 1/7 体积——浮点模型性能保留 99%,真正实现「轻量不降智」。

值得一提的是,这项成果已荣登 ICLR 2025 Spotlight(录取率仅 5%)。

它不仅展现了大模型压缩领域的创新性和实用性,更像一颗信号弹,宣告大模型在国产算力上的普及时代已然来临!

目前,CBQ 已作为可调用的算法之一,正式加入昇腾模型压缩工具包 ModelSlim,帮助开发者在昇腾芯片上实现 LLM 的高效部署。

极低比特量化,为何如此难?

长期以来,后训练量化(PTQ)一直是压缩大语言模型的「黑科技」——通过解决异常值和采用 layer-wise 或 block-wise 的 loss 优化技术取得了比较不错的结果。

但是当把参数比特「压得特别低」的时候,模型性能会严重下降。

为什么极低比特量化,如此困难?其实,答案隐藏在大模型的复杂结构中。

研究者们对 LLM 在低比特量化场景下的量化误差进行了深入分析,发现了问题的关键所在:

随着模型参数数量的增加和量化 bit 数的减少,模型内部的层间依赖(inter-layer dependencies)和层内依赖(intra-layer dependencies)会显著增强,这严重影响了量化精度。

如下实验所示,清晰展示了 LLAMA-7B 层间与层内的依赖关系。

图 1:Llama-7B 内部权重和层之间依赖关系的变化,以及层间缩放因子(scale)对误差的影响

图 1(a)为 LLAMA-7B 单一层中权重的 Hessian 矩阵绝对值可视化,2-bit 图比 4-bit 更模糊,非对角线噪声增多,表示在低比特下权重间的「干扰」增强了。

图 1(b)为 LLAMA-7B 32 层中损失相对于 scale 的 Hessian 矩阵可视化,在 2-bit 量化中,非对角线明显比 4-bit 更亮,说明层间依赖增强,模型更容易因为一层的误差影响到另一层。

以及图 1(c)LLAMA-7B 前两个 Transformer 块的平均 scale 与相应损失之间的关系,4-bit 情况下,误差平稳区域大,模型对 scale 不敏感。2-bit 情况下,误差对 scale 非常敏感,选择不当误差急剧上升,黑色区域更集中、易出错。

总结来说,将模型参数从高精度压缩到低精度,这一过程主要面临三大核心挑战:

1**.** 层间依赖的「雪球效应」

大模型由多个 Transformer 层组成,各层参数之间存在复杂的相互依赖。

在极低比特量化时,量化误差会在层间不断累积放大,就像「滚雪球」一样,导致整体性能严重下降。

然而,传统逐层量化的方法,无法有效捕捉这些层间依赖,进而造成了精度损失。

2**.** 层内依赖的复杂性

同一层内的参数并非独立存在,而是存在紧密的关联性。

极低比特量化会破坏这些精细的层内依赖,导致模型在处理复杂任务时「力不从心」。

比如,大模型语义理解或推理能力,可能因参数精度的降低而显著退化。

3**.** 权重和激活的异常值

模型的权重和激活值中的异常值,在低比特量化时会引发较大的误差。

传统的方法无法精确识别和处理这些异常值,进一步加剧了量化误差。

可见,这些挑战让低比特量化,成为大模型压缩的「拦路虎」。

那么,华为的 CBQ 方案,是如何突破这些瓶颈?让我们一探究竟!

CBQ 打破层间壁垒,精准又高效

CBQ 的核心思想是,通过跨块依赖(Cross-Block Dependency, CBD)机制和自适应 LoRA-Rounding 技术,同时优化多个 Transformer 块的量化参数,从而更好地保留模型内部的依赖关系。

具体来说,它通过三大技术创新,为极低比特量化注入了全新活力。

跨块依赖机制(CBD)

刚刚也提到,传统量化方法采用逐层优化,却忽视了层间依赖的复杂性。

CBQ 引入了 CBD 机制,通过滑动窗口的方式,同时优化多个 Transformer 块,并且相邻窗口之间会有重叠的块,以确保块之间的连接性和协作性。

这种方法,可以有效地捕捉到模型内部的长距离依赖关系,使得相邻的块能够共同参与到量化过程中,从而提高整体的量化性能。

在实验中,随着滑动窗口中块的数量增加,模型的性能也得到了显著提升。

自适应 LoRA-Rounding 技术

为了应对层内依赖的复杂性,CBQ 提出了自适应 LoRA-Rounding 技术,通过两个低秩矩阵来学习量化权重的自适应补偿值。

与传统的 AdaRound 方法相比,LoRA-Rounding 通过低秩分解大大减少了可学习参数,训练速度更快,GPU 内存消耗更低。

这种方法能够在训练过程中动态调整权重的量化精度,从而更好地适应模型的内部结构和数据分布。

粗到细的预处理策略(CFP)

针对异常值问题,CBQ 采用了粗到细的预处理策略(Coarse-to-Fine Preprocessing, CFP)。

CFP 策略从统计学的角度出发,通过分阶段检测和处理权重和激活中的异常值。

在粗粒度检测阶段,通过计算四分位数和四分位距来初步估计异常值的范围;在细粒度检测阶段,通过最小化异常值子集与正常值子集之间的距离,同时最大化子集内部的方差,来精确识别异常值的位置。

这种分阶段策略,有效减少了量化误差,确保模型在低比特场景下依然「稳如泰山」。

那么,CBQ 在场景中的真实表现又如何呢?

实验结果:性能与效率的双赢

一系列研究结果显示,CBQ 在华为盘古模型和开源模型的表现上,大放异彩。

盘古模型:端侧部署「杀手锏」

CBQ 量化技术已成功应用于华为盘古大模型 PanGu-7B 和 PanGu-1.5B 的端侧部署,凭借其高精度的量化性能,有效支撑了盘古大模型在多个业务场景的落地应用。

如下表所示,在 W8A8/W4A16 精度下,PanGu-1.5B 模型在中文(C-Eval/CMMLU)、多任务语言理解(MMLU)基准中的表现,毫不逊色于全精度模型的性能。

在中文、多语言理解、数学基准中,PanGu-7B 的表现同样如此。

这些成果,足以让盘古模型在手机等终端设备上,轻松运行。

开源模型:超越最优

此外,CBQ 在多个开源 LLM(如 OPT、LLaMA)上也取得了 SOTA。

例如,在 W4A16、W2A16 和 W4A8 等低比特量化设置下,CBQ 的性能均优于现有的最先进方法,并且与全精度模型的性能差距缩小到了 1% 以内。

更令人惊叹的是,CBQ 仅需 4.3 小时即可完成对 4 位权重的 LLaMA1-65B 模型的量化,展现了压缩率与精度之间的完美平衡(trade-off)。

未来展望

华为的 CBQ 方案,以跨块依赖机制、自适应 LoRA-Rounding 技术,以及粗到细的预处理策略,成功征服了极低比特量化的「三大高峰」。

这项创新有效地解决了,大模型在低比特量化场景下所面临的层间依赖和层内依赖难题。

它不仅在多种大语言模型和数据集上展现出了显著的性能提升,成功缩小了与全精度模型之间的差距,还以高效的量化效率实现了复杂模型的快速压缩。

最终,让盘古和各类开源模型,成功实现了在昇腾硬件上的高效部署,并为更加广泛的应用铺就坦途。

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

CBQ 大模型压缩 量化 华为
相关文章