Improving Endoscopy Lesion Classification Using Self-Supervised Deep Learning.
Journal:
Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference
PMID:
40039453
Abstract
In this work, we assess the impact of self-supervised learning (SSL) approaches on the detection of gastritis atrophy (GA) and intestinal metaplasia (IM) conditions. GA and IM are precancerous gastric lesions. Detecting these lesions is crucial to intervene early and prevent their progression to cancer. A set of experiments is conducted over the Chengdu dataset, by considering different amounts of annotated data in the training phase. Our results reveal that, when all available data is used for training, SSL approaches achieve a classification accuracy on par with a supervised learning baseline, (81.52% vs 81.76%). Interestingly, we observe that in low-data regimes (here represented as retaining only 12.5% of annotated data for training), the SSL model guarantees an accuracy gain with respect to the supervised learning baseline of approximately 1.5% (73.00% vs 71.52%). This observation hints at the potential of SSL models in leveraging unlabeled data, thus showcasing more robust performance improvements and generalization. Experimental results also show that SSL performance is significantly dependent on the specific data augmentation techniques and parameters adopted for contrastive learning, thus advocating for further investigations into the definition of optimal data augmentation frameworks specifically tailored for gastric lesion detection applications.