首页|期刊导航|南京大学学报(自然科学版)|基于输出激活引导的大模型通道级自适应稀疏度剪枝方法

基于输出激活引导的大模型通道级自适应稀疏度剪枝方法OA

OGAS:Output-activation guided pruning with adaptive sparsity for large language model

中文摘要英文摘要

针对大语言模型(Large Language Model,LLM)在端侧部署时面临的计算资源受限与显存占用过高问题,训练后剪枝(Post-Training Pruning,PTP)是一种高效的压缩手段.然而,现有的主流方法(如Wanda,SparseGPT)通常采用层级统一的稀疏度策略,忽视了不同层级与通道间显著的信息贡献异质性,而且,其评估多聚焦于输入侧强度,难以识别高能量静态冗余通道,导致高压缩率下模型精度严重衰减.为此,提出一种输出激活引导的通道级自适应稀疏度剪枝方法(OGAS).该方法首先构建融合输出激活能量范数与峰均比(Peak-to-Average Power Ratio,PAPR)的双重评价指标,从响应强度与特征特异性两个维度精准识别并保护稀疏关键特征;其次,设计了基于非线性曲率的连续映射机制,在连续空间内实现通道级稀疏度的动态自适应分配;此外,还引入黄金分割搜索算法来构建闭环优化流程,实现了关键超参数的层级自动寻优.在LLaMA-3和Mistral上的实验结果表明:在50%稀疏度下,OGAS将LLaMA-3.1-8B在WikiText-2数据集上的困惑度(Perplexity,PPL)降至7.99,相比于目前主流的一阶方法Wanda(PPL=8.85)取得了显著提升;在常识推理任务上的零样本平均准确率达到63.46%,和Wanda相比提升了1.6%.实验结果验证了OGAS能够更有效地保持大幅压缩后模型的语义理解与逻辑推理性能,在不同架构的模型上均表现出优异的稳健性与通用性.

Post-training pruning(PTP)has emerged as an efficient compression technique to address the challenges of limited computational resources and excessive memory footprint during the edge deployment of Large Language Models(LLMs).However,existing mainstream methods(e.g.,Wanda and SparseGPT)typically employ uniform layer-wise sparsity strategies,overlooking the significant heterogeneity in information contribution across different layers and channels.Moreover,their evaluation criteria predominantly focus on input-side intensity,making it difficult to identify high-energy static redundant channels,which leads to severe model performance degradation under high compression ratios.To address these limitations,this paper proposes OGAS,an Output-activation Guided Adaptive Sparsity pruning method at the channel level.First,a dual evaluation metric is constructed by integrating the output activation energy norm with the Peak-to-Average Power Ratio(PAPR)to accurately identify and protect sparse key features from the dimensions of response intensity and feature specificity.Second,a continuous mapping mechanism based on non-linear curvature is designed to achieve dynamic adaptive allocation of channel-level sparsity within a continuous space.Furthermore,a closed-loop optimization workflow is established by introducing the Golden Section Search algorithm to realize the automated layer-wise tuning of critical hyperparameters.Experimental results on mainstream open-source models,including LLaMA-3 and Mistral,demonstrate that at a 50%sparsity ratio,OGAS reduces the perplexity(PPL)of LLaMA-3.1-8B on the WikiText-2 dataset to 7.99,a significant improvement over the state-of-the-art first-order method Wanda(8.85).In common sense reasoning tasks,the average zero-shot accuracy reached 63.46%,representing a 1.6%improvement over Wanda.The results verify that OGAS effectively maintains the semantic understanding and logical reasoning capabilities of models after large-scale compression,exhibiting superior robustness and versatility across different model architectures.

李沛鸿;贺傍;周彤昕;李丽;傅玉祥

南京大学集成电路学院,苏州,215163南京大学集成电路学院,苏州,215163南京大学集成电路学院,苏州,215163VLSI实验室,南京大学电子科学与工程学院,南京,210023南京大学集成电路学院,苏州,215163

信息技术与安全科学

大语言模型训练后剪枝自适应稀疏度峰均比(PAPR)

large language modelmodel pruningadaptive sparsityPeak-to-Average Power Ratio

《南京大学学报(自然科学版)》 2026 (3)

422-433,12

国家重点研发计划(2023YFB2806800),国家自然科学基金(U21B2032),苏州市"揭榜挂帅"重点项目(SYG2024134)

10.13232/j.cnki.jnju.2026.03.008

评论