arXiv:2501.16385cs.LGcs.CL2025-01IJCAI被引 6

提出反馈量化方法,让大模型在手机等设备上更省内存、更准

FBQuant: FeedBack Quantization for Large Language Models

  • 借鉴自动控制的负反馈机制,稳定量化过程
  • 3比特量化下,Llama2-7B零样本准确率提升1.2%
  • 优化计算核,使额外推理时间减少60%

将大语言模型(LLMs)部署到边缘设备日益重要,可避免网络依赖、降低昂贵的API调用成本并提升用户隐私。然而,受限于边缘设备的计算资源,尤其是权重加载带来的内存带宽瓶颈,部署面临挑战。仅量化权重虽能减少内存访问,但常导致显著精度下降。近期引入子分支的方法虽有望缓解量化误差,但或缺乏稳健优化策略,或依赖次优目标。为此,我们提出反馈量化(FBQuant),灵感源自自动控制中的负反馈机制,确保重构权重始终受量化过程约束,从而降低过拟合风险。为缓解子分支带来的额外延迟,我们开发了高效CUDA内核,将额外推理时间减少60%。全面实验表明,FBQuant在多种LLM上均具高效性与有效性。值得注意的是,对于3比特的Llama2-7B,FBQuant将零样本准确率提升1.2%。

原文摘要 · Abstract (English)

Deploying Large Language Models (LLMs) on edge devices is increasingly important, as it eliminates reliance on network connections, reduces expensive API calls, and enhances user privacy. However, on-device deployment is challenging due to the limited computational resources of edge devices. In particular, the key bottleneck stems from memory bandwidth constraints related to weight loading. Weight-only quantization effectively reduces memory access, yet often induces significant accuracy degradation. Recent efforts to incorporate sub-branches have shown promise for mitigating quantization errors, but these methods either lack robust optimization strategies or rely on suboptimal objectives. To address these gaps, we propose FeedBack Quantization (FBQuant), a novel approach inspired by negative feedback mechanisms in automatic control. FBQuant inherently ensures that the reconstructed weights remain bounded by the quantization process, thereby reducing the risk of overfitting. To further offset the additional latency introduced by sub-branches, we develop an efficient CUDA kernel that decreases 60% of extra inference time. Comprehensive experiments demonstrate the efficiency and effectiveness of FBQuant across various LLMs. Notably, for 3-bit Llama2-7B, FBQuant improves zero-shot accuracy by 1.2%.

大模型量化边缘部署反馈机制低比特

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