Bridging Clinical Gaps: Multi-Dataset Integration for Reliable Multi-Class Lung Disease Classification with DeepCRINet and Occlusion Sensitivity
Naeem Ullah, Javed Ali Khan, Ivanoe De Falco, Giovanna Sannino
Abstract
This research presents DeepCRINet, a deep learning (DL) model designed for reliable performance across various Chest Radiography Images (CRIs) datasets, in response to the urgent need for quick and accurate lung disease identification utilizing CRIs. Our method builds on earlier research, which frequently used single-source datasets that might not adequately represent the heterogeneity present in clinical situations. Our model’s diagnostic adaptability and real-world dependability are improved by utilizing images from different datasets, which helps us overcome limitations such as dataset bias, robustness, generalizability, and underrepresentation of conditions. With validation on a broad dataset consisting of 14,096 images (from three different datasets), DeepCRINet provides a solution that demonstrates excellent flexibility in recognizing illnesses including TuBerculosis, Pneumonia, COVID-19, and Lung Opacity. Through data augmentation, we improve the dataset, supporting training and testing procedures and confirming the model’s ability to generalize. We used occlusion sensitivity as a kind of explainable AI to openly identify and visually emphasize regions important to proper classification. This ability not only shows that DeepCRINet is analytically better than other DL models and hybrid techniques, but it also improves patient outcomes and diagnosis, which makes it a vital tool for medical professionals like radiologists.