Optimization of Deep Neural Networks Using a Micro Genetic Algorithm

被引:1
作者
Landa, Ricardo [1 ]
Tovias-Alanis, David [1 ]
Toscano, Gregorio [2 ]
机构
[1] CINVESTAV, Tamaulipas Campus, Cd Victoria 87130, Mexico
[2] Catholic Univ Amer, Dept Elect Engn & Comp Sci, Washington, DC 20064 USA
关键词
micro genetic algorithm; deep neural networks; pruning; model complexity; accuracy;
D O I
10.3390/ai5040127
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This work proposes the use of a micro genetic algorithm to optimize the architecture of fully connected layers in convolutional neural networks, with the aim of reducing model complexity without sacrificing performance. Our approach applies the paradigm of transfer learning, enabling training without the need for extensive datasets. A micro genetic algorithm requires fewer computational resources due to its reduced population size, while still preserving a substantial degree of the search capabilities found in algorithms with larger populations. By exploring different representations and objective functions, including classification accuracy, hidden neuron ratio, minimum redundancy, and maximum relevance for feature selection, eight algorithmic variants were developed, with six variants performing both hidden layers reduction and feature-selection tasks. Experimental results indicate that the proposed algorithm effectively reduces the architecture of the fully connected layers in the convolutional neural network. The variant achieving the best reduction used only 44% of the convolutional features in the input layer, and only 9.7% of neurons in the hidden layers, without negatively impacting (statistically confirmed) classification accuracy when compared to a network model based on a full reference architecture and a representative method from the literature.
引用
收藏
页码:2651 / 2679
页数:29
相关论文
共 49 条
[1]  
Aggarwal C., 2018, Neural Networks and Deep Learning, DOI 10.1007/978-3-319-94463-0
[2]  
[Anonymous], 2016, github user: Yiweichen04 Cataract Dataset
[3]  
[Anonymous], 2016, kaggle user: Nitesh Yadav Chessman Image Dataset
[4]  
[Anonymous], 2020, kaggle user: Pranav Raikote COVID-19 Image Dataset
[5]   An Optimized Neural Network Using Genetic Algorithm for Cardiovascular Disease Prediction [J].
Arroyo, Jan Carlo T. ;
Delima, Allemar Jhone P. .
JOURNAL OF ADVANCES IN INFORMATION TECHNOLOGY, 2022, 13 (01) :95-99
[6]  
Ba J, 2014, ACS SYM SER
[7]   Hybridizing Evolutionary Computation and Deep Neural Networks: An Approach to Handwriting Recognition Using Committees and Transfer Learning [J].
Baldominos, Alejandro ;
Saez, Yago ;
Isasi, Pedro .
COMPLEXITY, 2019, 2019
[8]   Towards Interpretable Deep Learning: A Feature Selection Framework for Prognostics and Health Management Using Deep Neural Networks [J].
Barraza, Joaquin Figueroa ;
Droguett, Enrique Lopez ;
Martins, Marcelo Ramos .
SENSORS, 2021, 21 (17)
[9]   Deep features optimization based on a transfer learning, genetic algorithm, and extreme learning machine for robust content-based image retrieval [J].
Bibi, Ruqia ;
Mehmood, Zahid ;
Munshi, Asmaa ;
Yousaf, Rehan Mehmood ;
Ahmed, Syed Sohail .
PLOS ONE, 2022, 17 (10)
[10]  
Collaboration I.S.I, 2019, Skin Cancer: Malignant vs. Benign