Skip to content

Official code for the ECCV 2022 paper "Contrastive Vicinal Space for Unsupervised Domain Adaptation"

Notifications You must be signed in to change notification settings

NaJaeMin92/CoVi

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CoVi: Contrastive Vicinal Space for Unsupervised Domain Adaptation

PWC PWC

Contrastive Vicinal Space for Unsupervised Domain Adaptation
Jaemin Na, Dongyoon Han, Hyung Jin Chang, Wonjun Hwang
In ECCV 2022.


Abstract: Recent unsupervised domain adaptation methods have utilized vicinal space between the source and target domains. However, the equilibrium collapse of labels, a problem where the source labels are dom- inant over the target labels in the predictions of vicinal instances, has never been addressed. In this paper, we propose an instance-wise minimax strategy that minimizes the entropy of high uncertainty instances in the vicinal space to tackle the stated problem. We divide the vicinal space into two subspaces through the solution of the minimax prob- lem: contrastive space and consensus space. In the contrastive space, inter-domain discrepancy is mitigated by constraining instances to have contrastive views and labels, and the consensus space reduces the confusion between intra domain categories. The effectiveness of our method is demonstrated on public benchmarks, including Office-31, Office-Home, and VisDA-C, achieving state-of-the-art performances. We further show that our method outperforms the current state-of-the-art methods on PACS, which indicates that our instance-wise approach works well for multi-source domain adaptation as well.

Introduction

Video: Click the figure to watch the explanation video.

YouTube

Requirements

  • Linux
  • Python >= 3.7
  • PyTorch == 1.7.1
  • CUDA (must be a version supported by the pytorch version)

Getting Started

Training process.

Below we provide an example for training a CoVi on Office-31.

python main.py \
-gpu 0
-source amazon \
-target dslr \
-db_path $DATASET_PATH \
-baseline_path $BASELINE_PATH
  • $DATASET_PATH denotes the location where datasets are installed.
  • $BASELINE_PATH requires the path where pretrained models (DANN, MSTN, etc.) are stored.
  • For DANN, the following code may be used: pytorch-DANN

Contact

For questions, please contact: [email protected]

Citation

If you use this code in your research, please cite:

@article{na2021contrastive,
  title={Contrastive Vicinal Space for Unsupervised Domain Adaptation},
  author={Na, Jaemin and Han, Dongyoon and Chang, Hyung Jin and Hwang, Wonjun},
  journal={arXiv preprint arXiv:2111.13353},
  year={2021}
}

About

Official code for the ECCV 2022 paper "Contrastive Vicinal Space for Unsupervised Domain Adaptation"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages