Design and Compression Study for Convolutional Neural Networks Based on Evolutionary Optimization for Thoracic X-Ray Image Classification

被引:8
作者
Louati, Hassen [1 ]
Louati, Ali [2 ]
Bechikh, Slim [1 ]
Ben Said, Lamjed [1 ]
机构
[1] Univ Tunis, SMART Lab, ISG, Tunis, Tunisia
[2] Prince Sattam Bin Abdulaziz Univ, Coll Comp Engn & Sci, Dept Informat Syst, Al Kharj 11942, Saudi Arabia
来源
COMPUTATIONAL COLLECTIVE INTELLIGENCE, ICCCI 2022 | 2022年 / 13501卷
关键词
Deep CNN design; Deep CNN compression; Evolutionary algorithms; Thorax disease; Chest X-ray;
D O I
10.1007/978-3-031-16014-1_23
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Computer Vision has lately shown progress in addressing a variety of complex health care difficulties and has the potential to aid in the battle against certain lung illnesses, including COVID-19. Indeed, chest X-rays are one of the most commonly performed radiological techniques for diagnosing a range of lung diseases. Therefore, deep learning researchers have suggested that computer-aided diagnostic systems be built using deep learning methods. In fact, there are several CNN structures described in the literature. However, there are no guidelines for designing and compressing a specific architecture for a specific purpose; thus, such design remains highly subjective and heavily dependent on data scientists' knowledge and expertise. While deep convolutional neural networks have lately shown their ability to perform well in classification and dimension reduction tasks, the challenge of parameter selection is critical for these networks. However, since a CNN has a high number of parameters, its implementation in storage devices is difficult. This is due to the fact that the search space grows exponentially in size as the number of layers increases, and the large number of parameters necessitates extensive computation and storage, making it impractical for use on low-capacity devices. Motivated by these observations, we propose an automated method for CNN design and compression based on an evolutionary algorithm (EA) for X-Ray image classification that is capable of classifying radiography images and detecting possible chest abnormalities and infections, including COVID-19.Our evolutionary method is validated through a series of comparative experiments against relevant state-of-the-art architectures.
引用
收藏
页码:283 / 296
页数:14
相关论文
共 34 条
[1]  
[Anonymous], 2014, COMPUT RES REPOSITOR
[2]  
Blog G.R., 2017, AutoML for large scale image classification and object detection
[3]  
ELIAS P, 1975, IEEE T INFORM THEORY, V21, P194, DOI 10.1109/TIT.1975.1055349
[4]   OPTIMAL SOURCE CODES FOR GEOMETRICALLY DISTRIBUTED INTEGER ALPHABETS [J].
GALLAGER, RG ;
VANVOORHIS, DC .
IEEE TRANSACTIONS ON INFORMATION THEORY, 1975, 21 (02) :228-230
[5]  
Ge SM, 2017, IEEE INT CON MULTI, P667, DOI 10.1109/ICME.2017.8019465
[6]  
Han S., 2016, P INT C LEARN REPR
[7]   Licensed-Assisted Access for LTE in Unlicensed Spectrum: A MAC Protocol Design [J].
Han, Shiying ;
Liang, Ying-Chang ;
Chen, Qian ;
Soong, Boon-Hee .
2016 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2016,
[8]  
Han S, 2016, Arxiv, DOI arXiv:1510.00149
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]   Channel Pruning for Accelerating Very Deep Neural Networks [J].
He, Yihui ;
Zhang, Xiangyu ;
Sun, Jian .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :1398-1406