arXiv:2510.13713cs.LGmath.OC2025-10被引 2

用弗兰克-沃尔夫算法优化大模型剪枝,无需重训练也能高效降本。

Don't Be Greedy, Just Relax! Pruning LLMs via Frank-Wolfe

  • 将剪枝问题转化为凸松弛,用弗兰克-沃尔夫算法求解
  • 在GPT架构上显著降低每层剪枝误差,性能超越主流基线
  • 适合追求高效率剪枝的大模型部署场景

剪枝是降低神经网络计算与存储开销的常用技术。传统方法需通过重训练恢复剪枝导致的性能下降,而当前最先进的大语言模型(LLM)剪枝方法采用逐层剪枝,在小校准数据集上最小化每层剪枝误差,避免全量重训练——这对大模型而言成本过高。然而,寻找最优剪枝掩码是一个困难的组合优化问题,求解至最优不可行。现有方法依赖贪婪启发式,忽略权重间的相互作用。本文提出将组合约束进行凸松弛,并使用弗兰克-沃尔夫(FW)算法求解。该方法大幅降低每层剪枝误差,在主流GPT架构上优于强基线,且保持内存效率。我们通过理论证明:结合FW算法的收敛性,对松弛解进行整数取整后,可获得原组合问题的近似解。

原文摘要 · Abstract (English)

Pruning is a common technique to reduce the compute and storage requirements of Neural Networks. While conventional approaches typically retrain the model to recover pruning-induced performance degradation, state-of-the-art Large Language Model (LLM) pruning methods operate layer-wise, minimizing the per-layer pruning error on a small calibration dataset to avoid full retraining, which is considered computationally prohibitive for LLMs. However, finding the optimal pruning mask is a hard combinatorial problem and solving it to optimality is intractable. Existing methods hence rely on greedy heuristics that ignore the weight interactions in the pruning objective. In this work, we instead consider the convex relaxation of these combinatorial constraints and solve the resulting problem using the Frank-Wolfe (FW) algorithm. Our method drastically reduces the per-layer pruning error, outperforms strong baselines on state-of-the-art GPT architectures, and remains memory-efficient. We provide theoretical justification by showing that, combined with the convergence guarantees of the FW algorithm, we obtain an approximate solution to the original combinatorial problem upon rounding the relaxed solution to integrality.

大模型剪枝优化算法弗兰克-沃尔夫

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