An investigation into augmentation and preprocessing for optimising X-ray classification in limited datasets: a case study on necrotising enterocolitis.
Int J Comput Assist Radiol Surg
; 19(6): 1223-1231, 2024 Jun.
Article
in En
| MEDLINE
| ID: mdl-38652416
ABSTRACT
PURPOSE:
Obtaining large volumes of medical images, required for deep learning development, can be challenging in rare pathologies. Image augmentation and preprocessing offer viable solutions. This work explores the case of necrotising enterocolitis (NEC), a rare but life-threatening condition affecting premature neonates, with challenging radiological diagnosis. We investigate data augmentation and preprocessing techniques and propose two optimised pipelines for developing reliable computer-aided diagnosis models on a limited NEC dataset.METHODS:
We present a NEC dataset of 1090 Abdominal X-rays (AXRs) from 364 patients and investigate the effect of geometric augmentations, colour scheme augmentations and their combination for NEC classification based on the ResNet-50 backbone. We introduce two pipelines based on colour contrast and edge enhancement, to increase the visibility of subtle, difficult-to-identify, critical NEC findings on AXRs and achieve robust accuracy in a challenging three-class NEC classification task.RESULTS:
Our results show that geometric augmentations improve performance, with Translation achieving +6.2%, while Flipping and Occlusion decrease performance. Colour augmentations, like Equalisation, yield modest improvements. The proposed Pr-1 and Pr-2 pipelines enhance model accuracy by +2.4% and +1.7%, respectively. Combining Pr-1/Pr-2 with geometric augmentation, we achieve a maximum performance increase of 7.1%, achieving robust NEC classification.CONCLUSION:
Based on an extensive validation of preprocessing and augmentation techniques, our work showcases the previously unreported potential of image preprocessing in AXR classification tasks with limited datasets. Our findings can be extended to other medical tasks for designing reliable classifier models with limited X-ray datasets. Ultimately, we also provide a benchmark for automated NEC detection and classification from AXRs.Key words
Full text:
1
Collection:
01-internacional
Database:
MEDLINE
Main subject:
Enterocolitis, Necrotizing
Limits:
Female
/
Humans
/
Newborn
Language:
En
Journal:
Int J Comput Assist Radiol Surg
Journal subject:
RADIOLOGIA
Year:
2024
Document type:
Article
Affiliation country:
Publication country:
ALEMANHA
/
ALEMANIA
/
DE
/
DEUSTCHLAND
/
GERMANY