A Comparative Analysis of Chest X-ray Examination with AI Enhancement Using XAI Techniques

Öz

Chest X-ray analysis plays a vital role in diagnosing pneumonia, and recent advancements in Deep Learning (DL) methods have significantly improved the accuracy of automated diagnosis. This study explores the intersection of DL and explainable artificial intelligence (XAI) in the context of pneumonia diagnosis through chest X-rays. The dataset used in this study consists of 1,341 training images of healthy individuals and 3,875 images of pneumonia cases, with the test set comprising 234 healthy and 390 pneumonia cases. Additionally, the validation set includes 8 images for both categories. This diversity aims to enhance the model's ability to generalize across different scenarios. The Convolutional Neural Network (CNN) and Transfer Learning (TL) methods utilizing the ResNet50 model achieved accuracies of 95.23 and 96.67, respectively. Subsequently, the models were explained using XAI methods such as SHAP and Grad-CAM. The study concludes by highlighting the potential of DL and XAI to enhance the interpretability and reliability of pneumonia diagnoses through chest X-ray analysis, aiming to contribute to future research in this field.

Anahtar Kelimeler

• Chest X-ray analysis
• Deep Learning (DL) methods
• explainable AI (XAI)
• Convolutional Neural Networks (CNNs)
• Grad-CAM (Gradient-weighted Class Activation Mapping)

Kaynakça

1. [1] H. Sharma, et al., "Feature Extraction and Classification of Chest X-Ray Images Using CNN to Detect Pneumonia," in 2020 10th International Conference on Cloud Computing, Data Science & Engineering (Confluence), 2020, pp. 227–231.
2. [2] S.-H. Lo and Y. Yin, "A novel interaction-based methodology towards explainable AI with better understanding of Pneumonia Chest X-ray Images," Discover Artificial Intelligence, vol. 1, no. 1, p. 16, 2021.
3. [3] M. Eisen and A. Ribeiro, "Optimal Wireless Resource Allocation With Random Edge Graph Neural Networks," ARXIV-EESS.SP, 2019.
4. [4] M. Rahimzadeh and A. Attar, "A Modified Deep Convolutional Neural Network for Detecting COVID-19 and Pneumonia from Chest X-ray Images Based on The Concatenation of Xception and ResNet50V2," ARXIV-EESS.IV, 2020.
5. [5] G. F. Elsayed, B. Wohlberg, and S. Jastrzębski, "Deep Double Descent: Where Bigger Models and More Data Hurt," ARXIV-EESS.ST, 2020.
6. [6] Y. Yang, G. Mei, and F. Piccialli, "A Deep Learning Approach Considering Image Background for Pneumonia Identification Using Explainable AI (XAI)," IEEE/ACM Transactions on Computational Biology and Bioinformatics, vol. PP, 2022, doi: 10.1109/TCBB.2022.3190265.
7. [7] L. V. de Moura, C. Mattjie, C. M. Dartora, R. C. Barros, and A. M. Marques da Silva, "Explainable Machine Learning for COVID-19 Pneumonia Classification With Texture-Based Features Extraction in Chest Radiography," Frontiers in Digital Health, vol. 3, 2022, doi: 10.3389/fdgth.2021.662343.
8. [8] H. Ren, et al., "Interpretable Pneumonia Detection by Combining Deep Learning and Explainable Models With Multisource Data," IEEE Access, vol. 9, pp. 95872–95883, 2021, doi: 10.1109/ACCESS.2021.3094025.

9. [9] L. Zou, et al., "Ensemble image explainable AI (XAI) algorithm for severe community-acquired pneumonia and COVID-19 respiratory infections," IEEE Transactions on Artificial Intelligence, vol. 4, no. 2, pp. 242–254, 2022, doi: 10.1109/TAI.2022.3154871.
10. [10] O. Stephen, M. Sain, U. J. Maduh, and D. U. Jeong, "An efficient deep learning approach to pneumonia classification in healthcare," Journal of Healthcare Engineering, vol. 2019, Article ID 4180949, 2019, doi: 10.1155/2019/4180949.
11. [11] R. Alsharif, et al., "PneumoniaNet: Automated detection and classification of pediatric pneumonia using chest X-ray images and CNN approach," Electronics, vol. 10, no. 23, p. 2949, 2021, doi: 10.3390/electronics10232949.
12. [12] D. Varshni, K. Thakral, L. Agarwal, R. Nijhawan, and A. Mittal, "Pneumonia detection using CNN based feature extraction," in 2019 IEEE International Conference on Electrical, Computer and Communication Technologies (ICECCT), 2019, pp. 1–7, doi: 10.1109/ICECCT.2019.8869364.
13. [13] W. Zhang, et al., "Blind Image Quality Assessment Using A Deep Bilinear Convolutional Neural Network," ARXIV-EESS.IV, 2019.
14. [14] D. Valsesia, et al., "Deep Graph-Convolutional Image Denoising," ARXIV-EESS.IV, 2019.
15. [15] M. Gil-Martín, J. Montero, and R. San-Segundo, "Parkinson’s Disease Detection from Drawing Movements Using Convolutional Neural Networks," ELECTRONICS, 2019.
16. [16] H. Gao, et al., "PhyGeoNet: Physics-Informed Geometry-Adaptive Convolutional Neural Networks For Solving Parameterized Steady-State PDEs On Irregular Domain," ARXIV-EESS.IV, 2020.
17. [17] F. Eitel, K. Ritter, and Alzheimer’s Disease Neuroimaging Initiative (ADNI), "Testing the robustness of attribution methods for convolutional neural networks in MRI-based Alzheimer’s disease classification," in Interpretability of Machine Intelligence in Medical Image Computing and Multimodal Learning for Clinical Decision Support: Second International Workshop, iMIMIC 2019, and 9th International Workshop, ML-CDS 2019, Held in Conjunction with MICCAI 2019, Shenzhen, China, October 17, 2019, Proceedings, 2019, pp. 3–11.
18. [18] I. D. Apostolopoulos and T. Bessiana, "Covid-19: Automatic Detection From X-Ray Images Utilizing Transfer Learning With Convolutional Neural Networks," ARXIV-EESS.IV, 2020.
19. [19] H. S. Maghdid, et al., "Diagnosing COVID-19 Pneumonia From X-Ray And CT Images Using Deep Learning And Transfer Learning Algorithms," ARXIV-EESS.IV, 2020.
20. [20] N. E. M. Khalifa, et al., "Detection of Coronavirus (COVID-19) Associated Pneumonia Based on Generative Adversarial Networks and A Fine-Tuned Deep Transfer Learning Model Using Chest X-ray Dataset," ARXIV, 2020.
21. [21] T. Rahman, et al., "Transfer Learning With Deep Convolutional Neural Network (CNN) For Pneumonia Detection Using Chest X-ray," ARXIV-EESS.IV, 2020.
22. [22] P. R. A. S. Bassi and R. Attux, "A Deep Convolutional Neural Network for COVID-19 Detection Using Chest X-Rays," ARXIV-EESS.IV, 2020.
23. [23] Z. Zhou, et al., "Models Genesis: Generic Autodidactic Models For 3D Medical Image Analysis," ARXIV-EESS.IV, 2019.
24. [24] Y.-A. Chung and J. Glass, "Generative Pre-Training For Speech With Autoregressive Predictive Coding," ARXIV-EESS.AS, 2019.
25. [25] C. L. Srinidhi, O. Ciga, and A. L. Martel, "Deep Neural Network Models For Computational Histopathology: A Survey," ARXIV-EESS.IV, 2019.
26. [26] Z. Zhao, et al., "Applications of Unsupervised Deep Transfer Learning to Intelligent Fault Diagnosis: A Survey," ARXIV-EESS.SP, 2019.
27. [27] M. Goyal, et al., "Artificial Intelligence-Based Image Classification For Diagnosis Of Skin Cancer: Challenges And Opportunities," ARXIV-EESS.IV, 2019.
28. [28] S. M. Lundberg, et al., "A Unified Approach to Interpreting Model Predictions," in Advances in Neural Information Processing Systems, 2017.
29. [29] SHAP Contributors, "SHAP (SHapley Additive exPlanations) Documentation," 2020. [Online]. Available: https://shap.readthedocs.io/en/latest/
30. [30] S. Basu, S. Mitra, and N. Saha, "Deep Learning For Screening COVID-19 Using Chest X-Ray Images," ARXIV-EESS.IV, 2020.
31. [31] C. Xia, et al., "Vision Based Defects Detection for Keyhole TIG Welding Using Deep Learning with Visual Explanation," Journal of Manufacturing Processes, 2020.
32. [32] M. R. Karim, et al., "DeepCOVIDExplainer: Explainable COVID-19 Diagnosis Based On Chest X-ray Images," ARXIV-EESS.IV, 2020.
33. [33] S. Vijayarangan, et al., "Interpreting Deep Neural Networks For Single-Lead ECG Arrhythmia Classification," ARXIV-EESS.SP, 2020.
34. [34] M. Kim, et al., "Medinoid: Computer-Aided Diagnosis and Localization of Glaucoma Using Deep Learning," Applied Sciences, 2019.
35. [35] I. Elbouknify, et al., "CT-xCOV: A CT-scan Based Explainable Framework for COVID-19 Diagnosis," ARXIV-EESS.IV, 2023.