收录:
摘要:
Training of the convolution neural network (CNN) is a problem of global optimisation. This study proposed a hybrid modified particle swarm optimisation (MPSO) and conjugate gradient (CG) algorithm for efficient training of CNN. The training involves MPSO-CG to avoid trapping in local minima. Particularly, improvements in the MPSO by introducing a novel approach for control parameters, improved parameters updating criteria, a novel parameter in the velocity update equation, and fusion of the CG allows handling the issues in training CNN. In this study, the authors validate the proposed MPSO algorithm on three benchmark mathematical test functions and also compared with three different variants of the baseline particle swarm optimisation algorithm. Furthermore, the performance of the proposed MPSO-CG is also compared with other training algorithms focusing on the analysis of computational cost, convergence, and accuracy based on a standard problem specific to classification applications on CIFAR-10 dataset and face and skin detection dataset. © 2020 Institution of Engineering and Technology. All rights reserved.
关键词:
通讯作者信息:
电子邮件地址:
来源 :
IET Computer Vision
ISSN: 1751-9632
年份: 2020
期: 5
卷: 14
页码: 259-267
1 . 7 0 0
JCR@2022
ESI学科: COMPUTER SCIENCE;
ESI高被引阀值:132
归属院系: