Local-Global Context Aware Transformer for Language-Guided Video Segmentation

TL;DR

This paper introduces Locater, a novel Transformer-based model with a memory mechanism for language-guided video segmentation, effectively capturing long-term context and achieving state-of-the-art results on multiple datasets.

Contribution

The paper proposes a local-global context aware Transformer with a finite memory for efficient, accurate language-guided video segmentation, and introduces a new challenging dataset A2D-S+.

Findings

Locater outperforms previous methods on three LVS datasets.

Locater won 1st place in a major video object segmentation challenge.

The model processes videos with linear time complexity and constant memory size.

Abstract

We explore the task of language-guided video segmentation (LVS). Previous algorithms mostly adopt 3D CNNs to learn video representation, struggling to capture long-term context and easily suffering from visual-linguistic misalignment. In light of this, we present Locater (local-global context aware Transformer), which augments the Transformer architecture with a finite memory so as to query the entire video with the language expression in an efficient manner. The memory is designed to involve two components -- one for persistently preserving global video content, and one for dynamically gathering local temporal context and segmentation history. Based on the memorized local-global context and the particular content of each frame, Locater holistically and flexibly comprehends the expression as an adaptive query vector for each frame. The vector is used to query the corresponding frame for mask generation. The memory also allows Locater to process videos with linear time complexity and constant size memory, while Transformer-style self-attention computation scales quadratically with sequence length. To thoroughly examine the visual grounding capability of LVS models, we contribute a new LVS dataset, A2D-S+, which is built upon A2D-S dataset but poses increased challenges in disambiguating among similar objects. Experiments on three LVS datasets and our A2D-S+ show that Locater outperforms previous state-of-the-arts. Further, we won the 1st place in the Referring Video Object Segmentation Track of the 3rd Large-scale Video Object Segmentation Challenge, where Locater served as the foundation for the winning solution. Our code and dataset are available at: https://github.com/leonnnop/Locater