Transfer learning approach for pediatric pneumonia diagnosis using channel attention deep CNN architectures

被引:17
作者
Prakash, J. Arun [1 ]
Asswin, C. R. [1 ]
Kumar, K. S. Dharshan [1 ]
Dora, Avinash [1 ]
Ravi, Vinayakumar [2 ]
Sowmya, V [1 ]
Gopalakrishnan, E. A. [1 ]
Soman, K. P. [1 ]
机构
[1] Amrita Vishwa Vidyapeetham, Ctr Computat Engn & Networking CEN, Amrita Sch Engn, Coimbatore, India
[2] Prince Mohammad Bin Fahd Univ, Ctr Artificial Intelligence, Khobar, Saudi Arabia
关键词
Pediatric pneumonia; Chest X-rays; Computer -aided diagnosis; Deep learning; Transfer learning; Channel attention; Kernel PCA; Stacking classifier; CONVOLUTIONAL NEURAL-NETWORK; AUTOMATED DETECTION; CLASSIFICATION;
D O I
10.1016/j.engappai.2023.106416
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Chest X-ray is the most commonly adopted non-invasive and painless diagnostic test for pediatric pneumonia. However, the low radiation levels for diagnosis make accurate detection challenging, and this initiates the need for an unerring computer-aided diagnosis model. Our work proposes stacking ensemble learning on features extracted from channel attention deep CNN architectures. The features extracted from the channel attentionbased ResNet50V2, ResNet101V2, ResNet152V2, Xception, and DenseNet169 are individually passed through Kernel PCA for dimensionality reduction and concatenated. A stacking classifier with Support Vector Classifier, Logistic Regression, K-Nearest Neighbour, Nu-SVC, and XGBClassifier is employed for the final- Normal and Pneumonia classification. The stacking classifier achieves an accuracy of 96.15%, precision of 97.91%, recall of 95.90%, F1 score of 96.89%, and an AUC score of 96.24% on the publicly available pediatric pneumonia dataset. We expect this model to help the real-time diagnosis of pediatric pneumonia significantly.
引用
收藏
页数:21
相关论文
共 61 条
[1]  
Asnaoui K. E., 2021, Artificial intelligence and blockchain for future cybersecurity applications, P257
[2]  
Blagec K, 2021, Arxiv, DOI [arXiv:2008.02577, 10.48550/arXiv.2008.02577, DOI 10.48550/ARXIV.2008.02577]
[3]   Computer-assisted frameworks for classification of liver, breast and blood neoplasias via neural networks: A survey based on medical images [J].
Brunetti, Antonio ;
Carnimeo, Leonarda ;
Trotta, Gianpaolo Francesco ;
Bevilacqua, Vitoantonio .
NEUROCOMPUTING, 2019, 335 :274-298
[4]   Computer-aided diagnosis in the era of deep learning [J].
Chan, Heang-Ping ;
Hadjiiski, Lubomir M. ;
Samala, Ravi K. .
MEDICAL PHYSICS, 2020, 47 (05) :E218-E227
[5]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[6]   A Novel Transfer Learning Based Approach for Pneumonia Detection in Chest X-ray Images [J].
Chouhan, Vikash ;
Singh, Sanjay Kumar ;
Khamparia, Aditya ;
Gupta, Deepak ;
Tiwari, Prayag ;
Moreira, Catarina ;
Damasevicius, Robertas ;
de Albuquerque, Victor Hugo C. .
APPLIED SCIENCES-BASEL, 2020, 10 (02)
[7]   Can AI Help in Screening Viral and COVID-19 Pneumonia? [J].
Chowdhury, Muhammad E. H. ;
Rahman, Tawsifur ;
Khandakar, Amith ;
Mazhar, Rashid ;
Kadir, Muhammad Abdul ;
Bin Mahbub, Zaid ;
Islam, Khandakar Reajul ;
Khan, Muhammad Salman ;
Iqbal, Atif ;
Al Emadi, Nasser ;
Reaz, Mamun Bin Ibne ;
Islam, Mohammad Tariqul .
IEEE ACCESS, 2020, 8 :132665-132676
[8]   A Transfer Learning Method for Pneumonia Classification and Visualization [J].
Eduardo Lujan-Garcia, Juan ;
Yanez-Marquez, Cornelio ;
Villuendas-Rey, Yenny ;
Camacho-Nieto, Oscar .
APPLIED SCIENCES-BASEL, 2020, 10 (08)
[9]  
Fatima M, 2017, Journal of Intelligent Learning Systems and Applications, V09, P1, DOI [10.4236/jilsa.2017.91001, 10.4236/jilsa.2017.91001, DOI 10.4236/JILSA.2017.91001]
[10]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861