Scientific article
OA Policy
English

Contrastive learning vs. self-learning vs. deformable data augmentation in semantic segmentation of medical images

Published inJournal of Imaging Informatics in Medicine, vol. 37, no. 6, p. 3217-3230
Publication date2024-12
First online date2024-06-10
Abstract

To develop a robust segmentation model, encoding the underlying features/structures of the input data is essential to discriminate the target structure from the background. To enrich the extracted feature maps, contrastive learning and self-learning techniques are employed, particularly when the size of the training dataset is limited. In this work, we set out to investigate the impact of contrastive learning and self-learning on the performance of the deep learning-based semantic segmentation. To this end, three different datasets were employed used for brain tumor and hippocampus delineation from MR images (BraTS and Decathlon datasets, respectively) and kidney segmentation from CT images (Decathlon dataset). Since data augmentation techniques are also aimed at enhancing the performance of deep learning methods, a deformable data augmentation technique was proposed and compared with contrastive learning and self-learning frameworks. The segmentation accuracy for the three datasets was assessed with and without applying data augmentation, contrastive learning, and self-learning to individually investigate the impact of these techniques. The self-learning and deformable data augmentation techniques exhibited comparable performance with Dice indices of 0.913 ± 0.030 and 0.920 ± 0.022 for kidney segmentation, 0.890 ± 0.035 and 0.898 ± 0.027 for hippocampus segmentation, and 0.891 ± 0.045 and 0.897 ± 0.040 for lesion segmentation, respectively. These two approaches significantly outperformed the contrastive learning and the original model with Dice indices of 0.871 ± 0.039 and 0.868 ± 0.042 for kidney segmentation, 0.872 ± 0.045 and 0.865 ± 0.048 for hippocampus segmentation, and 0.870 ± 0.049 and 0.860 ± 0.058 for lesion segmentation, respectively. The combination of self-learning with deformable data augmentation led to a robust segmentation model with no outliers in the outcomes. This work demonstrated the beneficial impact of self-learning and deformable data augmentation on organ and lesion segmentation, where no additional training datasets are needed.

Keywords
  • Contrastive-learning
  • Data augmentation
  • Deep learning
  • Segmentation
  • Self-learning
Citation (ISO format)
ARABI, Hossein, ZAIDI, Habib. Contrastive learning vs. self-learning vs. deformable data augmentation in semantic segmentation of medical images. In: Journal of Imaging Informatics in Medicine, 2024, vol. 37, n° 6, p. 3217–3230. doi: 10.1007/s10278-024-01159-x
Main files (1)
Article (Published version)
Identifiers
Additional URL for this publicationhttps://link.springer.com/10.1007/s10278-024-01159-x
Journal ISSN2948-2925
72views
62downloads

Technical informations

Creation10/06/2024 21:06:17
First validation18/06/2024 09:02:42
Update06/12/2024 11:19:52
Status update06/12/2024 11:19:52
Last indexation06/12/2024 11:19:53
All rights reserved by Archive ouverte UNIGE and the University of GenevaunigeBlack