Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Label Propagation for Deep Semi-supervised Learning

About

Semi-supervised learning is becoming increasingly important because it can combine data carefully labeled by humans with abundant unlabeled data to train deep neural networks. Classic methods on semi-supervised learning that have focused on transductive learning have not been fully exploited in the inductive framework followed by modern deep learning. The same holds for the manifold assumption---that similar examples should get the same prediction. In this work, we employ a transductive label propagation method that is based on the manifold assumption to make predictions on the entire dataset and use these predictions to generate pseudo-labels for the unlabeled data and train a deep neural network. At the core of the transductive method lies a nearest neighbor graph of the dataset that we create based on the embeddings of the same network.Therefore our learning process iterates between these two steps. We improve performance on several datasets especially in the few labels regime and show that our work is complementary to current state of the art.

Ahmet Iscen, Giorgos Tolias, Yannis Avrithis, Ondrej Chum• 2019

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10--
507
Image ClassificationMini-Imagenet (test)
Top-1 Accuracy29.71
187
Image ClassificationCIFAR-10 4000 labels
Error Rate12.69
68
Image ClassificationCIFAR-10 4,000 labels (test)
Test Error Rate10.61
62
Image ClassificationCIFAR-100 10k labels
Test Error Rate35.92
44
Image ClassificationSVHN (test)
Top-1 Accuracy19.71
29
Image ClassificationCIFAR-10 4k labels
Error Rate11.82
23
Image ClassificationCIFAR-100
Top-1 Error Rate32.5
18
Image ClassificationCIFAR-100 4k labels
Error Rate43.73
13
Image ClassificationCIFAR-10 1k labels (test)
Test Error Rate16.93
9
Showing 10 of 23 rows

Other info

Follow for update