Learning to Annotate Part Segmentation with Gradient Matching
- URL: http://arxiv.org/abs/2211.03003v1
- Date: Sun, 6 Nov 2022 01:29:22 GMT
- Title: Learning to Annotate Part Segmentation with Gradient Matching
- Authors: Yu Yang, Xiaotian Cheng, Hakan Bilen, Xiangyang Ji
- Abstract summary: This paper focuses on tackling semi-supervised part segmentation tasks by generating high-quality images with a pre-trained GAN.
In particular, we formulate the annotator learning as a learning-to-learn problem.
We show that our method can learn annotators from a broad range of labelled images including real images, generated images, and even analytically rendered images.
- Score: 58.100715754135685
- License: http://creativecommons.org/licenses/by-nc-nd/4.0/
- Abstract: The success of state-of-the-art deep neural networks heavily relies on the
presence of large-scale labelled datasets, which are extremely expensive and
time-consuming to annotate. This paper focuses on tackling semi-supervised part
segmentation tasks by generating high-quality images with a pre-trained GAN and
labelling the generated images with an automatic annotator. In particular, we
formulate the annotator learning as a learning-to-learn problem. Given a
pre-trained GAN, the annotator learns to label object parts in a set of
randomly generated images such that a part segmentation model trained on these
synthetic images with their predicted labels obtains low segmentation error on
a small validation set of manually labelled images. We further reduce this
nested-loop optimization problem to a simple gradient matching problem and
efficiently solve it with an iterative algorithm. We show that our method can
learn annotators from a broad range of labelled images including real images,
generated images, and even analytically rendered images. Our method is
evaluated with semi-supervised part segmentation tasks and significantly
outperforms other semi-supervised competitors when the amount of labelled
examples is extremely limited.
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) - A Semi-Paired Approach For Label-to-Image Translation [6.888253564585197]
We introduce the first semi-supervised (semi-paired) framework for label-to-image translation.
In the semi-paired setting, the model has access to a small set of paired data and a larger set of unpaired images and labels.
We propose a training algorithm for this shared network, and we present a rare classes sampling algorithm to focus on under-represented classes.
arXiv Detail & Related papers (2023-06-23T16:13:43Z) - Semantic Segmentation with Generative Models: Semi-Supervised Learning
and Strong Out-of-Domain Generalization [112.68171734288237]
We propose a novel framework for discriminative pixel-level tasks using a generative model of both images and labels.
We learn a generative adversarial network that captures the joint image-label distribution and is trained efficiently using a large set of unlabeled images.
We demonstrate strong in-domain performance compared to several baselines, and are the first to showcase extreme out-of-domain generalization.
arXiv Detail & Related papers (2021-04-12T21:41:25Z) - Uncertainty guided semi-supervised segmentation of retinal layers in OCT
images [4.046207281399144]
We propose a novel uncertainty-guided semi-supervised learning based on a student-teacher approach for training the segmentation network.
The proposed framework is a key contribution and applicable for biomedical image segmentation across various imaging modalities.
arXiv Detail & Related papers (2021-03-02T23:14:25Z) - Information-Theoretic Segmentation by Inpainting Error Maximization [30.520622129165456]
We group image pixels into foreground and background, with the goal of minimizing predictability of one set from the other.
Our method does not involve training deep networks, is computationally cheap, class-agnostic, and even applicable in isolation to a single unlabeled image.
arXiv Detail & Related papers (2020-12-14T06:42:27Z) - 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) - A Few Guidelines for Incremental Few-Shot Segmentation [57.34237650765928]
Given a pretrained segmentation model and few images containing novel classes, our goal is to learn to segment novel classes while retaining the ability to segment previously seen ones.
We show how the main problems of end-to-end training in this scenario are.
i) the drift of the batch-normalization statistics toward novel classes that we can fix with batch renormalization and.
ii) the forgetting of old classes, that we can fix with regularization strategies.
arXiv Detail & Related papers (2020-11-30T20:45:56Z) - Grafit: Learning fine-grained image representations with coarse labels [114.17782143848315]
This paper tackles the problem of learning a finer representation than the one provided by training labels.
By jointly leveraging the coarse labels and the underlying fine-grained latent space, it significantly improves the accuracy of category-level retrieval methods.
arXiv Detail & Related papers (2020-11-25T19:06:26Z)
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.