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 条
[41]  
S K, 2019, BIOMED PHARMACOL J, V12, P1577, DOI [10.13005/bpj/1788, 10.13005/bpj/1788, DOI 10.13005/BPJ/1788, DOI 10.13005/bpj/1788]
[42]   Models of Learning to Classify X-ray Images for the Detection of Pneumonia using Neural Networks [J].
Saraiva, A. A. ;
Santos, D. B. S. ;
Costa, Nator Junior C. ;
Sousa, Jose Vigno M. ;
Fonseca Ferreira, N. M. ;
Valente, Antonio ;
Soares, Salviano .
BIOIMAGING: PROCEEDINGS OF THE 12TH INTERNATIONAL JOINT CONFERENCE ON BIOMEDICAL ENGINEERING SYSTEMS AND TECHNOLOGIES, VOL 2, 2019, :76-83
[43]  
Saraiva A. A., 2019, BIOIMAGING
[44]  
Siddiqi R, 2020, SN Computer Science, V1, DOI [10.1007/s42979-020-00361-2, 10.1007/s42979-020-00361-2]
[45]   Automated Pneumonia Diagnosis using a Customized Sequential Convolutional Neural Network [J].
Siddiqi, Raheel .
ICDLT 2019: 2019 3RD INTERNATIONAL CONFERENCE ON DEEP LEARNING TECHNOLOGIES, 2019, :64-70
[46]  
Simonyan K, 2015, Arxiv, DOI arXiv:1409.1556
[47]   Deep neural network ensemble for pneumonia localization from a large-scale chest x-ray database [J].
Sirazitdinov, Ilyas ;
Kholiavchenko, Maksym ;
Mustafaev, Tamerlan ;
Yuan Yixuan ;
Kuleev, Ramil ;
Ibragimov, Bulat .
COMPUTERS & ELECTRICAL ENGINEERING, 2019, 78 :388-399
[48]   An Efficient Deep Learning Approach to Pneumonia Classification in Healthcare [J].
Stephen, Okeke ;
Sain, Mangal ;
Maduh, Uchenna Joseph ;
Jeong, Do-Un .
JOURNAL OF HEALTHCARE ENGINEERING, 2019, 2019
[49]  
Szegedy C, 2017, AAAI CONF ARTIF INTE, P4278
[50]   Rethinking the Inception Architecture for Computer Vision [J].
Szegedy, Christian ;
Vanhoucke, Vincent ;
Ioffe, Sergey ;
Shlens, Jon ;
Wojna, Zbigniew .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :2818-2826