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

doi:10.1109/TMM.2024.3377123

2024-01-01

PRE

Tao Zhou

Bo Han

Tongliang Liu (Tongliang Liu)

Jiajia Zhao

龚晨 (Chen Gong) *

DOI：10.1109/TMM.2024.3377123

原文链接

原文求助

摘要

En 中文

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.

Keyword:

Prototypes

Training

Semisupervised learning

Perturbation methods

Entropy

Dogs

Computer science

Contrastive learning

intersectional class mismatch

private assignment suppression

prototype learning