分块交替更新+可学习步长,提升训练稳定性和效率
Neural Network Training via Stochastic Alternating Minimization with Trainable Step Sizes
- 将网络参数按层分块,交替优化以降低计算开销
- 在多个基准上用更少更新次数实现更好泛化性能
- 支持多种可学习步长,适合追求高效训练的研究者
深度神经网络训练本质上是非凸优化问题,传统方法如随机梯度下降(SGD)需同时更新所有参数,常导致收敛不稳定且计算成本高。为此,我们提出一种新方法——可学习步长的随机交替最小化(SAMT),将每层权重视为一个块,分块交替更新。该策略将整体优化分解为多个子问题,降低每步计算开销,并增强非凸环境下的训练稳定性。受元学习启发,我们在子问题求解中引入新型自适应步长机制,支持标量、逐元素、逐行、逐列等多种可学习步长形式,通过元学习实现各块的自适应步长选择。我们还提供了算法的理论收敛保证,证明其优化合理性。大量实验表明,SAMT在多个基准上以更少参数更新次数达到更优泛化性能,验证了其在神经网络优化中的有效性与潜力。
原文摘要 · Abstract (English)
The training of deep neural networks is inherently a nonconvex optimization problem, yet standard approaches such as stochastic gradient descent (SGD) require simultaneous updates to all parameters, often leading to unstable convergence and high computational cost. To address these issues, we propose a novel method, Stochastic Alternating Minimization with Trainable Step Sizes (SAMT), which updates network parameters in an alternating manner by treating the weights of each layer as a block. By decomposing the overall optimization into sub-problems corresponding to different blocks, this block-wise alternating strategy reduces per-step computational overhead and enhances training stability in nonconvex settings. To fully leverage these benefits, inspired by meta-learning, we proposed a novel adaptive step size strategy to incorporate into the sub-problem solving steps of alternating updates. It supports different types of trainable step sizes, including but not limited to scalar, element-wise, row-wise, and column-wise, enabling adaptive step size selection tailored to each block via meta-learning. We further provide a theoretical convergence guarantee for the proposed algorithm, establishing its optimization soundness. Extensive experiments for multiple benchmarks demonstrate that SAMT achieves better generalization performance with fewer parameter updates compared to state-of-the-art methods, highlighting its effectiveness and potential in neural network optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。