An optimized EfficientNetB0 framework with CLAHE-based preprocessing for accurate multi-class chest X-ray classification.
Authors
Affiliations (3)
Affiliations (3)
- Information Systems Department, Faculty of Information Systems and Computer Science, October 6 University, Giza, 12585, Egypt. [email protected].
- Department of Data Science and Artificial Intelligence, Faculty of Information Technology, Ajloun National University, P.O.43, Ajloun, 26810, Jordan.
- Department of Information Systems, Al-Alson Higher Institute, Cairo, Egypt.
Abstract
Chest radiography remains an essential diagnostic tool for thoracic diseases, yet interpreting overlapping anatomical structures is particularly challenging when multiple pathologies co-occur a common clinical scenario often oversimplified in deep learning approaches. This study presents an optimized EfficientNetB0 framework designed explicitly for multi-label classification of chest X-rays using the NIH dataset, integrating CLAHE-based contrast enhancement, strategic class balancing, and a comparative transfer learning strategy that preserves the dataset's inherent multi-label complexity. The proposed model achieved superior diagnostic performance with a macro-average AUC of 0.906 and recall of 0.824, outperforming DenseNet121 and MobileNetV2, and demonstrated strong per-class discrimination, especially for Pneumonia (AUC = 0.950) and Cardiomegaly (AUC = 0.946). These results confirm that the framework effectively balances learning capacity and generalization in a realistic multi-label clinical setting, offering a robust, interpretable solution suitable for computer-aided diagnosis where accurate detection of co-occurring thoracic pathologies is critical.