arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.03573cs.LGcs.AIcs.CV
收藏

平衡即本质:通过自适应梯度校正加速稀疏训练

Balance is Essence: Accelerating Sparse Training via Adaptive Gradient Correction

North Carolina State University(北卡罗来纳州立大学);Purdue University(普渡大学);Texas A&M University(德克萨斯农工大学)

Bowen Lei, Dongkuan Xu, Ruqi Zhang, Shuren He, Bani K. Mallick

AI总结:

本文提出自适应梯度校正方法,通过平衡当前与先前梯度来加速并稳定稀疏训练,在相同轮数下准确率最高提升5.0%,达到相同准确率最多减少52.1%的训练轮数。

摘要:

尽管深度神经网络性能出色,但它们需要大量的内存和计算成本,这限制了其在资源受限场景中的应用。稀疏训练是降低这些成本的最常用技术之一,然而,稀疏性约束增加了优化的难度,导致训练时间增加和不稳定。在这项工作中,我们旨在克服这一问题并实现时空协同高效。为了加速并稳定稀疏训练的收敛,我们分析了梯度变化并开发了一种自适应梯度校正方法。具体来说,我们近似当前梯度与先前梯度之间的相关性,并利用该相关性来平衡这两个梯度,从而获得校正后的梯度。我们的方法可用于标准和对抗设置下最流行的稀疏训练流程。理论上,我们证明了我们的方法能够加速稀疏训练的收敛速度。在多个数据集、模型架构和稀疏度上的大量实验表明,在相同训练轮数下,我们的方法在准确率上比领先的稀疏训练方法最多高出5.0%,并且在达到相同准确率时将训练轮数最多减少了52.1%。我们的代码可在以下网址获取:https://github.com/StevenBoys/AGENT。

英文摘要:

Despite impressive performance, deep neural networks require significant memory and computation costs, prohibiting their application in resource-constrained scenarios. Sparse training is one of the most common techniques to reduce these costs, however, the sparsity constraints add difficulty to the optimization, resulting in an increase in training time and instability. In this work, we aim to overcome this problem and achieve space-time co-efficiency. To accelerate and stabilize the convergence of sparse training, we analyze the gradient changes and develop an adaptive gradient correction method. Specifically, we approximate the correlation between the current and previous gradients, which is used to balance the two gradients to obtain a corrected gradient. Our method can be used with the most popular sparse training pipelines under both standard and adversarial setups. Theoretically, we prove that our method can accelerate the convergence rate of sparse training. Extensive experiments on multiple datasets, model architectures, and sparsities demonstrate that our method outperforms leading sparse training methods by up to \textbf{5.0\%} in accuracy given the same number of training epochs, and reduces the number of training epochs by up to \textbf{52.1\%} to achieve the same accuracy. Our code is available on: \url{https://github.com/StevenBoys/AGENT}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑