Alert button
Picture for Ajey Pai

Ajey Pai

Alert button

Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation

Jul 10, 2021
Prashant Pandey, Ajey Pai, Nisarg Bhatt, Prasenjit Das, Govind Makharia, Prathosh AP, Mausam

Figure 1 for Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation
Figure 2 for Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation
Figure 3 for Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation
Figure 4 for Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation

Contrastive Learning (CL) is a recent representation learning approach, which encourages inter-class separability and intra-class compactness in learned image representations. Since medical images often contain multiple semantic classes in an image, using CL to learn representations of local features (as opposed to global) is important. In this work, we present a novel semi-supervised 2D medical segmentation solution that applies CL on image patches, instead of full images. These patches are meaningfully constructed using the semantic information of different classes obtained via pseudo labeling. We also propose a novel consistency regularization (CR) scheme, which works in synergy with CL. It addresses the problem of confirmation bias, and encourages better clustering in the feature space. We evaluate our method on four public medical segmentation datasets and a novel histopathology dataset that we introduce. Our method obtains consistent improvements over state-of-the-art semi-supervised segmentation approaches for all datasets.

* MICCAI 2021 
Viaarxiv icon