Few-shot segmentation aims to segment objects with novel classes in a query image, given a support set which consists of few annotated support images. A key factor in few-shot segmentation is to effectively exploit information for the target classes from the support set. In addition, we argue that the overall quality of information available in each training episode varies depending on the given support samples. In this paper, we propose Mask-Guided Attention module to extract more beneficial features for few-shot segmentation from the support images. Taking advantage of the support masks, the area correlated to the foreground object is highlighted and enables the support encoder to extract comprehensive support features with contextual information. Furthermore, we propose Episode Adaptive Weight to balance the training between different episodes. It adaptively adjusts loss weight according to the difficulty of each episode determined by self-supervised segmentation loss of support images and encourages the model to pay more attention to more difficult episodes. Extensive experimental results including comparisons with the state-of-the-art methods and ablation studies demonstrate the effectiveness of the proposed method.
|Title of host publication||2022 IEEE International Conference on Image Processing, ICIP 2022 - Proceedings|
|Publisher||IEEE Computer Society|
|Number of pages||5|
|Publication status||Published - 2022|
|Event||29th IEEE International Conference on Image Processing, ICIP 2022 - Bordeaux, France|
Duration: 2022 Oct 16 → 2022 Oct 19
|Name||Proceedings - International Conference on Image Processing, ICIP|
|Conference||29th IEEE International Conference on Image Processing, ICIP 2022|
|Period||22/10/16 → 22/10/19|
Bibliographical noteFunding Information:
This research was supported by the Yonsei University Research Fund of 2021 (2021-22-0001).
This research was supported by RD program for Advanced Integrated-intelligence for Identification (AIID) through the National Research Foundation of KOREA(NRF) funded by Ministry of Science and ICT (NRF-2018M3E3A1057289).
© 2022 IEEE.
All Science Journal Classification (ASJC) codes
- Computer Vision and Pattern Recognition
- Signal Processing