Exploring Sparse Visual Prompt for Domain Adaptive Dense Prediction

Authors

  • Senqiao Yang Peking University
  • Jiarui Wu Peking University
  • Jiaming Liu Peking University
  • Xiaoqi Li Peking University
  • Qizhe Zhang Peking University
  • Mingjie Pan Peking University
  • Yulu Gan Peking University
  • Zehui Chen University of Science and Technology of China
  • Shanghang Zhang Peking University

DOI:

https://doi.org/10.1609/aaai.v38i15.29569

Keywords:

ML: Transfer, Domain Adaptation, Multi-Task Learning, CV: Applications

Abstract

The visual prompts have provided an efficient manner in addressing visual cross-domain problems. Previous works introduce domain prompts to tackle the classification Test-Time Adaptation (TTA) problem by placing image-level prompts on the input and fine-tuning prompts for each target domain. However, since the image-level prompts mask out continuous spatial details in the prompt-allocated region, it will suffer from inaccurate contextual information and limited domain knowledge extraction, particularly when dealing with dense prediction TTA problems. To overcome these challenges, we propose a novel Sparse Visual Domain Prompts (SVDP) approach, which applies minimal trainable parameters (e.g., 0.1%) to pixels across the entire image and reserves more spatial information of the input. To better apply SVDP in extracting domain-specific knowledge, we introduce the Domain Prompt Placement (DPP) method to adaptively allocates trainable parameters of SVDP on the pixels with large distribution shifts. Furthermore, recognizing that each target domain sample exhibits a unique domain shift, we design Domain Prompt Updating (DPU) strategy to optimize prompt parameters differently for each sample, facilitating efficient adaptation to the target domain. Extensive experiments were conducted on widely-used TTA and continual TTA benchmarks, and our proposed method achieves state-of-the-art performance in both semantic segmentation and depth estimation tasks.

Published

2024-03-24

How to Cite

Yang, S., Wu, J., Liu, J., Li, X., Zhang, Q., Pan, M., Gan, Y., Chen, Z., & Zhang, S. (2024). Exploring Sparse Visual Prompt for Domain Adaptive Dense Prediction. Proceedings of the AAAI Conference on Artificial Intelligence, 38(15), 16334-16342. https://doi.org/10.1609/aaai.v38i15.29569

Issue

Section

AAAI Technical Track on Machine Learning VI