References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Contrastive Semi-Supervised Learning for 2D Medical Image Segmentation
(Submitted on 12 Jun 2021 (v1), last revised 6 Aug 2021 (this version, v3))
Abstract: 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.
Submission history
From: . Mausam [view email][v1] Sat, 12 Jun 2021 15:43:24 GMT (1285kb,D)
[v2] Sat, 10 Jul 2021 03:45:49 GMT (1028kb,D)
[v3] Fri, 6 Aug 2021 16:33:11 GMT (0kb,I)
Link back to: arXiv, form interface, contact.