A multichannel EfficientNet deep learning-based stacking ensemble approach for lung disease detection using chest X-ray images

被引:16
|
作者
Ravi, Vinayakumar [1 ]
Acharya, Vasundhara [2 ]
Alazab, Mamoun [3 ]
机构
[1] Prince Mohammad Bin Fahd Univ, Ctr Artificial Intelligence, Khobar, Saudi Arabia
[2] Manipal Acad Higher Educ MAHE, Manipal Inst Technol MIT, Manipal, India
[3] Charles Darwin Univ, Coll Engn IT & Environm, Casuarina, NT, Australia
关键词
Lung disease; Pneumonia; COVID-19; Tuberculosis; Deep learning; Transfer learning; Multichannel; Stacking; Chest X-ray; PNEUMONIA DETECTION; COVID-19; TUBERCULOSIS; DIAGNOSIS; CLASSIFICATION; NETWORK;
D O I
10.1007/s10586-022-03664-6
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
This paper proposes a multichannel deep learning approach for lung disease detection using chest X-rays. The multichannel models used in this work are EfficientNetB0, EfficientNetB1, and EfficientNetB2 pretrained models. The features from EfficientNet models are fused together. Next, the fused features are passed into more than one non-linear fully connected layer. Finally, the features passed into a stacked ensemble learning classifier for lung disease detection. The stacked ensemble learning classifier contains random forest and SVM in the first stage and logistic regression in the second stage for lung disease detection. The performance of the proposed method is studied in detail for more than one lung disease such as pneumonia, Tuberculosis (TB), and COVID-19. The performances of the proposed method for lung disease detection using chest X-rays compared with similar methods with the aim to show that the method is robust and has the capability to achieve better performances. In all the experiments on lung disease, the proposed method showed better performance and outperformed similar lung disease existing methods. This indicates that the proposed method is robust and generalizable on unseen chest X-rays data samples. To ensure that the features learnt by the proposed method is optimal, t-SNE feature visualization was shown on all three lung disease models. Overall, the proposed method has shown 98% detection accuracy for pediatric pneumonia lung disease, 99% detection accuracy for TB lung disease, and 98% detection accuracy for COVID-19 lung disease. The proposed method can be used as a tool for point-of-care diagnosis by healthcare radiologists.Journal instruction requires a city for affiliations; however, this is missing in affiliation 3. Please verify if the provided city is correct and amend if necessary.correct
引用
收藏
页码:1181 / 1203
页数:23
相关论文
共 50 条
  • [1] Deep learning-based automatic detection of tuberculosis disease in chest X-ray images
    Showkatian, Eman
    Salehi, Mohammad
    Ghaffari, Hamed
    Reiazi, Reza
    Sadighi, Nahid
    POLISH JOURNAL OF RADIOLOGY, 2022, 87 : E118 - E124
  • [2] Diagnosis of COVID-19 Using Chest X-ray Images and Disease Symptoms Based on Stacking Ensemble Deep Learning
    AlMohimeed, Abdulaziz
    Saleh, Hager
    El-Rashidy, Nora
    Saad, Redhwan M. A.
    El-Sappagh, Shaker
    Mostafa, Sherif
    DIAGNOSTICS, 2023, 13 (11)
  • [3] Pneumonia detection in chest X-ray images using an ensemble of deep learning models
    Kundu, Rohit
    Das, Ritacheta
    Geem, Zong Woo
    Han, Gi-Tae
    Sarkar, Ram
    PLOS ONE, 2021, 16 (09):
  • [4] An efficient deep learning-based framework for tuberculosis detection using chest X-ray images
    Iqbal, Ahmed
    Usman, Muhammad
    Ahmed, Zohair
    TUBERCULOSIS, 2022, 136
  • [5] An Efficient Deep Learning Approach for Detecting Lung Disease from Chest X-Ray Images Using Transfer Learning and Ensemble Modeling
    Sagor, Mostofa Kamal
    Dipto, Shakib Mahmud
    Jahan, Ishrat
    Chowdhury, Susmita
    Reza, Md Tanzim
    Alam, Md Ashraful
    2021 IEEE ASIA-PACIFIC CONFERENCE ON COMPUTER SCIENCE AND DATA ENGINEERING (CSDE), 2021,
  • [6] Chest X-ray Images for Lung Disease Detection Using Deep Learning Techniques: A Comprehensive Survey
    Al-qaness, Mohammed A. A.
    Zhu, Jie
    AL-Alimi, Dalal
    Dahou, Abdelghani
    Alsamhi, Saeed Hamood
    Abd Elaziz, Mohamed
    Ewees, Ahmed A.
    ARCHIVES OF COMPUTATIONAL METHODS IN ENGINEERING, 2024, 31 (6) : 3267 - 3301
  • [7] Deep Learning-Based Computer-Aided Pneumothorax Detection Using Chest X-ray Images
    Malhotra, Priyanka
    Gupta, Sheifali
    Koundal, Deepika
    Zaguia, Atef
    Kaur, Manjit
    Lee, Heung-No
    SENSORS, 2022, 22 (06)
  • [8] Modeling of deep learning enabled lung disease detection and classification on chest X-ray images
    Saturi, Swapna
    Banda, Sandhya
    INTERNATIONAL JOURNAL OF HEALTHCARE MANAGEMENT, 2022,
  • [9] Deep Learning-based Detection of COVID-19 from Chest X-ray Images
    Manokaran, Jenita
    Zabihollahy, Fatemeh
    Hamilton-Wright, Andrew
    Ukwatta, Eranga
    MEDICAL IMAGING 2021: BIOMEDICAL APPLICATIONS IN MOLECULAR, STRUCTURAL, AND FUNCTIONAL IMAGING, 2021, 11600
  • [10] Ensemble learning-based COVID-19 detection by feature boosting in chest X-ray images
    Upadhyay, Kamini
    Agrawal, Monika
    Deepak, Desh
    IET IMAGE PROCESSING, 2020, 14 (16) : 4059 - 4066