Artificial Neural Network training using metaheuristics for medical data classification: An experimental study

被引:38
作者
Si, Tapas [1 ]
Bagchi, Jayri [1 ]
Miranda, Pericles B. C. [2 ]
机构
[1] Bankura Unnayani Inst Engn, Dept Comp Sci & Engn, Bankura 722146, W Bengal, India
[2] Univ Fed Rural Pernambuco UFRPE, Dept Comp DC, Rua Dom Manuel Medeiros S-N, BR-52171900 Recife, PE, Brazil
关键词
Artificial Neural Network; Metaheuristics; Medical data classification; Multi-criteria decision making; OPTIMIZER; ALGORITHMS;
D O I
10.1016/j.eswa.2021.116423
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The Artificial Neural Network (ANN) is an important machine learning tool used in medical data classification for disease diagnosis. The learning algorithm in ANN training plays a crucial role in classification performance. Various approaches have been successfully applied as a learning algorithm for ANN training. This paper performs an experimental study that investigates the performance of different metaheuristics as learning algorithms to train the ANN for medical data classification tasks. The experiments are carried out on 15 well-known medical datasets. A comparative study is conducted with the classical Levenberg-Marquardt (LM) and other thirteen recent and relevant metaheuristics. Different evaluation criteria such as accuracy, sensitivity, specificity, precision, Geometric Mean, F-Measure, false-positive rate (FPR) are considered for performance estimation. The classification results are analyzed using Multi-Criteria Decision Making (MCDM) method, and the results with analysis establish that the Equilibrium Optimizer algorithm outperforms all the other algorithms included in the comparative study.
引用
收藏
页数:22
相关论文
共 37 条
[1]   Optimizing connection weights in neural networks using the whale optimization algorithm [J].
Aljarah, Ibrahim ;
Faris, Hossam ;
Mirjalili, Seyedali .
SOFT COMPUTING, 2018, 22 (01) :1-15
[2]  
Asuncion A, 2007, UCI MACHINE LEARNING
[3]  
Bagchi J., 2020, INT C COMPUTATIONAL
[4]   Numerical optimization and feed-forward neural networks training using an improved optimization algorithm: multiple leader salp swarm algorithm [J].
Bairathi, Divya ;
Gopalani, Dinesh .
EVOLUTIONARY INTELLIGENCE, 2021, 14 (03) :1233-1249
[5]  
Bergstra J, 2012, J MACH LEARN RES, V13, P281
[6]  
Bhasin P, 2017, INT J ENG RES TECHNO
[7]   Artificial Bee Colony Training of Neural Networks [J].
Bullinaria, John A. ;
AlYahya, Khulood .
NATURE INSPIRED COOPERATIVE STRATEGIES FOR OPTIMIZATION (NICSO 2013), 2014, 512 :191-201
[8]  
Dutta RK., 2016, INT J COMPUT APPL, V137, P1
[9]   A trigonometric mutation operation to differential evolution [J].
Fan, HY ;
Lampinen, J .
JOURNAL OF GLOBAL OPTIMIZATION, 2003, 27 (01) :105-129
[10]   Marine Predators Algorithm: A nature-inspired metaheuristic [J].
Faramarzi, Afshin ;
Heidarinejad, Mohammad ;
Mirjalili, Seyedali ;
Gandomi, Amir H. .
EXPERT SYSTEMS WITH APPLICATIONS, 2020, 152