arXiv:2505.12216cs.CL2025-05ACL被引 3

一模型搞定多种压缩需求,提速28倍且不丢精度

One-for-All Pruning: A Universal Model for Customized Compression of Large Language Models

  • 用StratNet学习任意压缩请求对应的最优剪枝策略
  • 在64个请求下比基线快28倍,精度相当
  • 适合需要快速响应多用户压缩请求的场景

现有大模型剪枝方法虽能实现高压缩率并保持性能,但在处理多个并发请求时,耗时随请求数线性增长,难以满足真实场景需求。为此,我们提出通用定制化压缩模型UniCuCo,引入StratNet学习将任意压缩请求映射到最优剪枝策略。训练难点在于剪枝评估计算成本高且过程不可微,阻碍梯度传播。为此,我们采用高斯过程近似评估过程,其梯度可计算,从而近似非可微剪枝过程的梯度,实现StratNet更新。实验表明,UniCuCo在处理64个请求时比基线快28倍,且精度相当。

原文摘要 · Abstract (English)

Existing pruning methods for large language models (LLMs) focus on achieving high compression rates while maintaining model performance. Although these methods have demonstrated satisfactory performance in handling a single user's compression request, their processing time increases linearly with the number of requests, making them inefficient for real-world scenarios with multiple simultaneous requests. To address this limitation, we propose a Univeral Model for Customized Compression (UniCuCo) for LLMs, which introduces a StratNet that learns to map arbitrary requests to their optimal pruning strategy. The challenge in training StratNet lies in the high computational cost of evaluating pruning strategies and the non-differentiable nature of the pruning process, which hinders gradient backpropagation for StratNet updates. To overcome these challenges, we leverage a Gaussian process to approximate the evaluation process. Since the gradient of the Gaussian process is computable, we can use it to approximate the gradient of the non-differentiable pruning process, thereby enabling StratNet updates. Experimental results show that UniCuCo is 28 times faster than baselines in processing 64 requests, while maintaining comparable accuracy to baselines.

大模型压缩剪枝高效推理

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