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

Týr-the-Pruner:通过全局稀疏度分布优化实现大语言模型结构化剪枝

Týr-the-Pruner: Structural Pruning LLMs via Global Sparsity Distribution Optimization

Advanced Micro Devices, Inc.(先进微器件公司)

Guanchen Li, Yixing Xu, Zeping Li, Ji Liu, Xuanwu Yin, Dong Li, Emad Barsoum

AI总结:

本文提出端到端全局结构化剪枝框架 Týr-the-Pruner,通过超网络搜索优化各层稀疏度分布,在剪去 Llama-3.1-70B 半数参数时保留97%性能。

摘要:

结构化剪枝可提升大语言模型(LLM)与硬件无关的推理效率,但往往难以保持可比性能。局部剪枝能够执行高效的逐层压缩,却忽视全局拓扑。尽管全局剪枝旨在寻找最优稀疏模型,但直观方法通常采用两阶段范式:先评估子结构显著性,再执行全局剪枝,这忽视了结构间依赖,无法实现端到端优化。为克服这些局限,我们提出 Týr-the-Pruner,这是一个高效、基于搜索的端到端全局结构化剪枝框架。该框架通过在一系列稀疏率上对 LLM 的每一层反复应用局部剪枝来构建超网络,核心目标是在目标总体稀疏率下确定最优稀疏度分布。具体而言,我们引入有效的局部剪枝和期望误差累积方法以改进超网络构建。此外,我们采用具有由粗到细稀疏粒度的迭代剪枝与搜索策略,以确保高效的搜索收敛。实验结果表明,Týr-the-Pruner 实现了最先进的结构化剪枝效果:在移除 Llama-3.1-70B 高达50%的参数这一具有挑战性的设置下,仍保留稠密模型97%的性能。代码将发布于 https://github.com/AMD-AGI/Tyr-the-Pruner。

英文摘要:

Structural pruning enhances hardware-agnostic inference efficiency for large language models (LLMs) yet often fails to maintain comparable performance. Local pruning performs efficient layer-by-layer compression but ignores global topology. Although global pruning aims to identify an optimal sparse model, intuitive methods typically adopt a two-stage paradigm that first evaluates substructure saliency and then applies global pruning, which ignores inter-structure dependencies and fails to achieve end-to-end optimization. To address these limitations, we propose Týr-the-Pruner, an efficient end-to-end search-based global structural pruning framework. This framework constructs a supernet by repeatedly applying local pruning across a range of sparsity ratios to each layer in an LLM, with the core goal of determining the optimal sparsity distribution under a target overall sparsity ratio. Concretely, we introduce an effective local pruning and an expectation error accumulation approach to improve supernet construction. Furthermore, we employ an iterative prune-and-search strategy with coarse-to-fine sparsity granularity to ensure efficient search convergence. Experimental results show that Týr-the-Pruner achieves state-of-the-art structural pruning, retaining 97% of the dense model's performance while removing a challenging 50% of Llama-3.1-70B's parameters. Code will be available at https://github.com/AMD-AGI/Tyr-the-Pruner.

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

↑