cs.CVFeb 26, 2026

Pix2Key: Controllable Open-Vocabulary Retrieval with Semantic Decomposition and Self-Supervised Visual Dictionary Learning

Authors: Guoyizhe Wei, Yang Jiao, Nan Xi, Zhishen Huang, Jingjing Meng, Rama Chellappa, Yan Gao

Organizations: Johns Hopkins University, Baltimore, US · Amazon.com, Seattle, US

Abstract

Composed Image Retrieval (CIR) uses a reference image plus a natural-language edit to retrieve images that apply the requested change while preserving other relevant visual content. Classic fusion pipelines typically rely on supervised triplets and can lose fine-grained cues, while recent zero-shot approaches often caption the reference image and merge the caption with the edit, which may miss implicit user intent and return repetitive results. We present Pix2Key, which represents both queries and candidates as open-vocabulary visual dictionaries, enabling intent-aware constraint matching and diversity-aware reranking in a unified embedding space. A self-supervised pretraining component, V-Dict-AE, further improves the dictionary representation using only images, strengthening fine-grained attribute understanding without CIR-specific supervision. On the DFMM-Compose benchmark, Pix2Key improves Recall@10 up to 3.2 points, and adding V-Dict-AE yields an additional 2.3-point gain while improving intent consistency and maintaining high list diversity.

Figures & tables

Explore similar work

CardsList
  1. Less is More: Concise Retrieval and Critic-based Re-ranking for Composed Image Retrieval

    Oct 20, 2026Yuqian Zheng, Hyoju Ahn, Hyeongjin Ahn +1Composed Image RetrievalImage Retrieval

  2. DiCE-CIR: Direct Composition Learning for Efficient Zero-Shot Composed Image Retrieval

    Jul 6, 2026Gwang-Ho Na, Ho-Joong Kim, Seong-Whan LeeCross-Modal AlignmentCompositional Representation Learning