中文版 | English
CSpace
Efficient adaptive learning rate for convolutional neural network based on quadratic interpolation egret swarm optimization algorithm
Wei, Peiyang1,2,3,4; Shang, Mingsheng3,4; Zhou, Jiesan2; Shi, Xiaoyu3,4
2024-09-30
摘要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.
关键词Convolutional neural network Multi-domain image classification Adaptive learning rate Egret swarm optimization algorithm Quadratic interpolation
DOI10.1016/j.heliyon.2024.e37814
发表期刊HELIYON
卷号10期号:18页码:17
通讯作者Wei, Peiyang(weipy@cuit.edu.cn)
收录类别SCI
WOS记录号WOS:001317767700001
语种英语