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

Vinayakumar Ravi, Vasundhara Acharya, Mamoun Alazab

    Research output: Contribution to journalArticlepeer-review

    Abstract

    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.

    Original languageEnglish
    Pages (from-to)1181-1203
    Number of pages23
    JournalCluster Computing
    Volume26
    Issue number2
    Early online date2022
    DOIs
    Publication statusPublished - Apr 2023

    Fingerprint

    Dive into the research topics of 'A multichannel EfficientNet deep learning-based stacking ensemble approach for lung disease detection using chest X-ray images'. Together they form a unique fingerprint.

    Cite this