arXiv:2412.10869cs.LGcs.AI2024-12AAAI被引 10

TinySubNets通过剪枝与量化提升模型容量利用率,实现高效持续学习。

TinySubNets: An efficient and low capacity continual learning strategy

  • 采用多级剪枝与自适应量化,动态分配权重资源。
  • 在多个基准数据集上准确率超越现有最优方法。
  • 适合资源受限场景下的持续学习任务,尤其关注容量效率。

持续学习(CL)是当前机器学习研究中的重要方向。现有架构类和混合类方法虽能动态调整模型结构,但常因未有效利用模型稀疏性,导致容量饱和,限制可学习任务数量。本文提出一种新型架构式持续学习策略 TinySubNets(TSN),结合多级剪枝、自适应量化与权重共享机制。剪枝保留关键权重,释放冗余参数供后续任务使用;自适应量化将单个权重拆分为多份,按需分配至不同任务;任务间权重共享提升容量利用率并促进知识迁移。实验表明,TSN 在多个常见基准数据集和场景下,相较现有最先进方法显著提升准确率,并大幅优化了模型容量利用率。代码已开源:https://github.com/lifelonglab/tinysubnets。

原文摘要 · Abstract (English)

Continual Learning (CL) is a highly relevant setting gaining traction in recent machine learning research. Among CL works, architectural and hybrid strategies are particularly effective due to their potential to adapt the model architecture as new tasks are presented. However, many existing solutions do not efficiently exploit model sparsity, and are prone to capacity saturation due to their inefficient use of available weights, which limits the number of learnable tasks. In this paper, we propose TinySubNets (TSN), a novel architectural CL strategy that addresses the issues through the unique combination of pruning with different sparsity levels, adaptive quantization, and weight sharing. Pruning identifies a subset of weights that preserve model performance, making less relevant weights available for future tasks. Adaptive quantization allows a single weight to be separated into multiple parts which can be assigned to different tasks. Weight sharing between tasks boosts the exploitation of capacity and task similarity, allowing for the identification of a better trade-off between model accuracy and capacity. These features allow TSN to efficiently leverage the available capacity, enhance knowledge transfer, and reduce computational resource consumption. Experimental results involving common benchmark CL datasets and scenarios show that our proposed strategy achieves better results in terms of accuracy than existing state-of-the-art CL strategies. Moreover, our strategy is shown to provide a significantly improved model capacity exploitation. Code released at: https://github.com/lifelonglab/tinysubnets.

持续学习模型压缩剪枝量化

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