DQ-DETR: Dual Query Detection Transformer for Phrase Extraction and Grounding

By Siyi Liu, Yaoyuan Liang, Feng Li, Shijia Huang, Hao Zhang, Hang Su, Jun-Juan Zhu, Lei Zhang. 2022

45 citations recorded by the source corpus.

In this paper, we study the problem of visual grounding by considering both phrase extraction and grounding (PEG). In contrast to the previous phrase-known-at-test setting, PEG requires a model to extract phrases from text and locate objects from image simultaneously, which is a more practical setting in real applications. As phrase extraction can be regarded as a 1D text segmentation problem, we formulate PEG as a dual detection problem and propose a novel DQ-DETR model, which introduces dual queries to probe different features from image and text for object prediction and phrase mask prediction. Each pair of dual queries are designed to have shared positional parts but different content parts. Such a design effectively alleviates the difficulty of modality alignment between image and text (in contrast to a single query design) and empowers Transformer decoder to leverage phrase mask-guided attention to improve the performance. To evaluate the performance of PEG, we also propose a new metric CMAP (cross-modal average precision), analogous to the AP metric in object detection. The new metric overcomes the ambiguity of Recall@1 in many-box-to-one-phrase cases in phrase grounding. As a result, our PEG pre-trained DQ-DETR establishes new state-of-the-art results on all visual grounding benchmarks with a ResNet-101 backbone. For example, it achieves 91.04% and 83.51% in terms of recall rate on RefCOCO testA and testB with a ResNet-101 backbone.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Natural Language Processing, Information Extraction

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Feng Li
  2. Hao Zhang
  3. Shilong Liu
  4. Yukang Chen
  5. Lei Zhang

In Computer Vision and Pattern Recognition

  1. Feng Li
  2. Hao Zhang
  3. Shilong Liu
  4. Yukang Chen
  5. Lei Zhang
  6. Jianwei Yang
  7. Yu Liu
  8. Guanglu Song

In Natural Language Processing

  1. Feng Li
  2. Hao Zhang

Citations in the corpus