Chest X-ray Classification using Deep Convolution Models on Low-resolution images with Uncertain Labels
Journal:
arXiv
Published Date:
Apr 12, 2025
Abstract
Deep Convolutional Neural Networks have consistently proven to achieve
state-of-the-art results on a lot of imaging tasks over the past years'
majority of which comprise of high-quality data. However, it is important to
work on low-resolution images since it could be a cheaper alternative for
remote healthcare access where the primary need of automated pathology
identification models occurs. Medical diagnosis using low-resolution images is
challenging since critical details may not be easily identifiable. In this
paper, we report classification results by experimenting on different input
image sizes of Chest X-rays to deep CNN models and discuss the feasibility of
classification on varying image sizes. We also leverage the noisy labels in the
dataset by proposing a Randomized Flipping of labels techniques. We use an
ensemble of multi-label classification models on frontal and lateral studies.
Our models are trained on 5 out of the 14 chest pathologies of the publicly
available CheXpert dataset. We incorporate techniques such as augmentation,
regularization for model improvement and use class activation maps to visualize
the neural network's decision making. Comparison with classification results on
data from 200 subjects, obtained on the corresponding high-resolution images,
reported in the original CheXpert paper, has been presented. For pathologies
Cardiomegaly, Consolidation and Edema, we obtain 3% higher accuracy with our
model architecture.