Multi-Scale CNN: An Explainable AI-Integrated Unique Deep Learning Framework for Lung-Affected Disease Classification
-
Published:2023-09-30
Issue:5
Volume:11
Page:134
-
ISSN:2227-7080
-
Container-title:Technologies
-
language:en
-
Short-container-title:Technologies
Author:
Sarkar Ovi1ORCID, Islam Md. Robiul1, Syfullah Md. Khalid1ORCID, Islam Md. Tohidul2ORCID, Ahamed Md. Faysal3ORCID, Ahsan Mominul4ORCID, Haider Julfikar5ORCID
Affiliation:
1. Department of Electrical & Computer Engineering, Rajshahi University of Engineering & Technology, Rajshahi 6204, Bangladesh 2. Department of Information & Communication Engineering, University of Rajshahi, Rajshahi 6205, Bangladesh 3. Department of Computer Science & Engineering, International Standard University, Dhaka 1212, Bangladesh 4. Department of Computer Science, University of York, Deramore Lane, Heslington, York YO10 5GH, UK 5. Department of Engineering, Manchester Metropolitan University, Chester Street, Manchester M1 5GD, UK
Abstract
Lung-related diseases continue to be a leading cause of global mortality. Timely and precise diagnosis is crucial to save lives, but the availability of testing equipment remains a challenge, often coupled with issues of reliability. Recent research has highlighted the potential of Chest X-ray (CXR) images in identifying various lung diseases, including COVID-19, fibrosis, pneumonia, and more. In this comprehensive study, four publicly accessible datasets have been combined to create a robust dataset comprising 6650 CXR images, categorized into seven distinct disease groups. To effectively distinguish between normal and six different lung-related diseases (namely, bacterial pneumonia, COVID-19, fibrosis, lung opacity, tuberculosis, and viral pneumonia), a Deep Learning (DL) architecture called a Multi-Scale Convolutional Neural Network (MS-CNN) is introduced. The model is adapted to classify multiple numbers of lung disease classes, which is considered to be a persistent challenge in the field. While prior studies have demonstrated high accuracy in binary and limited-class scenarios, the proposed framework maintains this accuracy across a diverse range of lung conditions. The innovative model harnesses the power of combining predictions from multiple feature maps at different resolution scales, significantly enhancing disease classification accuracy. The approach aims to shorten testing duration compared to the state-of-the-art models, offering a potential solution toward expediting medical interventions for patients with lung-related diseases and integrating explainable AI (XAI) for enhancing prediction capability. The results demonstrated an impressive accuracy of 96.05%, with average values for precision, recall, F1-score, and AUC at 0.97, 0.95, 0.95, and 0.94, respectively, for the seven-class classification. The model exhibited exceptional performance across multi-class classifications, achieving accuracy rates of 100%, 99.65%, 99.21%, 98.67%, and 97.47% for two, three, four, five, and six-class scenarios, respectively. The novel approach not only surpasses many pre-existing state-of-the-art (SOTA) methodologies but also sets a new standard for the diagnosis of lung-affected diseases using multi-class CXR data. Furthermore, the integration of XAI techniques such as SHAP and Grad-CAM enhanced the transparency and interpretability of the model’s predictions. The findings hold immense promise for accelerating and improving the accuracy and confidence of diagnostic decisions in the field of lung disease identification.
Subject
Computer Science (miscellaneous)
Reference37 articles.
1. (2023, February 16). Coronavirus Disease (COVID-19) Situation Reports. Available online: https://www.who.int/emergencies/diseases/novel-coronavirus-2019/situation-reports. 2. Teixeira, L.O., Pereira, R.M., Bertolini, D., Oliveira, L.S., Nanni, L., Cavalcanti, G.D.C., and Costa, Y.M.G. (2021). Impact of Lung Segmentation on the Diagnosis and Explanation of COVID-19 in Chest X-ray Images. Sensors, 21. 3. Kim, S., Rim, B., Choi, S., Lee, A., Min, S., and Hong, M. (2022). Deep Learning in Multi-Class Lung Diseases’ Classification on Chest X-ray Images. Diagnostics, 12. 4. Alsharif, R., Al-Issa, Y., Alqudah, A.M., Qasmieh, I.A., Mustafa, W.A., and Alquran, H. (2021). PneumoniaNet: Automated Detection and Classification of Pediatric Pneumonia Using Chest X-ray Images and CNN Approach. Electronics, 10. 5. Shamrat, F.M.J.M., Azam, S., Karim, A., Islam, R., Tasnim, Z., Ghosh, P., and De Boer, F. (2022). LungNet22: A Fine-Tuned Model for Multiclass Classification and Prediction of Lung Disease Using X-ray Images. J. Pers. Med., 12.
Cited by
6 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|