-
题名基于动量项分离的深度学习优化算法
- 1
-
-
作者
文晨锐
杨歆豪
张嘉慧
张珂
-
机构
苏州大学机电工程学院
-
出处
《计算机仿真》
北大核心
2022年第2期337-342,共6页
-
基金
国家自然科学基金(61971297)。
-
文摘
针对神经网络处理参数更新的优化算法中出现的局部最优点振荡问题,改进带动量项的随机梯度下降算法,提出了一种动量项分离的优化算法。通过计算当前时刻目标函数的曲率半径,根据阈值适时分离动量项,从而缓解局部最优点振荡问题。实验表明,动量项分离的优化算法能够适用于不同的模型结构和不同数据集。相较于带动量项的随机梯度下降算法,具有更高的准确度,能够更快地稳定收敛。与同类一阶动量算法相比,其准确率上升明显,为深度神经网络的参数更新提供了一种新的有效的解决方案。
-
关键词
深度学习
曲率半径
动量项分离
图像识别
-
Keywords
Deep learning
Radius of curvature
Momentum separation
Image recognition
-
分类号
TP391
[自动化与计算机技术—计算机应用技术]
-