Novel Category Discovery Across Domains with Contrastive Learning and Adaptive Classifier

Shengyuan Yu,Yangyang Huang,Tianwen Yang,Jinhao Lin,Ronghua Luo
DOI: https://doi.org/10.1109/ijcnn60899.2024.10650890
2024-01-01
Abstract:Unsupervised domain adaptation (UDA) has been developed to transfer knowledge from a fully labeled source domain to an unlabeled target domain, mitigating domain shift. The presence of category shift has prompted the research in open-oet domain adaptation (ODA) and universal domain adaptation (UNDA). However, current ODA and UNDA models often treat all novel classes as a unified unknown class or set a priori number for novel classes, which fall short in the novel category discovery in domain adaptation (NCD in DA) problem. Accurately detecting the actual number of novel classes and learning their discriminative features in the DA process are two critical issues underexplored in the problem of NCD in DA. To address these issues, we propose a two-stage framework named Contrastive Adaptive Network (CAN) for the NCD in ODA and UNDA tasks. In Stage 1, we employ two types of contrastive learning for representation learning followed by k-means clustering with various values of k to search for the most probable novel class number. In Stage 2, we propose a known-unknown mapping protocol and group all the classes into several "known-unknown" pairs. Based on them, we design a novel Adaptive Classifier, the size of which is set according to the novel class number predicted in Stage 1. Then we design two tailored loss functions for the training of Adaptive Classifier to learn the discriminative features of novel classes while mitigating the domain shift of common classes, with source and target domain, respectively. Extensive experimental results demonstrate that CAN outperforms the performance of existing SOTA methods on ODA and UNDA problems and showcases significant improvements in the discrimination of novel class features. Implementation is available at https://github.com/yushengyuann/CAN.
What problem does this paper attempt to address?