arXiv:2502.01705cs.LG2025-02被引 6

提出一种渐进式二值化与稀疏剪枝结合的压缩方法,显著提升大模型部署效率。

Progressive Binarization with Semi-Structured Pruning for LLMs

  • 分步引入稀疏性并优化二值化参数,降低剪枝与量化误差
  • 在多个大模型上实现更低困惑度和更高下游准确率
  • 适合资源受限设备部署,尤其对推理加速有需求的研究者

大语言模型在自然语言处理中取得显著进展,但其高计算与内存开销限制了在资源受限设备上的部署。二值化是极端量化形式,但二值模型仍存在冗余可进一步消除。剪枝可有效去除冗余,但与二值化直接结合常导致性能严重下降。本文提出一种后训练框架 PBS²P,将二值化与半结构化剪枝无缝集成。首先提出分步半结构化剪枝与二值化优化(SPBO),逐步引入稀疏性并联合优化二值化参数,以降低剪枝与量化误差,实现更稳定高效的压缩。此外,提出粗到细搜索策略(CFS),先分配剪枝比例再细化元素选择,进一步提升性能。在多个大模型家族上的大量实验表明,PBS²P 在困惑度和下游准确率上均持续优于当前最优的二值化后训练量化方法。代码与模型将开源于 https://github.com/XIANGLONGYAN/PBS2P。

原文摘要 · Abstract (English)

Large language models (LLMs) have achieved remarkable progress in natural language processing, but their high computational and memory costs hinder deployment on resource-constrained devices. Binarization represents the most extreme form of quantization, yet binarized models still contain redundancy that can be further removed. Pruning provides a natural way to eliminate such redundancy, but naïve combination with binarization often results in severe performance degradation. In this paper, we propose Progressive Binarization with Semi-Structured Pruning (PBS$^2$P), a novel post-training framework that seamlessly integrates binarization and semi-structured pruning. We first propose Stepwise semi-structured Pruning with Binarization Optimization (SPBO), which progressively introduces sparsity while optimizing binarization parameters to jointly reduce pruning and quantization error, yielding more stable and accurate compression. Additionally, we propose a Coarse-to-Fine Search (CFS) that first allocates pruning ratios and then refines element selection, further enhancing overall performance. Extensive experiments across multiple LLM families show that PBS$^2$P consistently outperforms state-of-the-art (SOTA) binary post-training quantization methods in both perplexity and downstream accuracy. The code and models will be available at https://github.com/XIANGLONGYAN/PBS2P.

大模型压缩二值化剪枝推理优化

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