Skip to main navigation Skip to search Skip to main content

SENTENCE-LEVEL PROMPTS BENEFIT COMPOSED IMAGE RETRIEVAL

  • Yang Bai
  • , Xinxing Xu
  • , Yong Liu
  • , Salman Khan
  • , Fahad Khan
  • , Wangmeng Zuo
  • , Rick Siow Mong Goh
  • , Chun Mei Feng*
  • *Corresponding author for this work
  • Agency for Science, Technology and Research, Singapore
  • Mohamed Bin Zayed University of Artificial Intelligence
  • Australian National University

Research output: Contribution to conferencePaperpeer-review

Abstract

Composed image retrieval (CIR) is the task of retrieving specific images by using a query that involves both a reference image and a relative caption. Most existing CIR models adopt the late-fusion strategy to combine visual and language features. Besides, several approaches have also been suggested to generate a pseudo-word token from the reference image, which is further integrated into the relative caption for CIR. However, these pseudo-word-based prompting methods have limitations when target image encompasses complex changes on reference image, e.g., object removal and attribute modification. In this work, we demonstrate that learning an appropriate sentence-level prompt for the relative caption (SPRC) is sufficient for achieving effective composed image retrieval. Instead of relying on pseudo-word-based prompts, we propose to leverage pretrained V-L models, e.g., BLIP-2, to generate sentence-level prompts. By concatenating the learned sentence-level prompt with the relative caption, one can readily use existing text-based image retrieval models to enhance CIR performance. Furthermore, we introduce both image-text contrastive loss and text prompt alignment loss to enforce the learning of suitable sentence-level prompts. Experiments show that our proposed method performs favorably against the state-of-the-art CIR methods on the Fashion-IQ and CIRR datasets.

Original languageEnglish
StatePublished - 2024
Event12th International Conference on Learning Representations, ICLR 2024 - Hybrid, Vienna, Austria
Duration: 7 May 202411 May 2024

Conference

Conference12th International Conference on Learning Representations, ICLR 2024
Country/TerritoryAustria
CityHybrid, Vienna
Period7/05/2411/05/24

Fingerprint

Dive into the research topics of 'SENTENCE-LEVEL PROMPTS BENEFIT COMPOSED IMAGE RETRIEVAL'. Together they form a unique fingerprint.

Cite this