提出PISA算法,用预条件加速深度模型训练,抗数据异构性。
Preconditioned Inexact Stochastic ADMM for Deep Model
- 基于ADMM框架设计预条件随机优化,放宽收敛前提
- 在多类模型上优于现有最优优化器,支持并行计算
- 适合分布式训练或数据异构场景,对大模型训练友好
深度学习模型通常使用基于随机梯度下降的优化算法训练,但这类优化器存在收敛慢、收敛假设严格等固有局限。尤其在分布式设置下因数据异构带来的挑战,严重影响其理论与数值性能。本文提出一种新算法PISA(Preconditioned Inexact Stochastic ADMM),在仅需梯度在有界区域上满足Lipschitz连续性的条件下实现收敛,无需其他常见假设,有效应对数据异构问题。算法架构支持可扩展并行计算,并可融合二阶信息、二阶矩及通过Newton-Schulz迭代实现的正交化动量等预条件策略。引入后两种预条件得到两个高效变体SISA和NSISA。在视觉模型、大语言模型、强化学习模型、生成对抗网络和循环神经网络等多种深度模型的训练与微调任务中,实验表明SISA和NSISA在数值性能上显著优于多种前沿优化器。
原文摘要 · Abstract (English)
Deep learning models are usually trained with stochastic gradient descent-based algorithms, but these optimizers face inherent limitations, such as slow convergence and stringent assumptions for convergence. In particular, data heterogeneity arising from distributed settings poses significant challenges to their theoretical and numerical performance. This paper develops an algorithm, PISA (Preconditioned Inexact Stochastic Alternating Direction Method of Multipliers). Grounded in rigorous theoretical guarantees, the algorithm converges under the sole assumption of Lipschitz continuity of the gradient on a bounded region, thereby removing the need for other conditions commonly imposed by stochastic methods. This capability enables the proposed algorithm to tackle the challenge of data heterogeneity effectively. Moreover, the algorithmic architecture enables scalable parallel computing and supports various preconditions, such as second-order information, second moment, and orthogonalized momentum by Newton-Schulz iterations. Incorporating the latter two preconditions in PISA yields two computationally efficient variants: SISA and NSISA. Comprehensive experimental evaluations for training or fine-tuning diverse deep models, including vision models, large language models, reinforcement learning models, generative adversarial networks, and recurrent neural networks, demonstrate superior numerical performance of SISA and NSISA compared to various state-of-the-art optimizers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。