Class-Wise Contrastive Prototype Learning for Semi-Supervised Classification Under Intersectional Class Mismatch

Mingyu Li, Tao Zhou*, Bo Han, Tongliang Liu, Xinkai Liang, Jiajia Zhao, Chen Gong*

*Corresponding author for this work

Research output: Contribution to journalJournal articlepeer-review

Abstract

Traditional Semi-Supervised Learning (SSL) classification methods focus on leveraging unlabeled data to improve the model performance under the setting where labeled set and unlabeled set share the same classes. Nevertheless, the above-mentioned setting is often inconsistent with many real-world circumstances. Practically, both the labeled set and unlabeled set often hold some individual classes, leading to an intersectional class-mismatch setting for SSL. Under this setting, existing SSL methods are often subject to performance degradation attributed to these individual classes. To solve the problem, we propose a Class-wise Contrastive Prototype Learning (CCPL) framework, which can properly utilize the unlabeled data to improve the SSL classification performance. Specifically, we employ a supervised prototype learning strategy and a class-wise contrastive separation strategy to construct a prototype for each known class. To reduce the influence of the individual classes in unlabeled set (i.e., out-of-distribution classes), each unlabeled example can be weighted reasonably based on the prototypes during classifier training, which helps to weaken the negative influence caused by out-of-distribution classes. To reduce the influence of the individual classes in labeled set (i.e., private classes), we present a private assignment suppression strategy to suppress the improper assignments of unlabeled examples to the private classes with the help of the prototypes. Experimental results on four benchmarks and one real-world dataset show that our CCPL has a clear advantage over fourteen representative SSL methods as well as two supervised learning methods under the intersectional class-mismatch setting.
Original languageEnglish
Pages (from-to)8145-8156
Number of pages12
JournalIEEE Transactions on Multimedia
Volume26
DOIs
Publication statusPublished - 18 Mar 2024

User-Defined Keywords

  • Contrastive learning
  • intersectional class mismatch
  • private assignment suppression
  • prototype learning
  • semi-supervised learning

Fingerprint

Dive into the research topics of 'Class-Wise Contrastive Prototype Learning for Semi-Supervised Classification Under Intersectional Class Mismatch'. Together they form a unique fingerprint.

Cite this