• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2024, Vol. 46 ›› Issue (01): 132-141.

• Artificial Intelligence and Data Mining • Previous Articles     Next Articles

A focally discriminative loss for unsupervised domain adaptation method

WANG Shan-shan1,WANG Meng-zhu2,LUO Zhi-gang2   

  1. (1.The Key Laboratory of Intelligent Computing and Signal Processing,Ministry of Education,Anhui University,Hefei 230039;
    2.Laboratory of Parallel and Distributed Computing,College of Computer Science and Technology,
    National University of Defense Technology,Changsha 410073,China)
  • Received:2022-10-13 Revised:2022-11-28 Accepted:2024-01-25 Online:2024-01-25 Published:2024-01-15

Abstract: The maximum mean discrepancy (MMD), as a representative distribution metric between source domain and target domain, has been widely applied in unsupervised domain adaptation (UDA), where both domains follow different distributions, and the labels from source domain are merely available. However, MMD and its class-wise variants possibly ignore the intra-class compactness and inter-class separability, thus reducing discriminability of feature representation. This paper proposes a focally discriminative loss for unsupervised domain adaptation. This method endeavors to improve the discriminative ability of MMD from two aspects: (1) the weights are re-designed for MMD in order to align the distribution of relatively hard classes across domains; (2) a focally contrastive loss is explored to tradeoff the positive sample pairs and negative ones for better discrimination. The integration of both losses can not only make the intra-class features close, but also push away the inter-class features far from each other. Moreover, the improved loss is simple yet effective, and it can be extended to the network structure of the attention mechanism. Experiments on several domain adaptation datasets verify the effectiveness of the proposed method.

Key words: unsupervised domain adaptation, weighted maximum mean discrepancy, focally contrastive loss, attention mechanism