Efficient adaptive learning rate for convolutional neural network based on quadratic interpolation egret swarm optimization algorithm
Convolutional neural network (CNN) has recently become popular for addressing multi-domain image classification. However, most existing methods frequently suffer from poor performance, especially in performance and convergence for various datasets. Herein, we have proposed an algorithm for multi-domain image classification by introducing a novel adaptive learning rate rule to the conventional CNN. Specifically, we adopt the CNN to extract rich feature representations. Given that the hyperparameters of the learning rate have a positive effect on the prediction error, the Egret Swarm Optimization Algorithm (ESOA) is introduced to update the learning rate, which can jump out of local extrema during exploration. Therefore, combined with quadratic interpolation, the objective function can be approximated by a polynomial, thereby improving its prediction accuracy. To verify the robustness of the proposed algorithm, we conducted comprehensive experiments in five domain public datasets to fulfil the task of image classification. Meanwhile, the highest accuracy rate of 97.15 % was obtained on the test set. The performances of our method on 24 benchmark functions (CEC2017 and CEC2022) are compared with Particle Swarm Optimization (PSO), Genetic Algorithm(GA), Whale Optimization Algorithm(WOA), Catch Fish Optimization Algorithm(CFOA), GOOSE Algorithm(GO) and ESOA. In two benchmark sets, the performance metric values of our algorithm rank no. 1, especially in all unimodal functions in contrast with other baseline algorithms.
Authors
- Xiaoyu Shi (ORCID: https://orcid.org/0009-0003-3696-4442)
- Peiyang Wei (ORCID: https://orcid.org/0009-0008-9632-8006)
- Mingsheng Shang (ORCID: https://orcid.org/0000-0002-7024-2270)
- Jiesan Zhou
Institutions
- Chongqing University of Posts and Telecommunications (CN)
- Chengdu University of Information Technology (CN)
- Chongqing Institute of Green and Intelligent Technology (CN)
- University of Chinese Academy of Sciences (CN)
Publication Details
- Journal
- Heliyon
- Published
- 2024-09-01
- DOI
- https://doi.org/10.1016/j.heliyon.2024.e37814
- Citations
- 18
- Primary Topic
- Machine Learning and ELM
- Type
- article
- Field-Weighted Citation Impact
- 6.25
Funders
- Chengdu University of Information Technology
- Education Department of Sichuan Province
- Sichuan University
- Chengdu University