Boundary-aware Information Maximization for Self-supervised Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
Unsupervised pre-training has been proven as an effective approach to boost various downstream tasks given limited labeled data. Among various methods, contrastive learning learns a discriminative representation by constructing positive and negative pairs. However, it is not trivial to build reasonable pairs for a segmentation task in an unsupervised way. In this work, we propose a novel unsupervised pre-training framework that avoids the drawback of contrastive learning. Our framework consists of two principles: unsupervised over-segmentation as a pre-train task using mutual information maximization and boundary-aware preserving learning. Experimental results on two benchmark medical segmentation datasets reveal our method's effectiveness in improving segmentation performance when few annotated images are available.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 40 of its 55 references that are in the map. First posted on arXiv 4 Feb 2022.
Builds on
40 of 55 references.
References · 40
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2016DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFscited by 3,239
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,586
- 2015SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentationcited by 1,483
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,379
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,265
28 more
- 2016Colorful Image Colorizationcited by 1,113
- 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 763
- 2020Self-supervised Learning: Generative or Contrastivecited by 693
- 2018VoxelMorph: A Learning Framework for Deformable Medical Image Registrationcited by 657
- 2019Self-supervised Visual Feature Learning with Deep Neural Networks: A Surveycited by 656
- 2018ADVENT: Adversarial Entropy Minimization for Domain Adaptation in Semantic Segmentationcited by 623
- 2018Invariant Information Clustering for Unsupervised Image Classification and Segmentationcited by 454
- 2019Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentationcited by 378
- 2020Contrastive learning of global and local features for medical image segmentation with limited annotationscited by 309
- 2019Interpolation Consistency Training for Semi-Supervised Learningcited by 248
- 2021Unsupervised Semantic Segmentation by Contrasting Object Mask Proposalscited by 190
- 2018Not-so-supervised: a survey of semi-supervised, multi-instance, and transfer learning in medical image analysiscited by 178
- 2021PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in Clusteringcited by 163
- 2020Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Imagescited by 152
- 2017Learning Discrete Representations via Information Maximizing Self-Augmented Trainingcited by 149
- 20203D Self-Supervised Methods for Medical Imagingcited by 134
- 2020Self-Supervision with Superpixels: Training Few-shot Medical Image Segmentation without Annotationcited by 128
- 2019Self-Supervised Learning for Cardiac MR Image Segmentation by Anatomical Position Predictioncited by 70
- 2019Multimodal Self-Supervised Learning for Medical Image Analysiscited by 57
- 2021Positional Contrastive Learning for Volumetric Medical Image Segmentationcited by 56
- 2021Semi-supervised Contrastive Learning for Label-efficient Medical Image Segmentationcited by 52
- 2021Local contrastive loss with pseudo-label based self-training for semi-supervised medical image segmentationcited by 43
- 2019Shape-Aware Organ Segmentation by Predicting Signed Distance Mapscited by 41
- 2021Self-Paced Contrastive Learning for Semi-supervised Medical Image Segmentation with Meta-labelscited by 32
- 2018Deep semi-supervised segmentation with weight-averaged consistency targetscited by 21
- 2021Medical Image Segmentation with Limited Supervision: A Review of Deep Network Modelscited by 21
- 2021Boosting Semi-supervised Image Segmentation with Global and Local Mutual Information Regularizationcited by 14
- 2021A Spatial Guided Self-supervised Clustering Network for Medical Image Segmentationcited by 14
Built on by
Citing papers per year: 2023: 1; 2024: 1.
2 papers.