Unsupervised Image Segmentation using Mutual Mean-Teaching
- URL: http://arxiv.org/abs/2012.08922v1
- Date: Wed, 16 Dec 2020 13:13:34 GMT
- Title: Unsupervised Image Segmentation using Mutual Mean-Teaching
- Authors: Zhichao Wu and Lei Guo and Hao Zhang and Dan Xu
- Abstract summary: We propose an unsupervised image segmentation model based on the Mutual Mean-Teaching (MMT) framework to produce more stable results.
Experimental results demonstrate that the proposed model is able to segment various types of images and achieves better performance than the existing methods.
- Score: 12.784209596867495
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Unsupervised image segmentation aims at assigning the pixels with similar
feature into a same cluster without annotation, which is an important task in
computer vision. Due to lack of prior knowledge, most of existing model usually
need to be trained several times to obtain suitable results. To address this
problem, we propose an unsupervised image segmentation model based on the
Mutual Mean-Teaching (MMT) framework to produce more stable results. In
addition, since the labels of pixels from two model are not matched, a label
alignment algorithm based on the Hungarian algorithm is proposed to match the
cluster labels. Experimental results demonstrate that the proposed model is
able to segment various types of images and achieves better performance than
the existing methods.
Related papers
- UnSeg: One Universal Unlearnable Example Generator is Enough against All Image Segmentation [64.01742988773745]
An increasing privacy concern exists regarding training large-scale image segmentation models on unauthorized private data.
We exploit the concept of unlearnable examples to make images unusable to model training by generating and adding unlearnable noise into the original images.
We empirically verify the effectiveness of UnSeg across 6 mainstream image segmentation tasks, 10 widely used datasets, and 7 different network architectures.
arXiv Detail & Related papers (2024-10-13T16:34:46Z) - Unsupervised Universal Image Segmentation [59.0383635597103]
We propose an Unsupervised Universal model (U2Seg) adept at performing various image segmentation tasks.
U2Seg generates pseudo semantic labels for these segmentation tasks via leveraging self-supervised models.
We then self-train the model on these pseudo semantic labels, yielding substantial performance gains.
arXiv Detail & Related papers (2023-12-28T18:59:04Z) - Which Pixel to Annotate: a Label-Efficient Nuclei Segmentation Framework [70.18084425770091]
Deep neural networks have been widely applied in nuclei instance segmentation of H&E stained pathology images.
It is inefficient and unnecessary to label all pixels for a dataset of nuclei images which usually contain similar and redundant patterns.
We propose a novel full nuclei segmentation framework that chooses only a few image patches to be annotated, augments the training set from the selected samples, and achieves nuclei segmentation in a semi-supervised manner.
arXiv Detail & Related papers (2022-12-20T14:53:26Z) - One Weird Trick to Improve Your Semi-Weakly Supervised Semantic
Segmentation Model [8.388356030608886]
Semi-weakly supervised semantic segmentation (SWSSS) aims to train a model to identify objects in images based on a small number of images with pixel-level labels, and many more images with only image-level labels.
Most existing SWSSS algorithms extract pixel-level pseudo-labels from an image classifier - a very difficult task to do well.
We propose a method called prediction filtering, which instead of extracting pseudo-labels, just uses the classifier as a classifier.
Adding this simple post-processing method to baselines gives results competitive with or better than prior SWSSS algorithms.
arXiv Detail & Related papers (2022-05-02T21:46:41Z) - Learning Contrastive Representation for Semantic Correspondence [150.29135856909477]
We propose a multi-level contrastive learning approach for semantic matching.
We show that image-level contrastive learning is a key component to encourage the convolutional features to find correspondence between similar objects.
arXiv Detail & Related papers (2021-09-22T18:34:14Z) - SCNet: Enhancing Few-Shot Semantic Segmentation by Self-Contrastive
Background Prototypes [56.387647750094466]
Few-shot semantic segmentation aims to segment novel-class objects in a query image with only a few annotated examples.
Most of advanced solutions exploit a metric learning framework that performs segmentation through matching each pixel to a learned foreground prototype.
This framework suffers from biased classification due to incomplete construction of sample pairs with the foreground prototype only.
arXiv Detail & Related papers (2021-04-19T11:21:47Z) - Group-Wise Semantic Mining for Weakly Supervised Semantic Segmentation [49.90178055521207]
This work addresses weakly supervised semantic segmentation (WSSS), with the goal of bridging the gap between image-level annotations and pixel-level segmentation.
We formulate WSSS as a novel group-wise learning task that explicitly models semantic dependencies in a group of images to estimate more reliable pseudo ground-truths.
In particular, we devise a graph neural network (GNN) for group-wise semantic mining, wherein input images are represented as graph nodes.
arXiv Detail & Related papers (2020-12-09T12:40:13Z) - Unsupervised Learning of Image Segmentation Based on Differentiable
Feature Clustering [14.074732867392008]
The usage of convolutional neural networks (CNNs) for unsupervised image segmentation was investigated in this study.
We present a novel end-to-end network of unsupervised image segmentation that consists of normalization and an argmax function for differentiable clustering.
Third, we present an extension of the proposed method for segmentation with scribbles as user input, which showed better accuracy than existing methods.
arXiv Detail & Related papers (2020-07-20T10:28:36Z) - Autoregressive Unsupervised Image Segmentation [8.894935073145252]
We propose a new unsupervised image segmentation approach based on mutual information between different views constructed of the inputs.
The proposed method outperforms current state-of-the-art on unsupervised image segmentation.
arXiv Detail & Related papers (2020-07-16T10:47:40Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.