Low-rank Prompt Interaction for Continual Vision-Language Retrieval
Weicai Yan, Ye Wang, Wang Lin, Zirun Guo, Zhou Zhao, Tao Jin
TL;DR
This work tackles continual learning for vision–language retrieval by explicitly modeling cross-modal and cross-task interactions with a parameter-efficient framework. The proposed Low-rank Prompt Interaction (LPI) combines a shared low-rank decomposition for cross-modal prompts, hierarchical prompt alignment across layers, cross-modal prompt fusion, and cross-task prompt alignment driven by task semantic distances, trained with a composite loss that balances base task performance, cross-modal consistency, and cross-task discrimination. Key contributions include the Low-rank Interaction Decomposition (LID) to compress prompts, Hierarchical Prompt Alignment (HPA) to link prompts across layers, Cross-modal Prompt Fusion (CPF) for inter-layer collaboration, and Cross-task Prompt Alignment (CPA) leveraging task semantics; together they achieve superior recall with minimal parameter growth on image-text retrieval and referring expression comprehension under class-incremental settings. Empirical results show consistent gains over strong baselines and robust ablations confirm the value of each component, with analyses on computational cost and prompt visualization supporting practical viability and interpretability. The work advances efficient multimodal continual learning and offers a scalable approach for deploying vision–language models in evolving task streams; code is available at the provided repository.
Abstract
Research on continual learning in multi-modal tasks has been receiving increasing attention. However, most existing work overlooks the explicit cross-modal and cross-task interactions. In this paper, we innovatively propose the Low-rank Prompt Interaction (LPI) to address this general problem of multi-modal understanding, which considers both cross-modal and cross-task interactions. Specifically, as for the former, we employ multi-modal correlation modules for corresponding Transformer layers. Considering that the training parameters scale to the number of layers and tasks, we propose low-rank interaction-augmented decomposition to avoid memory explosion while enhancing the cross-modal association through sharing and separating common-specific low-rank factors. In addition, due to the multi-modal semantic differences carried by the low-rank initialization, we adopt hierarchical low-rank contrastive learning to ensure training robustness. As for the latter, we initially employ a visual analysis and identify that different tasks have clear distinctions in proximity. Therefore, we introduce explicit task contrastive constraints in the prompt learning process based on task semantic distances. Experiments on two retrieval tasks show performance improvements with the introduction of a minimal number of parameters, demonstrating the effectiveness of our method. Code is available at https://github.com/Kelvin-ywc/LPI.
