arXiv:2602.11184cs.LGcs.AI2026-02中稿 · ICLR被引 4

提出新方法让专家模型在极低比特下压缩不丢性能,适合边缘设备部署。

KBVQ-MoE: KLT-guided SVD with Bias-Corrected Vector Quantization for MoE Large Language Models

  • 用KLT-SVD共享专家间冗余权重,减少代码本浪费
  • 3比特量化后模型准确率接近原始精度(67.99 vs 68.07)
  • 适合资源受限场景,如移动端或嵌入式设备

混合专家(MoE)模型通过稀疏激活实现高性能与高效率的平衡,但其庞大的参数量和内存需求限制了在资源受限环境中的部署。向量量化(VQ)可通过代码本实现大语言模型的超低比特压缩,但直接应用于MoE常导致性能大幅下降,原因有二:一是专家间存在冗余表示,导致代码本重复量化相似特征;二是专家聚合放大输出偏差,引发量化后分布偏移。为此,本文提出KBVQ-MoE框架,融合两项技术:(1) 基于输入驱动的冗余消除,利用卡亨-洛维变换(KLT)引导的奇异值分解(SVD)提取主导权重分量并在专家间共享;(2) 偏差修正的输出稳定机制,仅对专家特异性(非冗余)表示进行量化,并通过通道级仿射补偿校正量化输出。在多个MoE大模型上的实验表明,该方法显著优于现有量化方法。例如,3比特量化Qwen1.5-MoE-A2.7B模型平均准确率达67.99,几乎与FP16基线(68.07)相当,验证了其在边缘设备等资源受限平台高效部署的潜力。

原文摘要 · Abstract (English)

Mixture of Experts (MoE) models have achieved great success by significantly improving performance while maintaining computational efficiency through sparse expert activation. However, their enormous parameter sizes and memory demands pose major challenges for deployment in resource-constrained environments. Vector Quantization (VQ) offers a promising approach for ultra-low-bit compression in Large Language Models (LLMs) by leveraging a codebook, where weight vectors are mapped to the most similar discrete codewords. Yet, directly applying VQ to MoEs often leads to substantial performance degradation due to two critical obstacles: (1) redundant representations among experts cause VQ to repeatedly quantize similar representations for each expert, resulting in inefficient use of limited codebook capacity; and (2) cumulative output bias is amplified by expert aggregation in MoE layers, leading to distributional shifts in the quantized outputs. To address these issues, we propose KBVQ-MoE, a novel VQ framework to enhance extremely low-bit quantization for MoE-based LLMs. KBVQ-MoE integrates two techniques: (1) input-driven redundancy elimination, where a Karhunen-Loeve Transform (KLT) guided singular value decomposition (SVD) extracts dominant weight components and shares them across experts; and (2) bias-corrected output stabilization, where vector quantization is applied only to expert-specific (non-redundant) representations and the quantized outputs are corrected via channel-wise affine compensation. Experiments on various MoE LLMs demonstrate that KBVQ-MoE preserves accuracy substantially better than existing quantization methods. For example, 3-bit quantization of Qwen1.5-MoE-A2.7B achieves an average accuracy of 67.99, nearly identical to the FP16 baseline of 68.07, underscoring KBVQ-MoE's potential for efficient deployment on edge devices and other resource-constrained platforms.

MoE模型向量量化低比特压缩边缘部署

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