arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2607.08027cs.CLcs.AI

通过幂变换和符号保留分数聚合与自适应特征保留对大语言模型进行结构化剪枝

Structured Pruning of Large Language Models via Power Transformation and Sign-Preserving Score Aggregation with Adaptive Feature Retention

Chubu University(楚ubu大学);Panasonic Holdings Corporation(松下电器控股公司)

Ryota Kobayashi, Tsubasa Hirakawa, Takayoshi Yamashita, Hironobu Fujiyoshi, Yasunori Ishii, Tomoyuki Okuno, Kazuki Kozuka

AI总结:

研究针对大语言模型结构化剪枝,改进自适应特征保留技术。结合幂变换、符号保留分数聚合及异常值去除方法,解决应用中出现的分布不匹配等问题。实验表明该方法能保持准确率,通过结构化剪枝加速推理。

摘要:

本文提出了一种改进的大语言模型结构化剪枝方法,解决了将非结构化剪枝技术自适应特征保留(AFR)应用于结构化剪枝时的关键挑战。应用AFR时会出现异构剪枝分数分布不匹配、指示优化方向一致性的符号信息丢失和异常值影响等问题。为此,提出了一种结合幂变换进行非线性分布对齐、符号保留分数聚合和基于百分位数的异常值去除的统一方法。在Llama-3-8B、Vicuna-v1.5-13B和LLaVA-v1.5-13B上的实验表明,该方法在保持与非结构化剪枝相当准确率的同时,通过结构化剪枝实现了实际推理加速。

英文摘要:

This paper proposes an improved structured pruning method for large language models (LLMs) that addresses key challenges in adapting Adaptive Feature Retention (AFR), an unstructured pruning technique, to structured pruning. When applying AFR to structured pruning, three major problems arise: distribution mismatch between heterogeneous pruning scores, loss of sign information indicating optimization direction consistency, and influence of outliers. To address these issues, we propose a unified approach combining power transformation for nonlinear distribution alignment, sign-preserving score aggregation, and percentile-based outlier removal. Experiments on Llama-3-8B, Vicuna-v1.5-13B, and LLaVA-v1.5-13B demonstrate that our method maintains accuracy comparable to unstructured pruning while achieving practical inference speedup through structured pruning.

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

↑