通过深度权重分解实现平滑稀疏训练,突破传统方法瓶颈
Deep Weight Factorization: Sparse Learning Through the Lens of Artificial Symmetries
- 将权重分解为多个因子,用可微的L2正则替代L1非光滑惩罚
- 在多个数据集上优于浅层分解和主流剪枝方法
- 提出专属初始化与学习率策略,保障深层因子网络训练稳定
稀疏正则化在机器学习中已有广泛应用,但其在神经网络中的应用仍具挑战,主要因L1范数不连续,无法与随机梯度下降兼容。一种有前景的方法是浅层权重分解,将权重拆分为两个因子,通过在因子上添加可微的L2正则,实现对L1惩罚的平滑优化。本文提出深度权重分解,将该方法扩展至超过两个因子。理论上证明了深度分解等价于非凸稀疏正则化,并分析其对训练动态与优化的影响。针对标准训练方式的局限性,我们设计了定制化初始化方案,并确定了训练因子网络的关键学习率要求。实验在多种架构与数据集上验证了深度权重分解的有效性,其性能持续优于浅层分解及广泛使用的剪枝方法。
原文摘要 · Abstract (English)
Sparse regularization techniques are well-established in machine learning, yet their application in neural networks remains challenging due to the non-differentiability of penalties like the $L_1$ norm, which is incompatible with stochastic gradient descent. A promising alternative is shallow weight factorization, where weights are decomposed into two factors, allowing for smooth optimization of $L_1$-penalized neural networks by adding differentiable $L_2$ regularization to the factors. In this work, we introduce deep weight factorization, extending previous shallow approaches to more than two factors. We theoretically establish equivalence of our deep factorization with non-convex sparse regularization and analyze its impact on training dynamics and optimization. Due to the limitations posed by standard training practices, we propose a tailored initialization scheme and identify important learning rate requirements necessary for training factorized networks. We demonstrate the effectiveness of our deep weight factorization through experiments on various architectures and datasets, consistently outperforming its shallow counterpart and widely used pruning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。