
Multi-Classification of Chest X-Ray Images Using an Enhanced Deep Learning Approach
Millions of people have died as a result of several epidemic lung diseases that have spread around the world, including COVID-19, lung opacity, and pneumonia. because these disorders exhibit small changes in Chest X-Ray (CXR) images, medical specialists have had difficulty in accurately diagnosing them. This research proposes a computer-aided lung disease detection technique based on CXR images to assist medical professionals. To meet the demand for practical and user-friendly diagnostic instruments, this research introduces new, enhanced deep-learning models designed explicitly for multi-class diagnosis of lung diseases, including normal, viral pneumonia, tuberculosis, lung opacity, fibrosis, and COVID-19 pneumonia, using CXR images. The four major phases of the proposed research are pre-processing, feature extraction, classification, and segmentation. First, the pre-processing phase employs a gaussian filtering approach to reduce input noise, thereby enhancing the image quality. Then, in the feature extraction stage, a custom Convolutional Neural Network (CNN) model is utilized to extract useful features and mitigate complexity problems. After feature extraction, the multiple classes of lung diseases are effectively classified by using the Enhanced ResNet50V2 (EResNet50V2) model. The region of infection in the chest radiographs is accurately identified and delineated using the Enhanced DeepLabV3+(EDeepLabV3+) model. The efficacy of the proposed approach is evaluated through extensive experiments against existing state-of-the-art methods. The research’s results demonstrated exceptional accuracy, with a 99.47% success rate. Throughout the six-class classification system, the average performance metrics for F1- score, recall, precision, Area Under the Curve (AUC), Intersection Over Union (IOU), and Dice Similarity Coefficient (DSC) were 0.9933, 0.9933, 0.9983, 0.9905, 0.9754, and 0.9867. This research advances the field of medical imaging and lays the foundation for developing deep learning-based systems for diagnosing lung diseases in the future.
- Abhishek G., Singla A., and Eshwar D., “DICA-Net: Optimizing Chest X-Ray Classification with Attention U-Net and Pigeon Local Search,” Network Modeling Analysis in Health Informatics and Bioinformatics, vol. 14, pp. 1-20, 2025. DOI:10.1007/s13721-025-00513-5
- Ali A., Wang Y., and Shi X., “Detection of Multi‐Class Lung Diseases Based on Customized Neural Network,” Computational Intelligence, vol. 40, no. 2, pp. e12649, 2024. https://doi.org/10.1111/coin.12649
- Almujally N., Mujtaba G., Alharbi S., Alshammari N., and Jalal A., “Swin Transformer-Enhanced UAV Surveillance: A Multi-Modal Feature Optimization for High-Precision Road Vehicles Detection,” The International Arab Journal of Information Technology, vol. 23, no. 1, pp. 144-153, 2026. https://doi.org/10.34028/iajit/23/1/13
- Arora R., Rao G., Banerjea S., and Rajitha B., “MLDC: Multi-Lung Disease Classification Using Quantum Classifier and Artificial Neural Networks,” Neural Computing and Applications, vol. 36, no. 7, pp. 3803-3816, 2024. https://doi.org/10.1007/s00521-023-09207-3
- Ashwini S., Arunkumar J., Prabu R., Singh N., and Singh N., “Diagnosis and Multi-Classification of Lung Diseases in CXR Images Using Optimized Deep Convolutional Neural Network,” Soft Computing, vol. 28, no. 7, pp. 6219-6233, 2024. https://doi.org/10.1007/s00500-023-09480-3
- Askr H., Moawad M., Darwish A., and Hassanien A., “Multi-Class Deep Learning Model for Predicting Lung Diseases based on Honey Badger Algorithm,” International Journal of Information Technology, vol. 17, no. 1, pp. 1147-1154, 2024. https://doi.org/10.1007/s41870-024-02046-y
- Bhosale R. and Yadav D., “Customized Convolutional Neural Network for Pulmonary Multi-Disease Classification Using Chest X-Ray Images,” Multimedia Tools and Applications, vol. 83, no. 6, pp. 18537-18571, 2024. https://doi.org/10.1007/s11042-023-16297-7
- Cahyani D., Oktoviana L., Yasin M., Wahyuningsih S., and et al., “Lung Diseases Identification Using Hybrid Transfer Learning and Bidirectional Long Short-Term Memory,” Bulletin of Electrical Engineering and Informatics, vol. 14, no. 2, pp. 1418-1427, 2025. https://doi.org/10.11591/eei.v14i2.9114
- Chen H., Qin Y., Liu X., Wang H., and Zhao J., “An Improved DeepLabv3+Lightweight Network for Remote-Sensing Image Semantic Segmentation,” Complex and Intelligent Systems, vol. 10, no. 2, pp. 1-11, 2024. DOI:10.1007/s40747-023-01304-z
- Chowdhury M. and Khandakar A., Kaggle, COVID-19 Radiography Dataset, https://www.kaggle.com/datasets/tawsifurrahman/covid19-radiography-database, Last Visited, 2024.
- Dandu R., Deepa B., and Murthy M., “Convolutional Neural Networks Framework for the Early Classification and Detection of Melanoma Skin Cancer,” SN Computer Science, vol. 5, no. 5, pp. 1-14, 2024. https://doi.org/10.1007/s42979-024-02937-8
- Dawood H., Nawaz M., Ilyas M., Nazir T., and Javed A., “Attention-Guided CenterNet Deep Learning Approach for Lung Cancer Detection,” Computers in Biology and Medicine, vol. 186, pp. 109613, 2025. https://doi.org/10.1016/j.compbiomed.2024.109613
- Deeplake, NIH ChestX-ray Dataset, https://datasets.activeloop.ai/docs/ml/datasets/nih-chest-x-ray-dataset/, Last Visited, 2024.
- Egala R. and Sairam M., “Multi-Layer Stacked Residual Coordinate Termite Alate Network for Multi-Class Lung Diseases Detection from Chest X-Ray Images,” Applied Soft Computing, vol. 179, pp. 113393, 2025. https://doi.org/10.1016/j.asoc.2025.113393
- El-Ghandour M. and Obayya M., “Pneumonia Detection in Chest X-Ray Images Using an Optimized Ensemble with XGBoost Classifier,” Multimedia Tools and Applications, vol. 84, pp. 5491-5521, 2025.
https://doi.org/10.1007/s11042-024-18975-6 - Geethu Lakshmi G. and Nagaraj P., “Lung Cancer Detection and Classification Using Optimized CNN Features and Squeeze-Inception-ResNeXt Model,” Computational Biology and Chemistry, vol. 117, pp. 108437, 2025. DOI:10.1016/j.compbiolchem.2025.108437
- Gite S., Mishra A., and Kotecha K., “Enhanced Lung Image Segmentation Using Deep Learning,” Neural Computing and Applications, vol. 35, pp. 22839-22853, 2023. https://doi.org/10.1007/s00521-021-06719-8
- Guan C., Ai H., Wang W., and Singh R., “Multi-CNNs with Variational Information Bottleneck for Chest X-Ray Classification,” The Journal of Supercomputing, vol. 81, no. 13, pp. 1-22, 2025. https://doi.org/10.1007/s11227-025-07767-w
- Hu W., Yang S., Guo W., Xiao N., and et al., “STC-UNet: Renal Tumor Segmentation Based on Enhanced Feature Extraction at Different Network Levels,” BMC Medical Imaging, vol. 24, no. 179, pp. 1-14, 2024. https://doi.org/10.1186/s12880-024-01359-5
- Ishwerlal R., Agarwal R., and Sujatha K., “Lung Disease Classification Using Chest X-Ray Image: an Optimal Ensemble of Classification with Hybrid Training,” Biomedical Signal Processing and Control, vol. 91, pp. 105941, 2024. https://doi.org/10.1016/j.bspc.2023.105941
- Karmakar M., Hota A., and Nag A., “Convolutional Neural Network (CNN) and Federated Learning-Based Approach for Lung Disease Detection,” Iran Journal of Computer Science, vol. 8, pp. 2387-2408, 2025. https://doi.org/10.1007/s42044-025-00320-1
- Kordnoori S., Sabeti M., Mostafaei H., and Banihashemi S., “LungXpertAI: A Deep Multi-Task Learning Model for Chest CT Scan Analysis and COVID-19 Detection,” Biomedical Signal Processing and Control, vol. 99, pp. 106866, 2025. https://doi.org/10.1016/j.bspc.2024.106866
- Kumar S. and Bhowmik B., “EffiCOVID-Net: A Highly Efficient Convolutional Neural Network for COVID-19 Diagnosis Using Chest X-Ray Imaging,” Methods, vol. 240, pp. 81-100, 2025. https://doi.org/10.1016/j.ymeth.2025.04.008
- Latha D. and Mahesh T., “Feature-Driven Breast Cancer Classification via Hybrid Model Using Mammogram Images,” Journal of Medical Engineering and Technology, vol. 49, no. 7, pp. 276-292, 2025. https://doi.org/10.1080/03091902.2025.2530938
- Malik H., Anees T., and Din M., “BDCNet: Multi-Classification Convolutional Neural Network Model for Classification of COVID-19, Pneumonia, and Lung Cancer from Chest Radiographs,” Multimedia Systems, vol. 28, no. 3, pp. 815-829, 2022. DOI:10.1007/s00530-021-00878-3
- Munawwar S. and Rao P., “An Efficient Deep Learning Based Multi-Level Feature Extraction Network for Multi-Modal Medical Image Fusion,” The International Arab Journal of Information Technology, vol. 22, no. 3, pp. 429-447, 2025. https://doi.org/10.34028/iajit/22/3/2
- Mutukuru R., Rajesh A., Ponduri V., Ahammed J., and Kothala L., “Transformer-Based RT-DETR Framework for Accurate Chest X-ray Disease Detection,” IRBM, vol. 46, no. 6, pp. 100912, 2025. https://doi.org/10.1016/j.irbm.2025.100912
- Naeem A., Malik H., Din M., Sadeghi-Niaraki A., and et al., “Skindwnet: A Novel Deep Learning Model for Multi-Class Classification of Skin Cancers Using Dermoscopic Images,” Multimedia Systems, vol. 31, no. 314, pp. 1-28, 2025. https://doi.org/10.1007/s00530-025-01952-w
- Nayantara P., Kamath S., Kadavigere R., and Manjunath K., “Automatic Liver Segmentation from Multiphase CT Using Modified Segnet and ASPP Module,” SN Computer Science, vol. 5, no. 4, pp. 1-18, 2024. https://doi.org/10.1007/s42979-024-02719-2
- Nitha V. and Vinod S., “Novel CEFNet Framework for Lung Disease Detection and Infection Region Identification,” Biomedical Signal Processing and Control, vol. 9, pp. 106624, 2024. https://doi.org/10.1016/j.bspc.2024.106624
- Rao G., Rajitha B., Srinivasu P., Ijaz M., and Wozniak M., “Hybrid Framework for Respiratory Lung Diseases Detection Based on Classical CNN and Quantum Classifiers from Chest X-Rays,” Biomedical Signal Processing and Control, vol. 8, no. 8, pp. 105567, 2024. https://doi.org/10.1016/j.bspc.2023.105567
- Rashad M., Alebiary D., Aldawsari M., El-Sawy A., and AbuEl-Atta A., “CCNN-SVM: Automated Model for Emotion Recognition Based on Custom Convolutional Neural Networks with SVM,” Information, vol. 15, no. 7, pp. 1-16, 2024. https://doi.org/10.3390/info15070384
- Samee N., Houssein E., Saber E., Hu G., and Wang M., “Integrated Deep Learning-Based IRACE and Convolutional Neural Networks for Chest X-Ray Image Classification,” Knowledge-Based Systems, vol. 329, pp. 114293, 2025. https://doi.org/10.1016/j.knosys.2025.114293
- Satapathy S., Cho S., Mishra S., Sah S., and Mohanty S., “A Federated Learning Approach for Classifying Chest Diseases from Chest X-Ray Images,” Biomedical Signal Processing and Control, vol. 100, pp. 107107, 2025. https://doi.org/10.1016/j.bspc.2024.107107
- Sayeed A., Khansur N., Srizon A., Faruk M., and et al., “An Effective Screening of COVID‐19 Pneumonia by Employing Chest X‐Ray Segmentation and Attention‐Based Ensembled Classification,” IET Image Processing, vol. 18, no. 9, pp. 2400-2416, 2024. https://doi.org/10.1049/ipr2.13106
- Shariff V., Paritala C., and Ankala K., “Optimizing Non-Small Cell Lung Cancer Detection with Convolutional Neural Networks and Differential Augmentation,” Scientific Reports, vol. 15, no. 15640, pp. 1-27, 2025. https://doi.org/10.1038/s41598-025-98731-4
- Sharma V., Nillmani., Gupta S., and Shukla K., “Deep Learning Models for Tuberculosis Detection and Infected Region Visualization in Chest X-Ray Images,” Intelligent Medicine, vol. 4, no. 2, pp. 104-113, 2024. https://doi.org/10.1016/j.imed.2023.06.001
- Shastri A., Prajapati Y., Katariya H., Paliwal M., and Sabale K., “Enhancing Clinical Outcomes Using Deep Learning Solution for Accurate Lung Cancer Classification,” Sensing and imaging, vol. 26, pp. 1-19, 2025. https://doi.org/10.1007/s11220-025-00548-y
- Shyni H. and E C., “Pulmonu-Net: A Semantic Lung Disease Segmentation Model Leveraging the Benefit of Multi-Scale Feature Concatenation and Leaky Relu,” Automatika, vol. 65, no. 2, pp. 641-651, 2024. https://doi.org/10.1080/00051144.2024.2314918
- Singh B. and Kumar P., “A Unified Framework for Lung Disease Screening Using Chest X-Ray Images,” Engineering Applications of Artificial Intelligence, vol. 161, pp. 112073, 2025. https://doi.org/10.1016/j.engappai.2025.112073
- Srivastava R., Kumar N., and Sandhan T., “Binary Classification of Laryngeal Images Utilising ResNet-50 CNN Architecture,” Indian Journal of Otolaryngology and Head and Neck Surgery, vol. 77, no. 2, pp. 644-651, 2025. DOI:10.1007/s12070-024-05202-9
- Tam W., Babyn P., and Alirezaie J., “Robust Lung Segmentation in Chest X-Ray Images Using Modified U-Net with Deeper Network and Residual Blocks,” Computer Methods and Programs in Biomedicine Update, vol. 8, pp. 100211, 2025. https://doi.org/10.1016/j.cmpbup.2025.100211
- Timpano G., Veltri P., Vizza P., Cascini G., and Manti F., “Deep Learning-Based 3D and 2D Approaches for Skeletal Muscle Segmentation on Low-Dose CT Images,” Journal of Imaging Informatics in Medicine, pp. 1-13, 2025. https://doi.org/10.1007/s10278-025-01646-9
- Truong T. and Do T., “Multimodal Approach for Lung Disease Classification: Fusing Chest X-Ray Images and Clinical Texts,” SN Computer Science, vol. 6, pp. 607, 2025. https://doi.org/10.1007/s42979-025-04138-3
- Ullah N., Khan J., Almakdi S., Alshehri M., and et al., “ChestCovidNet: An Effective DL-based Approach for COVID-19, Lung Opacity, and Pneumonia Detection Using Chest Radiographs Images,” Biochemistry and Cell Biology, vol. 104, pp. 1-16, 2024. DOI:10.1139/bcb-2023-0265
- Varadharajan I. and Rathinavelayutham S., “An Efficient Lung Disease Classification Using Dove Swarm Optimization Based Multi-Scale Faster-RCNN Model,” Evolving Systems, vol. 16, no. 3, pp. 1-22, 2025. https://doi.org/10.1007/s12530-025-09708-7
- Vasudevan P. and Ekambaram C., “HYAQP: A Hybrid Meta-Heuristic Optimization Model for Air Quality Prediction Using Unsupervised Machine Learning Paradigms,” The International Arab Journal of Information Technology, vol. 21, no. 5, pp. 953-966, 2024. https://doi.org/10.34028/iajit/21/5/15
- Verma K., Sikka G., Swaraj A., Kumar S., and Kumar A., “Classification of COVID-19 on Chest X-Ray Images Using Deep Learning Model with Histogram Equalization and Lung Segmentation,” SN Computer Science, vol. 5, pp. 1-15, 2024. https://doi.org/10.1007/s42979-024-02695-7
- Wei S., Hu Z., and Tan L., “Res-ECA-Unet++: An Automatic Segmentation Model for Ovarian Tumor Ultrasound Images Based on Residual Networks and Channel Attention Mechanism,” Frontiers in Medicine, vol. 12, pp. 1589356, 2025. DOI:10.3389/fmed.2025.1589356
- Weli M. and Abdullah O., “Digital Image Noise Reduction Based on Proposed Smoothing and Sharpening Filters,” The Indonesian Journal of Computer Science, vol. 13, no. 4, pp. 5001-5021, 2024. https://doi.org/10.33022/ijcs.v13i4.4151
- Wen M., Li C., Xue Y., Xu M., and et al., “YOFIR: High Precise Infrared Object Detection Algorithm Based on YOLO and Fasternet,” Infrared Physics and Technology, vol. 144, pp. 105627, 2025. https://doi.org/10.1016/j.infrared.2024.105627
- Yadav S., Rizvi S., and Agarwal P., “Detection of Lung Diseases for Pneumonia, Tuberculosis, and COVID-19 with Artificial Intelligence Tools,” SN Computer Science, vol. 5, pp. 1-18, 2024. https://doi.org/10.1007/s42979-024-02617-7
- Yang L., Wan Y., and Pan F., “Enhancing Chest X-ray Diagnosis with a Multimodal Deep Learning Network by Integrating Clinical History to Refine Attention,” Journal of Imaging Informatics in Medicine, vol. 38, pp. 3568-3583, 2025. https://doi.org/10.1007/s10278-025-01446-1
- Ye Y., Chen Y., Wang R., Zhu D., and et al., “Image Segmentation Using Improved U-Net Model and Convolutional Block Attention Module Based on Cardiac Magnetic Resonance Imaging,” Journal of Radiation Research and Applied Sciences, vol. 17, no. 1, pp. 100816, 2024. https://doi.org/10.1016/j.jrras.2023.100816
- Yilmaz E., Gorurgoz C., Kıs H., Canger E., and Oztaş B., “Forensic Dental Age Estimation with Deep Learning: A Modified Xception Model for Panoramic X-Ray Images,” Forensic Science, Medicine and Pathology, vol. 21, no. 2, pp. 565- 579, 2025. DOI:10.1007/s12024-025-00962-4
- Zou L., Li J., Chen H., Liang M., and et al., “MCG-Net: Medical Chief Complaint-Guided Multi-Modal Masked Content Pre-training for Chest Image Classification,” Expert Systems with Applications, vol. 271, pp. 126660, 2025. https://doi.org/10.1016/j.eswa.2025.126660