用进化算法动态优化大模型压缩,提升精度并打破传统假设
EvoPress: Accurate Dynamic Model Compression via Evolutionary Search
- 通过进化搜索自动寻找每层最优压缩策略,不依赖独立性假设
- 在Llama/Mistral/Phi上实现当前最佳动态压缩效果,性能超越现有方法
- 支持结构化剪枝、无结构稀疏和动态位宽量化,通用性强
大型语言模型(LLM)的高计算开销催生了大量压缩研究,如量化、稀疏化或结构化剪枝。近期前沿方向是动态非均匀压缩方法,即按块或逐层调整压缩程度(如稀疏度),以最小化精度损失并满足全局压缩阈值。然而,现有方法依赖于对各层重要性的估计,隐含假设各层贡献独立。我们观察到这一独立性假设在实际中通常不成立:进一步剪枝甚至可能显著恢复性能。为此,我们提出EvoPress,一种新型进化框架用于动态LLM压缩。通过将动态压缩建模为通用优化问题,EvoPress高效识别最优压缩配置,并适用于多种模型与压缩技术。利用EvoPress,我们在Llama、Mistral和Phi模型上实现了动态压缩的最先进性能,为结构化剪枝(块/层删除)、无结构稀疏性和动态位宽量化设立了新基准。代码已开源。
原文摘要 · Abstract (English)
The high computational costs of large language models (LLMs) have led to a flurry of research on LLM compression, via methods such as quantization, sparsification, or structured pruning. A new frontier in this area is given by dynamic, non-uniform compression methods, which adjust the compression levels (e.g., sparsity) per-block or even per-layer in order to minimize accuracy loss, while guaranteeing a global compression threshold. Yet, current methods rely on estimating the importance of a given layer, implicitly assuming that layers contribute independently to the overall compression error. We begin from the motivating observation that this independence assumption does not generally hold for LLM compression: pruning a model further may even significantly recover performance. To address this, we propose EvoPress, a novel evolutionary framework for dynamic LLM compression. By formulating dynamic compression as a general optimization problem, EvoPress identifies optimal compression profiles in a highly efficient manner, and generalizes across diverse models and compression techniques. Via EvoPress, we achieve state-of-the-art performance for dynamic compression of Llama, Mistral, and Phi models, setting new benchmarks for structural pruning (block/layer dropping), unstructured sparsity, and quantization with dynamic bitwidths. Our code is available at https://github.com/IST-DASLab/EvoPress}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。