arXiv:2512.07142cs.LGcs.AI2025-12

提出新方法在初始化时高效找到高精度稀疏子网络。

Winning the Lottery by Preserving Network Training Dynamics with Concrete Ticket Search

  • 将子网络搜索建模为整体优化问题,用连续松弛和梯度平衡控制稀疏度。
  • 在CIFAR10上99.3%稀疏度下准确率达74.0%,仅需7.9分钟,远快于传统方法。
  • 适合追求高效稀疏化、关注训练初期动态的模型压缩研究者。

彩票券假说指出,在密集随机初始化的神经网络中存在高度稀疏且可训练的子网络(‘赢家票券’)。然而,当前最先进的抽票方法如彩票券重置(LTR)计算成本过高,而更高效的基于显著性的初始化剪枝(PaI)技术则存在显著的准确率-稀疏度权衡,且无法通过基本合理性检验。本文认为,PaI依赖的一阶显著性指标忽略了权重间依赖关系,是导致性能差距的关键原因。为此,我们提出基于连续松弛的子网络搜索(CTS),将子网络发现建模为整体组合优化问题,并引入新型梯度平衡方案(GRADBALANCE)控制稀疏度,无需敏感超参数调优即可高效识别初始化附近高性能子网络。受近期彩票券训练动态研究启发,我们进一步提出基于知识蒸馏的剪枝目标,发现最小化稀疏与密集网络输出间的反向KL散度(CTS-KL)尤为有效。在多种图像分类任务上的实验表明,CTS生成的子网络能稳健通过合理性检验,准确率可媲美或超越LTR,同时仅需极小计算量。例如,在ResNet-20/CIFAR10上,99.3%稀疏度下达到74.0%准确率,耗时仅7.9分钟,而LTR在相同稀疏度下准确率为68.3%,耗时95.2分钟。在所有稀疏度下,CTS均优于基于显著性的方法,尤其在高度稀疏场景下优势更显著。

原文摘要 · Abstract (English)

The Lottery Ticket Hypothesis asserts the existence of highly sparse, trainable subnetworks ('winning tickets') within dense, randomly initialized neural networks. However, state-of-the-art methods of drawing these tickets, like Lottery Ticket Rewinding (LTR), are computationally prohibitive, while more efficient saliency-based Pruning-at-Initialization (PaI) techniques suffer from a significant accuracy-sparsity trade-off and fail basic sanity checks. In this work, we argue that PaI's reliance on first-order saliency metrics, which ignore inter-weight dependencies, contributes substantially to this performance gap, especially in the sparse regime. To address this, we introduce Concrete Ticket Search (CTS), an algorithm that frames subnetwork discovery as a holistic combinatorial optimization problem. By leveraging a Concrete relaxation of the discrete search space and a novel gradient balancing scheme (GRADBALANCE) to control sparsity, CTS efficiently identifies high-performing subnetworks near initialization without requiring sensitive hyperparameter tuning. Motivated by recent works on lottery ticket training dynamics, we further propose a knowledge distillation-inspired family of pruning objectives, finding that minimizing the reverse Kullback-Leibler divergence between sparse and dense network outputs (CTS-KL) is particularly effective. Experiments on varying image classification tasks show that CTS produces subnetworks that robustly pass sanity checks and achieve accuracy comparable to or exceeding LTR, while requiring only a small fraction of the computation. For example, on ResNet-20 on CIFAR10, it reaches 99.3% sparsity with 74.0% accuracy in 7.9 minutes, while LTR attains the same sparsity with 68.3% accuracy in 95.2 minutes. CTS's subnetworks outperform saliency-based methods across all sparsities, but its advantage over LTR is most pronounced in the highly sparse regime.

子网络搜索稀疏化模型压缩

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