arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2305.18445cs.LGcs.CV
收藏

深度神经网络的智能梯度放大

Intelligent gradient amplification for deep neural networks

Georgia State University(佐治亚州立大学);Georgia Institute of Technology(佐治亚理工学院);Emory University(埃默里大学);Chinese Academy of Sciences(中国科学院);Shenzhen Institute of Advanced Technology(深圳先进技术研究院)

Sunitha Basodi, Krishna Pusuluri, Xueli Xiao, Yi Pan

AI总结:

本文提出一种智能梯度放大方法,通过分析训练中梯度波动选择放大层,以缓解梯度消失并加速训练,在CIFAR-10和CIFAR-100上分别提升约2.5%和4.5%的准确率。

摘要:

深度学习模型在各种任务和领域中相比其他机器学习技术提供了更优越的性能,但也带来了自身的挑战。特别是,随着模型深度的增加,深度学习模型需要更长的训练时间,并且会遭受梯度消失问题。若干解决方案独立地解决了这些问题,但很少有工作致力于确定一种集成解决方案,通过解决梯度消失问题来提升模型性能,同时加速训练过程,以在更大的学习率下实现更高的性能。在本工作中,我们使用一种分析训练期间各层梯度波动的形式化方法,智能地确定对深度学习模型的哪些层应用梯度放大。我们使用两种不同的智能度量方法和两种不同的阈值来确定放大层,并采用仅在特定训练轮次放大梯度的训练策略,对较简单和较深的神经网络进行了详细实验。结果表明,与原始模型相比,我们的放大方法提供了更好的性能,在CIFAR-10数据集上实现了约2.5%的准确率提升,在CIFAR-100数据集上实现了约4.5%的准确率提升,即使模型在更高的学习率下训练也是如此。

英文摘要:

Deep learning models offer superior performance compared to other machine learning techniques for a variety of tasks and domains, but pose their own challenges. In particular, deep learning models require larger training times as the depth of a model increases, and suffer from vanishing gradients. Several solutions address these problems independently, but there have been minimal efforts to identify an integrated solution that improves the performance of a model by addressing vanishing gradients, as well as accelerates the training process to achieve higher performance at larger learning rates. In this work, we intelligently determine which layers of a deep learning model to apply gradient amplification to, using a formulated approach that analyzes gradient fluctuations of layers during training. Detailed experiments are performed for simpler and deeper neural networks using two different intelligent measures and two different thresholds that determine the amplification layers, and a training strategy where gradients are amplified only during certain epochs. Results show that our amplification offers better performance compared to the original models, and achieves accuracy improvement of around 2.5% on CIFAR- 10 and around 4.5% on CIFAR-100 datasets, even when the models are trained with higher learning rates.

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

↑