UCDR-Adapter: Exploring Adaptation of Pre-Trained Vision-Language Models for Universal Cross-Domain Retrieval

Autor:	Jiang, Haoyu, Cheng, Zhi-Qi, Moreira, Gabriel, Zhu, Jiawen, Sun, Jingdong, Ren, Bukun, He, Jun-Yan, Dai, Qi, Hua, Xian-Sheng
Rok vydání:	2024
Předmět:	Computer Science - Computer Vision and Pattern Recognition Computer Science - Information Retrieval Computer Science - Multimedia
Druh dokumentu:	Working Paper
Popis:	Universal Cross-Domain Retrieval (UCDR) retrieves relevant images from unseen domains and classes without semantic labels, ensuring robust generalization. Existing methods commonly employ prompt tuning with pre-trained vision-language models but are inherently limited by static prompts, reducing adaptability. We propose UCDR-Adapter, which enhances pre-trained models with adapters and dynamic prompt generation through a two-phase training strategy. First, Source Adapter Learning integrates class semantics with domain-specific visual knowledge using a Learnable Textual Semantic Template and optimizes Class and Domain Prompts via momentum updates and dual loss functions for robust alignment. Second, Target Prompt Generation creates dynamic prompts by attending to masked source prompts, enabling seamless adaptation to unseen domains and classes. Unlike prior approaches, UCDR-Adapter dynamically adapts to evolving data distributions, enhancing both flexibility and generalization. During inference, only the image branch and generated prompts are used, eliminating reliance on textual inputs for highly efficient retrieval. Extensive benchmark experiments show that UCDR-Adapter consistently outperforms ProS in most cases and other state-of-the-art methods on UCDR, U(c)CDR, and U(d)CDR settings. Comment: Accepted to WACV 2025. Project link: https://github.com/fine68/UCDR2024
Databáze:	arXiv
Externí odkaz:	http://arxiv.org/abs/2412.10680 Zobrazit plný text záznamu View this record from Arxiv