Comparison of Augmentation and Preprocessing Techniques for Improved Generalization Performance in Deep Learning based Chest X-Ray Classification
- Title
- Comparison of Augmentation and Preprocessing Techniques for Improved Generalization Performance in Deep Learning based Chest X-Ray Classification
- Creator
- Prakash S.; Ramamurthy B.
- Description
- Convolutional Neural Network (CNN) models are well known for image classification; however, the downside of CNN is the ineffectiveness to generalize and inclination towards over-fitting in case of a small train dataset. A balanced and sufficient data is thus essential to effectively train a CNN model, but this is not always possible, especially in the case of medical imaging data, as often patients with the same disease are not always available. Image augmentation addresses the given issue by creating new data points artificially with slight modifications. This study, investigates ten different methods with various parameters and probability and their combined effect on the test dataset's generalization performance and F1 Score. For the study, three pre-Trained CNN models, namely ResNetl8, ResNet34, and ResNet50, are fine tuned on a small training dataset of 500 Pneumonia and 160 Non-Pneumonia(Normal) Images for each augmentation setting. The test accuracy, F1 Score, and generalization performance were calculated for a test dataset consisting of 50 Pneumonia and 16 Non-Pneumonia(Normal) Images. 2022 IEEE.
- Source
- 2022 International Conference on Trends in Quantum Computing and Emerging Business Technologies, TQCEBT 2022
- Date
- 2022-01-01
- Publisher
- Institute of Electrical and Electronics Engineers Inc.
- Subject
- Augmentation; CNN; ResNet; Transfer Learning; X Ray
- Coverage
- Prakash S., CHRIST (Deemed to Be University), Computer Science Department, Bengaluru, India; Ramamurthy B., CHRIST (Deemed to Be University), Computer Science Department, Bengaluru, India
- Rights
- Restricted Access
- Relation
- ISBN: 978-166545361-5
- Format
- Online
- Language
- English
- Type
- Conference paper
Collection
Citation
Prakash S.; Ramamurthy B., “Comparison of Augmentation and Preprocessing Techniques for Improved Generalization Performance in Deep Learning based Chest X-Ray Classification,” CHRIST (Deemed To Be University) Institutional Repository, accessed February 25, 2025, https://archives.christuniversity.in/items/show/20181.