arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.07596cs.LGcs.AI
收藏

Astro: 基于激活引导的结构正则化用于鲁棒性LLM后训练量化

Astro: Activation-guided Structured Regularization for Outlier-Robust LLM Post-Training Quantization

Beijing Institute of Technology(北京理工大学);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Hebei Province Key Laboratory of Big Data Science and Intelligent Technology(河北省大数据科学与智能技术重点实验室)

Xi Chen, Ming Li, Junxi Li, Changsheng Li, Peisong Wang, Lizhong Ding, Ye Yuan, Guoren Wang

AI总结:

Astro通过激活引导的结构正则化方法,在高效且无延迟的情况下提升LLM后训练量化的鲁棒性与性能。

摘要:

仅权重的后训练量化(PTQ)对于高效大型语言模型(LLM)部署至关重要,但受到权重和激活异常值导致的精度下降的困扰。现有的缓解策略往往面临关键限制:要么导致抑制不足,要么带来显著的部署效率问题,如推理延迟、预处理负担重或依赖复杂的操作融合。为了解决这些限制,我们利用一个关键见解:过参数化的LLM通常收敛到平坦极小值,这意味着存在一个庞大的等效解空间,其中权重可以调整而不影响精度。基于此,我们提出了Astro,一种基于激活引导的结构正则化框架,旨在以硬件友好的高效方式抑制异常值的负面影响。利用激活引导的正则化目标,Astro主动重建内在稳健的权重,强烈抑制与高幅激活对应的权重异常值,而不牺牲模型精度。关键的是,Astro引入了零推理延迟,并且与主流量化方法如GPTQ正交。广泛的实验表明,Astro实现了高度竞争性的性能;值得注意的是,在LLaMA-2-7B上,其性能优于复杂的基于学习的旋转方法,几乎在量化时间上减少了三分之一。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑