arXiv:2504.05357cs.LGcs.AI2025-04ICLR被引 3

只靠参数符号就能让随机初始化网络表现媲美原模型。

Find A Winning Sign: Sign Is All We Need to Win the Lottery

  • 用参数符号替代复杂结构,传递泛化能力。
  • 新方法在多种架构和数据集上实现低误差路径连接。
  • 适合关注模型压缩与高效训练的研究者。

彩票理论假设存在一个稀疏子网络(即‘赢家票’),其在从零训练时的泛化能力可媲美原始过参数化网络。现有迭代剪枝(IP)方法通常依赖归一化层参数来保持泛化性能,但受限于特定初始化或小规模模型,或通过在训练权重上应用掩码绕过问题。本文发现,参数符号配置对泛化信息传递至关重要。通过线性模式连通性分析,我们发现保留参数符号和归一化层参数可维持训练后稀疏网络的吸引域。为减少对归一化层的依赖,我们设计方法使稀疏网络与初始化归一化参数的对应网络之间线性路径误差屏障更低。实验表明,任意随机初始化网络仅继承稀疏结构与参数符号即可在该路径上达到低误差,性能接近原模型。代码已开源。

原文摘要 · Abstract (English)

The Lottery Ticket Hypothesis (LTH) posits the existence of a sparse subnetwork (a.k.a. winning ticket) that can generalize comparably to its over-parameterized counterpart when trained from scratch. The common approach to finding a winning ticket is to preserve the original strong generalization through Iterative Pruning (IP) and transfer information useful for achieving the learned generalization by applying the resulting sparse mask to an untrained network. However, existing IP methods still struggle to generalize their observations beyond ad-hoc initialization and small-scale architectures or datasets, or they bypass these challenges by applying their mask to trained weights instead of initialized ones. In this paper, we demonstrate that the parameter sign configuration plays a crucial role in conveying useful information for generalization to any randomly initialized network. Through linear mode connectivity analysis, we observe that a sparse network trained by an existing IP method can retain its basin of attraction if its parameter signs and normalization layer parameters are preserved. To take a step closer to finding a winning ticket, we alleviate the reliance on normalization layer parameters by preventing high error barriers along the linear path between the sparse network trained by our method and its counterpart with initialized normalization layer parameters. Interestingly, across various architectures and datasets, we observe that any randomly initialized network can be optimized to exhibit low error barriers along the linear path to the sparse network trained by our method by inheriting its sparsity and parameter sign information, potentially achieving performance comparable to the original. The code is available at https://github.com/JungHunOh/AWS\_ICLR2025.git

模型剪枝彩票理论参数符号泛化能力

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