Performance Evaluation of CNN Architectures on NIH Chest X-Ray Dataset with Boosting Ensemble
Lung diseases, including atelectasis, pneumonia, and pneumothorax, are leading causes of mortality worldwide, making early and accurate diagnosis crucial for effective treatment. This study applies a boosting ensemble method for thoracic disease classification using the National Institutes of Health (NIH) Chest X-ray dataset, which contains images across 14 disease categories. Five pre-trained convolutional neural network (CNN) models—VGG16, VGG19(Visual Geometry Group), ResNet50 (Residual Networks), MobileNetV2, and DenseNet201—were fine-tuned to extract features from the chest X-ray images. These models are widely used in medical image analysis due to their ability to automatically extract relevant features from complex images. To address data imbalance, often problematic in medical datasets, a downsampling approach for the majority class is applied. The study further employs a boosting ensemble method to improve classification accuracy of 92.58%. Comparisons between the individual CNN models and the ensemble method show that the ensemble consistently outperforms individual models, especially in handling imbalanced data. This approach demonstrates the potential of ensemble learning in enhancing the accuracy of lung disease detection through medical imaging.
Authors
- Mrityunjay Singh (ORCID: https://orcid.org/0000-0001-8672-4250)
- Prashansa Taneja
- Aman Sharma (ORCID: https://orcid.org/0000-0003-2263-7902)
Institutions
- Jaypee University of Information Technology (IN)
- Indian Institute of Information Technology Una (IN)
Publication Details
- Published
- 2024-12-18
- DOI
- https://doi.org/10.1109/pdgc64653.2024.10984390
- Citations
- 2
- Primary Topic
- AI in cancer detection
- Type
- article
- Field-Weighted Citation Impact
- 0.69