Reviewing the Forgotten Classes for Domain Adaptation of Black-Box Predictors

Authors

  • Shaojie Zhang Jilin University
  • Chun Shen Jilin University
  • Shuai Lü Jilin University
  • Zeyu Zhang Jilin University

DOI:

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

Keywords:

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

Abstract

For addressing the data privacy and portability issues of domain adaptation, Domain Adaptation of Black-box Predictors (DABP) aims to adapt a black-box source model to an unlabeled target domain without accessing both the source-domain data and details of the source model. Although existing DABP approaches based on knowledge distillation (KD) have achieved promising results, we experimentally find that these methods all have the minority class forgetting issue, which refers that the trained model completely forgets some minority classes. To address this issue, we propose a method called Reviewing the Forgotten Classes (RFC), which including two main modules. Firstly, we propose a simple but effective component called selection training (ST). ST selects classes that the model tends to forget according to the learning status of the model and obtains clean samples of the selected classes with the small-loss criterion for enhanced training. ST is orthogonal to previous methods and can effectively alleviate their minority class forgetting issue. Secondly, we find that neighborhood clustering (NC) can help the model learn more balanced than KD so that further alleviate the minority class forgetting issue. However, NC is based on the fact that target features from the source model already form some semantic structure, while DABP is unable to obtain the source model. Thus, we use KD and ST to warm up the target model to form a certain semantic structure. Overall, our method inherits the merits of both ST and NC, and achieves state of the art on three DABP benchmarks.

Published

2024-03-24

How to Cite

Zhang, S., Shen, C., Lü, S., & Zhang, Z. (2024). Reviewing the Forgotten Classes for Domain Adaptation of Black-Box Predictors. Proceedings of the AAAI Conference on Artificial Intelligence, 38(15), 16830-16837. https://doi.org/10.1609/aaai.v38i15.29624

Issue

Section

AAAI Technical Track on Machine Learning VI