通过激活引导的结构化正则化,提升大模型量化抗异常值能力。
Astro: Activation-guided Structured Regularization for Outlier-Robust LLM Post-Training Quantization
- 利用激活信息引导权重重构,抑制高激活对应的异常权重。
- 在LLaMA-2-7B上量化时间仅为复杂旋转方法的1/3,性能更优。
- 零推理延迟,兼容主流量化方法,适合高效部署场景。
权重仅量化后训练量化(Weight-only PTQ)对大型语言模型高效部署至关重要,但受权重与激活异常值影响导致精度下降。现有缓解策略常存在抑制不足或引入显著部署效率问题,如推理延迟、繁重预处理或依赖复杂算子融合。我们基于一个关键洞察:过参数化的LLM通常收敛至平坦极小值,意味着存在巨大等效解空间,可调整权重而不损害精度。基于此,我们提出Astro——一种激活引导的结构化正则化框架,以硬件友好且高效的方式抑制异常值负面影响。借助激活引导的正则化目标,Astro主动重构内在鲁棒权重,对高幅激活对应的权重异常值进行强抑制,同时保持模型精度。关键的是,Astro引入零推理延迟,且与GPTQ等主流量化方法正交。大量实验表明,Astro表现极具竞争力;尤其在LLaMA-2-7B上,其性能优于复杂学习型旋转方法,量化时间几乎仅为后者的1/3。
原文摘要 · Abstract (English)
Weight-only post-training quantization (PTQ) is crucial for efficient Large Language Model (LLM) deployment but suffers from accuracy degradation caused by weight and activation outliers. Existing mitigation strategies often face critical limitations: they either yield insufficient outlier suppression or incur significant deployment inefficiencies, such as inference latency, heavy preprocessing, or reliance on complex operator fusion. To resolve these limitations, we leverage a key insight: over-parameterized LLMs often converge to Flat Minima, implying a vast equivalent solution space where weights can be adjusted without compromising accuracy. Building on this, we propose Astro, an Activation-guided Structured Regularization framework designed to suppress the negative effects of outliers in a hardware-friendly and efficient manner. Leveraging the activation-guided regularization objective, Astro actively reconstructs intrinsically robust weights, aggressively suppressing weight outliers corresponding to high-magnitude activations without sacrificing model accuracy. Crucially, Astro introduces zero inference latency and is orthogonal to mainstream quantization methods like GPTQ. Extensive experiments show that Astro achieves highly competitive performance; notably, on LLaMA-2-7B, it achieves better performance than complex learning-based rotation methods with almost 1/3 of the quantization time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。