Semantics-Aware Dynamic Localization and Refinement for Referring Image Segmentation

Authors

  • Zhao Yang University of Oxford
  • Jiaqi Wang Shanghai AI Laboratory
  • Yansong Tang Tsinghua-Berkeley Shenzhen Institute, Tsinghua University
  • Kai Chen Shanghai AI Laboratory
  • Hengshuang Zhao The University of Hong Kong
  • Philip H.S. Torr University of Oxford

DOI:

https://doi.org/10.1609/aaai.v37i3.25428

Keywords:

CV: Multi-modal Vision, CV: Language and Vision, CV: Segmentation, ML: Representation Learning

Abstract

Referring image segmentation segments an image from a language expression. With the aim of producing high-quality masks, existing methods often adopt iterative learning approaches that rely on RNNs or stacked attention layers to refine vision-language features. Despite their complexity, RNN-based methods are subject to specific encoder choices, while attention-based methods offer limited gains. In this work, we introduce a simple yet effective alternative for progressively learning discriminative multi-modal features. The core idea of our approach is to leverage a continuously updated query as the representation of the target object and at each iteration, strengthen multi-modal features strongly correlated to the query while weakening less related ones. As the query is initialized by language features and successively updated by object features, our algorithm gradually shifts from being localization-centric to segmentation-centric. This strategy enables the incremental recovery of missing object parts and/or removal of extraneous parts through iteration. Compared to its counterparts, our method is more versatile—it can be plugged into prior arts straightforwardly and consistently bring improvements. Experimental results on the challenging datasets of RefCOCO, RefCOCO+, and G-Ref demonstrate its advantage with respect to the state-of-the-art methods.

Downloads

Published

2023-06-26

How to Cite

Yang, Z., Wang, J., Tang, Y., Chen, K., Zhao, H., & Torr, P. H. (2023). Semantics-Aware Dynamic Localization and Refinement for Referring Image Segmentation. Proceedings of the AAAI Conference on Artificial Intelligence, 37(3), 3222-3230. https://doi.org/10.1609/aaai.v37i3.25428

Issue

Section

AAAI Technical Track on Computer Vision III