Prompt-based Distribution Alignment for Unsupervised Domain Adaptation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bai, Shuanghao, Zhang, Min, Zhou, Wanqi, Huang, Siteng, Luan, Zhirong, Wang, Donglin, Chen, Badong
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909083331723264
author Bai, Shuanghao
Zhang, Min
Zhou, Wanqi
Huang, Siteng
Luan, Zhirong
Wang, Donglin
Chen, Badong
author_facet Bai, Shuanghao
Zhang, Min
Zhou, Wanqi
Huang, Siteng
Luan, Zhirong
Wang, Donglin
Chen, Badong
contents Recently, despite the unprecedented success of large pre-trained visual-language models (VLMs) on a wide range of downstream tasks, the real-world unsupervised domain adaptation (UDA) problem is still not well explored. Therefore, in this paper, we first experimentally demonstrate that the unsupervised-trained VLMs can significantly reduce the distribution discrepancy between source and target domains, thereby improving the performance of UDA. However, a major challenge for directly deploying such models on downstream UDA tasks is prompt engineering, which requires aligning the domain knowledge of source and target domains, since the performance of UDA is severely influenced by a good domain-invariant representation. We further propose a Prompt-based Distribution Alignment (PDA) method to incorporate the domain knowledge into prompt learning. Specifically, PDA employs a two-branch prompt-tuning paradigm, namely base branch and alignment branch. The base branch focuses on integrating class-related representation into prompts, ensuring discrimination among different classes. To further minimize domain discrepancy, for the alignment branch, we construct feature banks for both the source and target domains and propose image-guided feature tuning (IFT) to make the input attend to feature banks, which effectively integrates self-enhanced and cross-domain features into the model. In this way, these two branches can be mutually promoted to enhance the adaptation of VLMs for UDA. We conduct extensive experiments on three benchmarks to demonstrate that our proposed PDA achieves state-of-the-art performance. The code is available at https://github.com/BaiShuanghao/Prompt-based-Distribution-Alignment.
format Preprint
id arxiv_https___arxiv_org_abs_2312_09553
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Prompt-based Distribution Alignment for Unsupervised Domain Adaptation
Bai, Shuanghao
Zhang, Min
Zhou, Wanqi
Huang, Siteng
Luan, Zhirong
Wang, Donglin
Chen, Badong
Computer Vision and Pattern Recognition
Recently, despite the unprecedented success of large pre-trained visual-language models (VLMs) on a wide range of downstream tasks, the real-world unsupervised domain adaptation (UDA) problem is still not well explored. Therefore, in this paper, we first experimentally demonstrate that the unsupervised-trained VLMs can significantly reduce the distribution discrepancy between source and target domains, thereby improving the performance of UDA. However, a major challenge for directly deploying such models on downstream UDA tasks is prompt engineering, which requires aligning the domain knowledge of source and target domains, since the performance of UDA is severely influenced by a good domain-invariant representation. We further propose a Prompt-based Distribution Alignment (PDA) method to incorporate the domain knowledge into prompt learning. Specifically, PDA employs a two-branch prompt-tuning paradigm, namely base branch and alignment branch. The base branch focuses on integrating class-related representation into prompts, ensuring discrimination among different classes. To further minimize domain discrepancy, for the alignment branch, we construct feature banks for both the source and target domains and propose image-guided feature tuning (IFT) to make the input attend to feature banks, which effectively integrates self-enhanced and cross-domain features into the model. In this way, these two branches can be mutually promoted to enhance the adaptation of VLMs for UDA. We conduct extensive experiments on three benchmarks to demonstrate that our proposed PDA achieves state-of-the-art performance. The code is available at https://github.com/BaiShuanghao/Prompt-based-Distribution-Alignment.
title Prompt-based Distribution Alignment for Unsupervised Domain Adaptation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.09553