A Separation and Alignment Framework for Black-Box Domain Adaptation

Authors

  • Mingxuan Xia School of Software Technology, Zhejiang University State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Junbo Zhao State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Gengyu Lyu Faculty of Information Technology, Beijing University of Technology
  • Zenan Huang State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Tianlei Hu State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Gang Chen State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Haobo Wang School of Software Technology, Zhejiang University State Key Laboratory of Blockchain and Data Security, Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v38i14.29532

Keywords:

ML: Transfer, Domain Adaptation, Multi-Task Learning, ML: Unsupervised & Self-Supervised Learning

Abstract

Black-box domain adaptation (BDA) targets to learn a classifier on an unsupervised target domain while assuming only access to black-box predictors trained from unseen source data. Although a few BDA approaches have demonstrated promise by manipulating the transferred labels, they largely overlook the rich underlying structure in the target domain. To address this problem, we introduce a novel separation and alignment framework for BDA. Firstly, we locate those well-adapted samples via loss ranking and a flexible confidence-thresholding procedure. Then, we introduce a novel graph contrastive learning objective that aligns under-adapted samples to their local neighbors and well-adapted samples. Lastly, the adaptation is finally achieved by a nearest-centroid-augmented objective that exploits the clustering effect in the feature space. Extensive experiments demonstrate that our proposed method outperforms best baselines on benchmark datasets, e.g. improving the averaged per-class accuracy by 4.1% on the VisDA dataset. The source code is available at: https://github.com/MingxuanXia/SEAL.

Published

2024-03-24

How to Cite

Xia, M., Zhao, J., Lyu, G., Huang, Z., Hu, T., Chen, G., & Wang, H. (2024). A Separation and Alignment Framework for Black-Box Domain Adaptation. Proceedings of the AAAI Conference on Artificial Intelligence, 38(14), 16005-16013. https://doi.org/10.1609/aaai.v38i14.29532

Issue

Section

AAAI Technical Track on Machine Learning V