Image to Pseudo-Episode: Boosting Few-Shot Segmentation by Unlabeled Data

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhang, Jie, Li, Yuhan, Wang, Yude, Lin, Stephen, Shan, Shiguang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914796164612096
author Zhang, Jie
Li, Yuhan
Wang, Yude
Lin, Stephen
Shan, Shiguang
author_facet Zhang, Jie
Li, Yuhan
Wang, Yude
Lin, Stephen
Shan, Shiguang
contents Few-shot segmentation (FSS) aims to train a model which can segment the object from novel classes with a few labeled samples. The insufficient generalization ability of models leads to unsatisfactory performance when the models lack enough labeled data from the novel classes. Considering that there are abundant unlabeled data available, it is promising to improve the generalization ability by exploiting these various data. For leveraging unlabeled data, we propose a novel method, named Image to Pseudo-Episode (IPE), to generate pseudo-episodes from unlabeled data. Specifically, our method contains two modules, i.e., the pseudo-label generation module and the episode generation module. The former module generates pseudo-labels from unlabeled images by the spectral clustering algorithm, and the latter module generates pseudo-episodes from pseudo-labeled images by data augmentation methods. Extensive experiments on PASCAL-$5^i$ and COCO-$20^i$ demonstrate that our method achieves the state-of-the-art performance for FSS.
format Preprint
id arxiv_https___arxiv_org_abs_2405_08765
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Image to Pseudo-Episode: Boosting Few-Shot Segmentation by Unlabeled Data
Zhang, Jie
Li, Yuhan
Wang, Yude
Lin, Stephen
Shan, Shiguang
Computer Vision and Pattern Recognition
Few-shot segmentation (FSS) aims to train a model which can segment the object from novel classes with a few labeled samples. The insufficient generalization ability of models leads to unsatisfactory performance when the models lack enough labeled data from the novel classes. Considering that there are abundant unlabeled data available, it is promising to improve the generalization ability by exploiting these various data. For leveraging unlabeled data, we propose a novel method, named Image to Pseudo-Episode (IPE), to generate pseudo-episodes from unlabeled data. Specifically, our method contains two modules, i.e., the pseudo-label generation module and the episode generation module. The former module generates pseudo-labels from unlabeled images by the spectral clustering algorithm, and the latter module generates pseudo-episodes from pseudo-labeled images by data augmentation methods. Extensive experiments on PASCAL-$5^i$ and COCO-$20^i$ demonstrate that our method achieves the state-of-the-art performance for FSS.
title Image to Pseudo-Episode: Boosting Few-Shot Segmentation by Unlabeled Data
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.08765