An Improved Reacceleration Optimization Algorithm Based on the Momentum Method for Image Recognition

被引:1
|
作者
Sun, Haijing [1 ]
Cai, Ying [2 ]
Tao, Ran [3 ]
Shao, Yichuan [1 ]
Xing, Lei [4 ]
Zhang, Can [2 ]
Zhao, Qian [5 ]
机构
[1] Shenyang Univ, Sch Intelligent Sci & Engn, Shenyang 110044, Peoples R China
[2] Shenyang Univ, Sch Informat Engn, Shenyang 110044, Peoples R China
[3] Shanghai Maruka Comp Informat Technol Co Ltd, Shanghai 200052, Peoples R China
[4] Univ Surrey, Sch Chem & Chem Engn, Guildford GU2 7XH, Surrey, England
[5] Shenyang Univ Technol, Sch Sci, Shenyang 110044, Peoples R China
关键词
momentum acceleration; optimization algorithm; deep learning; image recognition; gradient descent algorithm;
D O I
10.3390/math12111759
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
The optimization algorithm plays a crucial role in image recognition by neural networks. However, it is challenging to accelerate the model's convergence and maintain high precision. As a commonly used stochastic gradient descent optimization algorithm, the momentum method requires many epochs to find the optimal parameters during model training. The velocity of its gradient descent depends solely on the historical gradients and is not subject to random fluctuations. To address this issue, an optimization algorithm to enhance the gradient descent velocity, i.e., the momentum reacceleration gradient descent (MRGD), is proposed. The algorithm utilizes the point division of the current momentum and the gradient relationship, multiplying it with the gradient. It can adjust the update rate and step size of the parameters based on the gradient descent state, so as to achieve faster convergence and higher precision in training the deep learning model. The effectiveness of this method is further proven by applying the reacceleration mechanism to the Adam optimizer, resulting in the MRGDAdam algorithm. We verify both algorithms using multiple image classification datasets, and the experimental results show that the proposed optimization algorithm enables the model to achieve higher recognition accuracy over a small number of training epochs, as well as speeding up model implementation. This study provides new ideas and expansions for future optimizer research.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Smoke Image Recognition Method Based on the optimization of SVM parameters with Improved Fruit Fly Algorithm
    Liu, Jingwen
    Tan, Junshan
    Qin, Jiaohua
    Xiang, Xuyu
    [J]. KSII TRANSACTIONS ON INTERNET AND INFORMATION SYSTEMS, 2020, 14 (08): : 3534 - 3549
  • [2] Optimization of Image Segmentation Method Based on Improved Wavelet Algorithm
    Li, Sun
    [J]. PROCEEDINGS OF THE 2015 INTERNATIONAL CONFERENCE ON AUTOMATION, MECHANICAL CONTROL AND COMPUTATIONAL ENGINEERING, 2015, 124 : 1019 - 1022
  • [3] Image Retrieval Method based on An Improved Particle Swarm Optimization Algorithm
    Hu, Guanghao
    Yang, Fei
    [J]. 2015 INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION, BIG DATA AND SMART CITY (ICITBS), 2016, : 466 - 469
  • [4] Method of Image Recognition for Lepidopteran Insects Based on Improved Differential Evolution Algorithm
    Lin D.
    Huang S.
    Zhang F.
    Liang G.
    Wu S.
    Hu X.
    Wang R.
    [J]. Linye Kexue/Scientia Silvae Sinicae, 2020, 56 (03): : 73 - 81
  • [5] An image enhancement method based on improved teaching-learning-based optimization algorithm
    Bi X.
    Pan T.
    [J]. Bi, Xiaojun (zl12306124@163.com), 1716, Editorial Board of Journal of Harbin Engineering (37): : 1716 - 1721
  • [6] Color Image Edge Detection Method Based on the Improved Whale Optimization Algorithm
    Liu, Dujin
    Zhou, Shiji
    Shen, Rong
    Luo, Xuegang
    [J]. IEEE ACCESS, 2023, 11 : 5981 - 5989
  • [7] Image Noise Recognition Algorithm Based on Improved DenseNet
    Huang, Mengxing
    Zeng, Lirong
    Zhang, Yu
    Li, Yuchun
    Ni, Zehao
    Wu, Di
    Feng, Siling
    [J]. WEB INFORMATION SYSTEMS AND APPLICATIONS (WISA 2021), 2021, 12999 : 455 - 467
  • [8] Image Enhancement Based on Improved Antlion Optimization Algorithm
    Guo, Wen-yan
    Zhang, Xuan
    Liu, Ke-xin
    Zhang, Jiao-jiao
    [J]. 2018 INTERNATIONAL CONFERENCE ON COMMUNICATION, NETWORK AND ARTIFICIAL INTELLIGENCE (CNAI 2018), 2018, : 418 - 423
  • [9] A fast SAR image segmentation method based on improved chicken swarm optimization algorithm
    Liang, Jianhui
    Wang, Lifang
    Ma, Miao
    Zhang, Jian
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2018, 77 (24) : 31787 - 31805
  • [10] A fast SAR image segmentation method based on improved chicken swarm optimization algorithm
    Jianhui Liang
    Lifang Wang
    Miao Ma
    Jian Zhang
    [J]. Multimedia Tools and Applications, 2018, 77 : 31787 - 31805