arXiv:2602.02891cs.LGcs.CL2026-02

无需训练即可高效找到最优大模型剪枝方案,提升推理效率。

TraceNAS: Zero-shot LLM Pruning via Gradient Trace Correlation

  • 通过梯度迹相关性构建零样本代理,全局评估模型结构重要性。
  • 单卡8.5小时完成剪枝搜索,节省90%以上计算资源。
  • 适合追求高效部署的大模型应用开发者与研究者。

结构化剪枝对大语言模型的高效部署至关重要。由于模型不同子模块对剪枝的敏感性差异,需识别最优非均匀剪枝方案。现有方法仅局部评估层、注意力头或权重通道的重要性,忽略模型内部复杂的全局结构依赖。训练感知剪枝虽能捕捉全局依赖,但计算开销巨大。为此,我们提出TraceNAS,一种无需训练的神经架构搜索框架,联合探索大模型深度与宽度的结构化剪枝。TraceNAS利用尺度不变的零样本代理,识别损失曲面与预训练模型高度对齐的剪枝模型,有效筛选出后剪枝训练中性能潜力最大的模型。该方法极为高效,在单张GPU上8.5小时内完成高保真剪枝模型发现,相比训练感知方法减少10倍GPU小时消耗。在Llama与Qwen系列模型上的评估表明,TraceNAS在常识与推理基准上表现媲美训练感知基线。

原文摘要 · Abstract (English)

Structured pruning is essential for efficient deployment of Large Language Models (LLMs). The varying sensitivity of LLM sub-blocks to pruning necessitates the identification of optimal non-uniformly pruned models. Existing methods evaluate the importance of layers, attention heads, or weight channels in isolation. Such localized focus ignores the complex global structural dependencies that exist across the model. Training-aware structured pruning addresses global dependencies, but its computational cost can be just as expensive as post-pruning training. To alleviate the computational burden of training-aware pruning and capture global structural dependencies, we propose TraceNAS, a training-free Neural Architecture Search (NAS) framework that jointly explores structured pruning of LLM depth and width. TraceNAS identifies pruned models that maintain a high degree of loss landscape alignment with the pretrained model using a scale-invariant zero-shot proxy, effectively selecting models that exhibit maximal performance potential during post-pruning training. TraceNAS is highly efficient, enabling high-fidelity discovery of pruned models on a single GPU in 8.5 hours, yielding a 10$\times$ reduction in GPU-hours compared to training-aware methods. Evaluations on the Llama and Qwen families demonstrate that TraceNAS is competitive with training-aware baselines across commonsense and reasoning benchmarks.

大模型剪枝零样本搜索高效部署结构化剪枝

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