arXiv:2602.07804cs.CLcs.AI2026-02中稿 · ICLR被引 4

用博弈论评估各层贡献,让大模型剪枝更精准高效

Pruning as a Cooperative Game: Surrogate-Assisted Layer Contribution Estimation for Large Language Models

  • 将模型层视为玩家,用代理网络估算每层边际贡献
  • 在多个数据集上降低困惑度并提升零样本准确率
  • 适合需要高效部署大模型的研究者和工程师

大型语言模型(LLMs)虽在各类任务中表现优异,但其实际部署受限于高昂的计算成本。层剪枝是降低推理开销的常用策略,但现有方法多依赖静态启发式规则,未能考虑层间的相互依赖关系,限制了剪枝效果。本文提出一种基于博弈论的框架,将层剪枝建模为合作博弈,每个层为参与者,模型性能为收益。由于精确计算Shapley值对大模型不可行,我们设计了一个轻量级代理网络,可低成本预测任意层组合下的模型性能。同时采用分层蒙特卡洛掩码采样进一步降低计算开销。该方法能捕捉层间依赖关系,动态识别关键层进行剪枝。大量实验表明,本方法在困惑度与零样本准确率上均具一致性优势,实现了更高效、更有效的层剪枝。

原文摘要 · Abstract (English)

While large language models (LLMs) demonstrate impressive performance across various tasks, their deployment in real-world scenarios is still constrained by high computational demands. Layer-wise pruning, a commonly employed strategy to mitigate inference costs, can partially address this challenge. However, existing approaches generally depend on static heuristic rules and fail to account for the interdependencies among layers, thereby limiting the effectiveness of the pruning process. To this end, this paper proposes a game-theoretic framework that formulates layer pruning as a cooperative game in which each layer acts as a player and model performance serves as the utility. As computing exact Shapley values is computationally infeasible for large language models (LLMs), we propose using a lightweight surrogate network to estimate layer-wise marginal contributions. This network can predict LLM performance for arbitrary layer combinations at a low computational cost. Additionally, we employ stratified Monte Carlo mask sampling to further reduce the cost of Sharpley value estimation. This approach captures inter-layer dependencies and dynamically identifies critical layers for pruning. Extensive experiments demonstrate the consistent superiority of our method in terms of perplexity and zero-shot accuracy, achieving more efficient and effective layer-wise pruning for large language models.

模型剪枝博弈论大模型优化

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