wvangansbeke / Unsupervised-Classification

SCAN: Learning to Classify Images without Labels, incl. SimCLR. [ECCV 2020]
https://arxiv.org/abs/2005.12320
Other
1.37k stars 268 forks source link

Results without false positives in the nearest neighbors #99

Closed skr998 closed 2 years ago

skr998 commented 2 years ago

Hello, thanks for your outstanding work.

There is a question about the "Results without false positives in the nearest neighbors" in your paper, because if you remove some samples in the nearest neighbors set, you can't obtain K nearest neighbors for every samples.

So how did you solve this problem?

wvangansbeke commented 2 years ago

Hi @skr998,

If all neighbors are incorrect, we only trained on augmented views for that specific image. Keep in mind, this is only to provide an approximation of the upperbound of our method since this requires lables.

skr998 commented 2 years ago

Thanks @wvangansbeke , you said "If all neighbors are incorrect, we only trained on augmented views for that specific image.", is that means you also only select 1 from the K neighbors set corrected by labels, to compute the scan loss?

Moreover, you evaluate the model every epoch, and select 5 neighbors to compute the loss together. Have you used the labels to correct this neighbor set in the evaluation phase every epoch? I guess you didn't.

wvangansbeke commented 2 years ago

Hi @skr998,