Confusing Pair Correction Based on Category Prototype for Domain Adaptation under Noisy Environments

Authors

  • Churan Zhi Beijing Jiaotong University, Beijing, China
  • Junbao Zhuo Institute of Computing Technology, Chinese Academy of Sciences, Beijing, China
  • Shuhui Wang Institute of Computing Technology, Chinese Academy of Sciences, Beijing, China

DOI:

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

Keywords:

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

Abstract

In this paper, we address unsupervised domain adaptation under noisy environments, which is more challenging and practical than traditional domain adaptation. In this scenario, the model is prone to overfitting noisy labels, resulting in a more pronounced domain shift and a notable decline in the overall model performance. Previous methods employed prototype methods for domain adaptation on robust feature spaces. However, these approaches struggle to effectively classify classes with similar features under noisy environments. To address this issue, we propose a new method to detect and correct confusing class pair. We first divide classes into easy and hard classes based on the small loss criterion. We then leverage the top-2 predictions for each sample after aligning the source and target domain to find the confusing pair in the hard classes. We apply label correction to the noisy samples within the confusing pair. With the proposed label correction method, we can train our model with more accurate labels. Extensive experiments confirm the effectiveness of our method and demonstrate its favorable performance compared with existing state-of-the-art methods. Our codes are publicly available at https://github.com/Hehxcf/CPC/.

Published

2024-03-24

How to Cite

Zhi, C., Zhuo, J., & Wang, S. (2024). Confusing Pair Correction Based on Category Prototype for Domain Adaptation under Noisy Environments. Proceedings of the AAAI Conference on Artificial Intelligence, 38(15), 17060-17068. https://doi.org/10.1609/aaai.v38i15.29650

Issue

Section

AAAI Technical Track on Machine Learning VI