科技与创新2021,Issue(22) :10-13.DOI:10.15913/j.cnki.kjycx.2021.22.005

基于VGG16网络的超参数调整策略的研究

张铠臻 李艳武 刘博 李杰 谢辉 张忠义
科技与创新2021,Issue(22) :10-13.DOI:10.15913/j.cnki.kjycx.2021.22.005

基于VGG16网络的超参数调整策略的研究

张铠臻 1李艳武 1刘博 1李杰 1谢辉 1张忠义1
扫码查看

作者信息

  • 1. 重庆三峡学院电子与信息工程学院,重庆 404100
  • 折叠

摘要

在利用卷积神经网络进行训练时,往往会因为超参数调整不适而导致训练的模型不佳.研究了学习率调整方法,学习率是神经网络的一种超参数.分别对VGG16网络利用了指数衰减、余弦变化非自适应调整学习率策略与Adagrad、RMSProp、Aadm自适应调整学习率策略,还调整了不同算法中的其他相关参数.将以上策略运用在VGG16网络中,在Cifar10数据集上得到了最终的训练损失和测试精度.针对这些数据,从收敛速度、模型稳定性、训练损失、测试精度的角度进行了分析.经过对参数的调整,最后得出了在VGG16上利用SGD优化器的余弦变化策略达到的测试精度最高,其测试精度达到了88.33%,同时收敛速度比自适应学习率策略更快.

关键词

卷积神经网络/学习率/测试精度/自适应

引用本文复制引用

基金项目

重庆市三峡库区地质环境监测与灾害预警重点实验室开放基金(YB2020C0102)

出版年

2021
科技与创新
中国计算机用户协会

科技与创新

ISSN:1008-0570
被引量6
参考文献量5
段落导航相关论文