Uncertainty-aware semi-supervised few shot segmentation

Soopil Kim

doi:10.1016/j.patcog.2022.109292

Uncertainty-aware semi-supervised few shot segmentation

2023-05-01

12

OA

AI

DOI：10.1016/j.patcog.2022.109292

Original

Save

View PDF

Abstract

En 中文

Few shot segmentation (FSS) aims to learn pixel-level classification of a target object in a query image using only a few annotated support samples. This is challenging as it requires modeling appearance vari-ations of target objects and the diverse visual cues between query and support images with limited in-formation. To address this problem, we propose a semi-supervised FSS strategy that leverages additional prototypes from unlabeled images with uncertainty guided pseudo label refinement. To obtain reliable prototypes from unlabeled images, we meta-train a neural network to jointly predict segmentation and estimate the uncertainty of predictions. We employ the uncertainty estimates to exclude predictions with high degrees of uncertainty for pseudo label construction to obtain additional prototypes from the refined pseudo labels. During inference, query segmentation is predicted using prototypes from both support and unlabeled images including low-level features of the query images. Our approach can easily supplement existing approaches without the requirement of additional training when employing unlabeled samples. Extensive experiments on PASCAL -5 i and COCO- 20i demonstrate that our model can effectively remove unreliable predictions to refine pseudo labels and significantly improve upon baseline performance.(c) 2023 Elsevier Ltd. All rights reserved.

Keywords:

Few shot segmentation

Meta learning

Uncertainty estimation

Semi-supervised learning

Prototype

Uncertainty-aware semi-supervised few shot segmentation

Abstract

AI Summary

Journal

Organization