提出无需调参的高效结构化剪枝方法,一键完成模型压缩与优化。
HESSO: Towards Automatic Efficient and User Friendly Any Neural Network Training and Pruning
- 设计自适应混合优化器HESSO,自动实现结构化稀疏训练
- 在多个视觉与语言任务上达到或超越现有最优性能
- 引入纠错循环机制,防止剪枝后性能崩溃,适合各类网络
结构化剪枝是将大型深度神经网络(DNN)压缩为高性能小模型的主流方法。现有方法多依赖多阶段流程,需大量工程投入和人工经验。最近提出的“仅训练一次”(OTO)系列通过自动完成搜索空间生成、结构化稀疏优化和子网络构建来简化流程。然而,OTO系列内置的稀疏优化器(如半空间投影梯度族)仍需超参数调优,并隐含控制稀疏探索,仍需人工干预。为此,本文提出混合高效结构化稀疏优化器(HESSO),可自动高效训练DNN并生成高性能子网络,几乎无需调参,且易于集成到通用训练框架中。针对剪枝中常见的不可逆性能下降问题,进一步提出纠正冗余识别循环(CRIC),可靠识别关键结构。数值实验表明,HESSO及其增强版HESSO-CRIC在计算机视觉、自然语言处理及大语言模型等多种应用中均表现优异,性能媲美甚至超越现有最优方法,支持多数DNN架构,且能有效避免性能崩溃。
原文摘要 · Abstract (English)
Structured pruning is one of the most popular approaches to effectively compress the heavy deep neural networks (DNNs) into compact sub-networks while retaining performance. The existing methods suffer from multi-stage procedures along with significant engineering efforts and human expertise. The Only-Train-Once (OTO) series has been recently proposed to resolve the many pain points by streamlining the workflow by automatically conducting (i) search space generation, (ii) structured sparse optimization, and (iii) sub-network construction. However, the built-in sparse optimizers in the OTO series, i.e., the Half-Space Projected Gradient (HSPG) family, have limitations that require hyper-parameter tuning and the implicit controls of the sparsity exploration, consequently requires intervening by human expertise. To address such limitations, we propose a Hybrid Efficient Structured Sparse Optimizer (HESSO). HESSO could automatically and efficiently train a DNN to produce a high-performing subnetwork. Meanwhile, it is almost tuning-free and enjoys user-friendly integration for generic training applications. To address another common issue of irreversible performance collapse observed in pruning DNNs, we further propose a Corrective Redundant Identification Cycle (CRIC) for reliably identifying indispensable structures. We numerically demonstrate the efficacy of HESSO and its enhanced version HESSO-CRIC on a variety of applications ranging from computer vision to natural language processing, including large language model. The numerical results showcase that HESSO can achieve competitive even superior performance to varying state-of-the-arts and support most DNN architectures. Meanwhile, CRIC can effectively prevent the irreversible performance collapse and further enhance the performance of HESSO on certain applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。