arXiv:2510.01650cs.LGcs.AI2025-10被引 5

提出无需代理目标的剪枝方法,实现90%极稀疏下仍保持高精度。

The Unseen Frontier: Pushing the Limits of LLM Sparsity with Surrogate-Free ADMM

  • 用无代理的ADMM优化直接解决传统剪枝方法瓶颈
  • 在LLaMA-2-7B上90%稀疏时困惑度降低7.8倍
  • 支持极端稀疏(95%)且推理速度提升3.98倍,适合大模型部署

神经网络剪枝是缓解大语言模型计算与内存开销的有前景技术。然而,当前方法难以突破中等稀疏度(50%-60%)而不严重损害模型性能。本文提出名为$ exttt{Elsa}$的原理性有效方法,实现高达90%的极稀疏度同时保持高模型保真度。该方法通过识别现有实践中的多个局限——均源于对代理目标函数的依赖——并利用标准且成熟的基于ADMM的约束优化技术直接应对。大量实验覆盖多种模型与规模显示,$ exttt{Elsa}$ 显著优于现有方法;例如,在LLaMA-2-7B上90%稀疏时,困惑度比最佳现有方法低7.8倍。此外,$ exttt{Elsa}$ 在极端稀疏(如95%)下依然稳定,相较稠密模型可实现最高3.98倍推理加速和7.80倍内存压缩。我们还提出$ exttt{Elsa}_{-L}$量化变体,可扩展至270亿参数大模型,并建立其理论收敛保证。这些结果表明,大模型稀疏前沿取得实质性进展,且尚未探索的方向仍蕴藏巨大提升潜力。

原文摘要 · Abstract (English)

Neural network pruning is a promising technique to mitigate the excessive computational and memory requirements of large language models (LLMs). Despite its promise, however, progress in this area has diminished, as conventional methods are seemingly unable to surpass moderate sparsity levels (50-60%) without severely degrading model accuracy. This work breaks through the current impasse, presenting a principled and effective method called $\texttt{Elsa}$, which achieves extreme sparsity levels of up to 90% while retaining high model fidelity. This is done by identifying several limitations in current practice, all of which can be traced back to their reliance on a surrogate objective formulation. $\texttt{Elsa}$ tackles this issue directly and effectively via standard and well-established constrained optimization techniques based on ADMM. Our extensive experiments across a wide range of models and scales show that $\texttt{Elsa}$ achieves substantial improvements over existing methods; e.g., it achieves 7.8$\times$ less perplexity than the best existing method on LLaMA-2-7B at 90% sparsity. Moreover, we show that $\texttt{Elsa}$ remains stable even at extreme sparsity (e.g., 95\%), yielding up to $\times$3.98 inference speedup and $\times$7.80 memory compression over its dense counterpart. We also present $\texttt{Elsa}_{-L}$, a quantized variant that scales to extremely large models (27B), and establish its theoretical convergence guarantees.These results highlight meaningful progress in advancing the frontier of LLM sparsity, while promising that significant opportunities for further advancement may remain in directions that have so far attracted limited exploration.

模型剪枝大模型压缩稀疏性优化算法

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。