为边缘设备定制轻量级视觉Transformer,提升特定类别识别精度。
NuWa: Deriving Lightweight Class-Specific Vision Transformers for Edge Devices
- 通过自知识净化去除干扰类别的权重,实现精准压缩
- 无需重训练,推理速度更快,特定类别准确率超越原始模型
- 适合资源受限设备上的高效定制化部署,尤其适用于无人机等场景
视觉Transformer(ViT)常需压缩以部署于无人机、智能汽车等资源受限的边缘设备。然而,现有压缩方法忽略多数边缘应用仅需特定类别知识的事实,导致模型保留冗余信息,特定类别性能不佳。我们发现,仅更换校准数据集不足以解决此问题,因存在两类根本挑战:其一,存在干扰特定类别的权重,移除后可提升专精性能;其二,目标类别多样且资源受限,需大量定制模型,而现有方法耗时且计算开销大,难以扩展。本文提出NuWa,一种低成本方法,从基础ViT中高效衍生出针对特定类别的轻量级模型。NuWa通过自知识净化剔除类损害权重,并利用闭式优化快速生成紧凑模型。无需后修剪重训练,推导出的边缘模型在特定类别准确率上超越基础模型,且推理加速。实验证明,相比最先进无训练剪枝方法,NuWa在特定任务上最高提升29.00%准确率;相较最优有训练剪枝方法,剪枝速度提升33.69倍,成本降低高达99.83%,平均仅损失0.61%准确率。项目页面:https://github.com/CGCL-codes/NuWa。
原文摘要 · Abstract (English)
Vision Transformers (ViTs) often need to be compressed for deployment on resource-constrained edge devices like drones and smart vehicles. However, existing model compression methods ignore that many edge devices only require the knowledge of specific classes for their applications. As a result, the derived all-class ViTs retain redundant knowledge and perform suboptimally on these classes. We discovered that simply replacing the calibration dataset with class-specific data does not suffice to address this issue, as these methods face two fundamental limitations. First, they overlook the existence of class-detrimental weights, which interfere with specialization, while removing them can improve class-specific performance. Second, the diversity of target classes and resource constraints on edge devices demand numerous customized models. Existing methods are time-consuming and computationally expensive, thus unscalable. In this work, we present NuWa, a cost-efficient method that addresses these challenges by deriving small ViTs from base ViTs for edge devices with specific class requirements. NuWa performs self-knowledge purification to prune class-detrimental weights and efficiently derives compact ViTs through closed-form optimization. Without post-pruning retraining, the derived edge ViTs surpass the base ViT in class-specific accuracy and accelerate inference. Comprehensive experiments demonstrate that NuWa outperforms state-of-the-art training-free pruning methods on class-specific tasks by up to 29.00\% in accuracy. Compared with the best-performing training-dependent pruning method, NuWa achieves a 33.69x pruning speedup and reduces pruning cost by up to 99.83\%, with only a 0.61\% average accuracy loss. Project Page: https://github.com/CGCL-codes/NuWa.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。