Few-shot object segmentation with a new feature aggregation module

doi:10.1016/j.displa.2023.102459

Few-shot object segmentation with a new feature aggregation module

2023-07-01

5

PRE

AI

Palaiahnakote Shivakumara

Y

Yue Lu

DOI：10.1016/j.displa.2023.102459

Original

Original request for help

Save

Abstract

En 中文

The success of convolutional neural network for object segmentation depends on a large amount of training data and high-quality samples. But annotating such high-quality training data for pixel-wise segmentation is labor-intensive. To reduce the massive labor work, few-shot learning has been introduced to segment objects, which uses a few samples for training without compromising the performance. However, the current few-shot models are biased towards the seen classes rather than being class-irrelevant due to lack of global context prior attention. Therefore, this study aims at proposing a few-shot object segmentation model with a new feature aggregation module. Specifically, the proposed work develops a detail-aware module to enhance the discrimination of details with diversified attributes. To enhance the semantics of each pixel, we propose a global attention module to aggregate detailed features containing semantic information. Furthermore, to improve the performance of the proposed model, the model uses support samples that represents class-specific prototype obtained by respective category prototype block. Next, the proposed model predicts label of each pixel of query sample by estimating the distance between the pixel and prototypes. Experiments on standard datasets demonstrate significance of the proposed model over SOTA in terms of segmentation with a few training samples.

Keywords:

Few-shot learning

Object segmentation

Feature aggregation

Attention mechanism

Few-shot object segmentation with a new feature aggregation module

Abstract

Journal

Organization