论文精选73°

主动学习与剪枝结合实现高效模型压缩

One Loop, Two Gains: Can Active Learning win the Lottery for Free?

精选理由

这篇论文展示了如何将主动学习和模型剪枝结合,无需额外计算成本就能获得稀疏模型,解决了主动学习在大规模架构上的计算瓶颈问题。

研究人员提出Improve & Prune (I&P)方法,将迭代幅度剪枝整合到主动学习的重训练循环中。该方法在多个获取函数、架构族和图像分类数据集上进行了测试,包括主动微调场景。实验结果表明,I&P在每个主动学习迭代中都能产生稀疏、可部署的模型,在95%稀疏度下仍能保持与密集模型相当的准确性。

原文 · arXiv cs.LG

One Loop, Two Gains: Can Active Learning win the Lottery for Free?

The lottery ticket hypothesis posits the existence of winning tickets: sparse subnetworks that, when trained in isolation from their original initialization, match the accuracy of the full dense network. The predominant method for discovering such tickets, iterative magnitude pruning, alternates pruning with full retraining from scratch until convergence over many cycles. Similarly, deep active learning also retrains a model from scratch after each acquisition round as new labels become available. Despite this shared reliance on iterative retraining with a substantial computational overhead, the two paradigms have been studied separately. We observe that the iterative training loop inherent to pool-based active learning already provides the exact computational structure that iterative magnitude pruning exploits, and propose Improve & Prune (I&P), a method that integrates magnitude pruning into each active learning retraining cycle at practically no additional cost. This raises a key empirical question: can iterative magnitude pruning produce winning tickets under the non-stationary data regime of active learning? We investigate this question across multiple acquisition functions, architecture families, and image classification datasets, including an active fine-tuning scenario. Our results demonstrate that I&P yields sparse, deployable models at each active learning iteration. Those match the accuracy of their dense counterparts at sparsities up to 95%, effectively obtaining winning tickets as a byproduct of the active learning pipeline. These per-iteration sparse models can address two computational bottlenecks - per-round model retraining and acquisition scoring over the unlabeled pool - that currently prevent the practical adoption of DAL on large architectures and large unlabeled pools.