Optimizing a Deep Residual Neural Network with Genetic Algorithm for Acute Lymphoblastic Leukemia Classification

被引:25
作者
Rodrigues, Larissa Ferreira [1 ]
Backes, Andre Ricardo [1 ]
Nassif Travencolo, Bruno Augusto [1 ]
Barbosa de Oliveira, Gina Maira [1 ]
机构
[1] Fed Univ Uberlandia UFU, Fac Comp FACOM, Uberlandia, MG, Brazil
关键词
Leukemia classification; Convolutional neural networks; Genetic algorithm; Hyperparameter optimization; Fine-tuning; DIAGNOSIS; FUTURE; SEARCH;
D O I
10.1007/s10278-022-00600-3
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Acute lymphoblastic leukemia (ALL) is the most common childhood cancer worldwide, and it is characterized by the production of immature malignant cells in the bone marrow. Computer vision techniques provide automated analysis that can help specialists diagnose this disease. Microscopy image analysis is the most economical method for the initial screening of patients with ALL, but this task is subjective and time-consuming. In this study, we propose a hybrid model using a genetic algorithm (GA) and a residual convolutional neural network (CNN), ResNet-50V2, to predict ALL using microscopy images available in ALL-IDB dataset. However, accurate prediction requires suitable hyperparameters setup, and tuning these values manually still poses challenges. Hence, this paper uses GA to find the best hyperparameters that lead to the highest accuracy rate in the models. Also, we compare the performance of GA hyperparameter optimization with Random Search and Bayesian optimization methods. The results show that GA optimization improves the accuracy of the classifier, obtaining 98.46% in terms of accuracy. Additionally, our approach sheds new perspectives on identifying leukemia based on computer vision strategies, which could be an alternative for applications in a real-world scenario.
引用
收藏
页码:623 / 637
页数:15
相关论文
共 69 条
[1]  
[Anonymous], 1982, Pattern recognition: A statistical approach
[2]  
[Anonymous], 2018, P WORKSH VIS COMP IL
[3]   A convolutional neural network-based learning approach to acute lymphoblastic leukaemia detection with automated feature extraction [J].
Anwar, Shamama ;
Alam, Afrin .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2020, 58 (12) :3113-3121
[4]  
Backes AR, 2020, INT CONF SYST SIGNAL, P290, DOI [10.1109/IWSSIP48289.2020.9145325, 10.1109/iwssip48289.2020.9145325]
[5]  
Bengio Yoshua, 2012, Neural Networks: Tricks of the Trade. Second Edition: LNCS 7700, P437, DOI 10.1007/978-3-642-35289-8_26
[6]  
Bergstra J., 2011, Adv. Neural Inf. Process. Syst., P2546
[7]  
Bergstra J, 2012, J MACH LEARN RES, V13, P281
[8]  
Bhattacharjee R, 2015, 2015 IEEE POWER, COMMUNICATION AND INFORMATION TECHNOLOGY CONFERENCE (PCITC-2015), P657, DOI 10.1109/PCITC.2015.7438079
[9]  
Chollet F., 2015, Keras
[10]  
Claro M, 2020, INT CONF SYST SIGNAL, P63, DOI [10.1109/iwssip48289.2020.9145406, 10.1109/IWSSIP48289.2020.9145406]