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 条
  • [41] An Improved SAR Image Target Recognition Algorithm
    Kong, Yunbo
    Yang, Fan
    Zhou, Hai
    Zhang, Fan
    Fu, Haojun
    [J]. 3RD ANNUAL INTERNATIONAL CONFERENCE ON INFORMATION SYSTEM AND ARTIFICIAL INTELLIGENCE (ISAI2018), 2018, 1069
  • [42] New Improved Optimized Method for Medical Image Enhancement Based on Modified Shark Smell Optimization Algorithm
    Zhou, Yuebin
    Ye, Jianmin
    Du, Yujin
    Sheykhahmad, Fatima Rashid
    [J]. SENSING AND IMAGING, 2020, 21 (01):
  • [43] A fuzzy image clustering method based on an improved backtracking search optimization algorithm with an inertia weight parameter
    Toz, Guliz
    Yucedag, Ibrahim
    Erdogmus, Pakize
    [J]. JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2019, 31 (03) : 295 - 303
  • [44] New Improved Optimized Method for Medical Image Enhancement Based on Modified Shark Smell Optimization Algorithm
    Yuebin Zhou
    Jianmin Ye
    Yujin Du
    Fatima Rashid Sheykhahmad
    [J]. Sensing and Imaging, 2020, 21
  • [45] Fast Image Registration Method Based on Improved AKAZE Algorithm
    Zhao Weidong
    Liu Junde
    Wang Manman
    Li Dan
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (06)
  • [46] Improved method on image stitching based on optical flow algorithm
    Lee, Hyunchul
    Lee, Sungmin
    Choi, Okkyung
    [J]. INTERNATIONAL JOURNAL OF ENGINEERING BUSINESS MANAGEMENT, 2020, 12
  • [47] A Novel Image Restore Method Based on Improved Genetic Algorithm
    Chen Wenjie
    Dou Lihua
    [J]. PROCEEDINGS OF THE 29TH CHINESE CONTROL CONFERENCE, 2010, : 3081 - 3086
  • [48] A Gastrointestinal Image Classification Method Based on Improved Adam Algorithm
    Sun, Haijing
    Cui, Jiaqi
    Shao, Yichuan
    Yang, Jiapeng
    Xing, Lei
    Zhao, Qian
    Zhang, Le
    [J]. MATHEMATICS, 2024, 12 (16)
  • [49] A Method of Image Segmentation Based on Improved Adaptive Genetic Algorithm
    Yu, Wenjiao
    Huang, Mengxing
    Zhu, Donghai
    Li, Xuegang
    [J]. FOUNDATIONS OF INTELLIGENT SYSTEMS (ISKE 2011), 2011, 122 : 507 - 516
  • [50] Image denoising method based on improved wavelet threshold algorithm
    Zhu, Guowu
    Liu, Bingyou
    Yang, Pan
    Fan, Xuan
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (26) : 67997 - 68011