-
题名用于图像分类的模糊策略学习率ResNet
被引量:1
- 1
-
-
作者
张睿权
覃华
-
机构
广西大学计算机与电子信息学院
-
出处
《计算机工程与设计》
北大核心
2023年第8期2305-2311,共7页
-
基金
国家自然科学基金项目(51667004、61762009)。
-
文摘
ResNet深度神经网络用于图像分类时,全连接层训练算法收敛性差降低了分类效果。针对此不足,提出一种模糊策略梯度算法训练ResNet。推导出ResNet全连接层权重的迭代公式,用历史梯度信息修正当前一阶小批量梯度,用模糊策略学习率更新权重,通过上下边界函数处理学习率的过大或过小而引发的迭代振荡,改善训练算法收敛性。在CINIC-10和CIFAR-100数据集上的实验结果表明,所提算法训练的ResNet分类效果优于相比较算法。特别是在综合性分类指标Kappa系数上,所提算法训练的ResNet较最新的AdaBound算法平均提高了9.29%,改进效果显著。
-
关键词
图像分类
全连接层
训练算法收敛性
深度神经网络
小批量梯度
模糊策略学习率
上下边界函数
-
Keywords
image classification
full connection layer
convergence of training algorithms
deep neural network
mini-batch gradient
fuzzy policy learning rate
upper and lower boundary functions
-
分类号
TP301.6
[自动化与计算机技术—计算机系统结构]
-