跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

EdgeRazor:一种通过混合精度量化感知蒸馏实现的大语言模型轻量级框架

文章摘要

EdgeRazor是一种新的轻量级框架,用于在资源受限设备上部署大语言模型。它采用混合精度量化感知蒸馏,在极低比特宽度下实现高压缩比和优越性能,同时训练预算远低于现有方法。

来源arXiv Machine Learning作者: Shu-Hao Zhang, Le-Tong Huang, Xiang-Sheng Deng, Xin-Yi Zou, Chen Wu, Nan Li, Shao-Qun Zhang
EdgeRazor:一种通过混合精度量化感知蒸馏实现的大语言模型轻量级框架
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

近年来,将大语言模型(LLM)部署到智能手机、物联网设备等资源受限的边缘计算环境成为研究热点。量化技术通过将全精度(如16位)的权重和激活值转换为低位(如8位、4位甚至更低)表示,显著减少了模型存储和计算开销。然而,传统量化方法存在明显瓶颈:训练后量化(PTQ)仅在小数据集上校准参数,无需重训练,但在4位以下精度时性能急剧下降;量化感知训练(QAT)使用代理梯度搜索低比特参数,但计算资源需求巨大;量化感知蒸馏(QAD)结合了QAT与全精度教师的知识迁移,却需要手动选择蒸馏特征并严重依赖教师数据。

针对上述缺陷,由Shu-Hao Zhang等人提出的EdgeRazor框架提供了一种全新的解决方案。该框架由三个创新模块构成:一是混合精度量化感知蒸馏(Mixed-Precision Quantization-Aware Distillation),它允许为模型不同部分分配不同的比特宽度,实现精度的细粒度控制;二是自适应特征蒸馏(Adaptive Feature Distillation),能够从16位的教师模型自动导出不同比特宽度的学生模型,无需手动特征选择;三是熵感知KL散度(Entropy-Aware KL Divergence),该损失函数在人工标注数据集和蒸馏生成的数据集上自动平衡前向和反向KL散度,其平衡权重完全由教师输出的分布熵决定,消除了手工调参的需求。

研究者们在基础模型、指令微调模型以及多模态LLM上进行了广泛的实验验证。结果令人瞩目:仅1.88比特的EdgeRazor模型不仅超越了所有3比特的竞争方法,甚至比当前领先的2比特PTQ方法高出11.3个百分点的准确率,而训练预算仅为领先QAT方法的4到10分之一。在所有比特宽度下,EdgeRazor均实现了更高的压缩比:例如,1.58比特的Qwen3-0.6B模型存储从原始的1.41 GB骤降至0.28 GB,解码速度相对于16位基线提升了15.1倍。

这一研究成果为边缘设备上的大语言模型部署开辟了新路径,有望推动LLM在移动端和物联网场景中的实际应用,大幅降低推理成本和能耗。

展开要点与分析

文章情报

工程师进阶

要点

  • EdgeRazor引入了混合精度量化感知蒸馏,实现精度的细粒度控制。
  • 在1.88比特下超越所有3比特竞争者,并超过领先的2比特PTQ方法11.3个百分点。
  • 大幅减少存储(如Qwen3-0.6B从1.41 GB降至0.28 GB),解码速度提升15.1倍。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。

EdgeRazor:一种通过混合精度量化感知蒸馏实现的大语言模型轻量级框架 | AI News Hub