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

被引:8
|
作者
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
相关论文
共 50 条
  • [31] The Effectiveness of Transfer Learning for Arabic Handwriting Recognition using Deep CNN
    Elleuch, Mohamed
    Jraba, Safa
    Kherallah, Monji
    [J]. JOURNAL OF INFORMATION ASSURANCE AND SECURITY, 2021, 16 (02): : 85 - 93
  • [32] Intelligent Motor Bearing Fault Diagnosis Using Channel Attention-Based CNN
    Yin, Jianguo
    Cen, Gang
    [J]. WORLD ELECTRIC VEHICLE JOURNAL, 2022, 13 (11)
  • [33] Delamination Diagnosis System Using Nonlinear Transformation-Based Augmentation Approach for CNN Transfer Learning
    Dong-Yoon Kim
    Yeon-Jun Woo
    Seong-Gyu Sim
    Gil Ho Yoon
    [J]. Journal of Vibration Engineering & Technologies, 2024, 12 : 3213 - 3230
  • [34] Deep transfer learning CNN based approach for COVID-19 detection
    Muhammad, Wazir
    Bhutto, Zuhaibuddin
    Shah, Syed Ali Raza
    Shah, Jalal
    Shaikh, Murtaza Hussain
    Hussain, Ayaz
    Thaheem, Imdadullah
    Ali, Shamshad
    [J]. INTERNATIONAL JOURNAL OF ADVANCED AND APPLIED SCIENCES, 2022, 9 (04): : 44 - 52
  • [35] Delamination Diagnosis System Using Nonlinear Transformation-Based Augmentation Approach for CNN Transfer Learning
    Kim, Dong-Yoon
    Woo, Yeon-Jun
    Sim, Seong-Gyu
    Yoon, Gil Ho
    [J]. JOURNAL OF VIBRATION ENGINEERING & TECHNOLOGIES, 2024, 12 (03) : 3213 - 3230
  • [36] A Novel Deep Transfer Learning Method for Intelligent Fault Diagnosis Based on Variational Mode Decomposition and Efficient Channel Attention
    Liu, Caiming
    Zheng, Xiaorong
    Bao, Zhengyi
    He, Zhiwei
    Gao, Mingyu
    Song, Wenlong
    [J]. ENTROPY, 2022, 24 (08)
  • [37] Deep learning-based CNN for multiclassification of ocular diseases using transfer learning
    Deepak, G. Divya
    Bhat, Subraya Krishna
    [J]. COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2024, 12 (01):
  • [38] Bearing fault diagnosis based on ID CNN attention gated recurrent network and transfer learning
    Shi J.
    Hou L.
    [J]. Zhendong yu Chongji/Journal of Vibration and Shock, 2023, 42 (03): : 159 - 164and173
  • [39] Deep Learning Architectures for Diagnosis of Diabetic Retinopathy
    Solano, Alberto
    Dietrich, Kevin N.
    Martinez-Sober, Marcelino
    Barranquero-Cardenosa, Regino
    Vila-Tomas, Jorge
    Hernandez-Camara, Pablo
    [J]. APPLIED SCIENCES-BASEL, 2023, 13 (07):
  • [40] Super-resolution of remotely sensed data using channel attention based deep learning approach
    Wang, Peijuan
    Bayram, Bulent
    Sertel, Elif
    [J]. INTERNATIONAL JOURNAL OF REMOTE SENSING, 2021, 42 (16) : 6050 - 6067