Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Negative Margin Matters: Understanding Margin in Few-shot Classification

About

This paper introduces a negative margin loss to metric learning based few-shot learning methods. The negative margin loss significantly outperforms regular softmax loss, and achieves state-of-the-art accuracy on three standard few-shot classification benchmarks with few bells and whistles. These results are contrary to the common practice in the metric learning field, that the margin is zero or positive. To understand why the negative margin loss performs well for the few-shot classification, we analyze the discriminability of learned features w.r.t different margins for training and novel classes, both empirically and theoretically. We find that although negative margin reduces the feature discriminability for training classes, it may also avoid falsely mapping samples of the same novel class to multiple peaks or clusters, and thus benefit the discrimination of novel classes. Code is available at https://github.com/bl0/negative-margin.few-shot.

Bin Liu, Yue Cao, Yutong Lin, Qi Li, Zheng Zhang, Mingsheng Long, Han Hu• 2020

Related benchmarks

TaskDatasetResultRank
Few-shot classificationMini-ImageNet
1-shot Acc63.85
175
Few-Shot Class-Incremental LearningminiImageNet (test)
Accuracy (Session 1)66.64
173
5-way Few-shot ClassificationMiniImagenet
Accuracy (5-shot)81.57
150
Few-shot classificationCUB (test)
Accuracy89.4
145
5-way Few-shot ClassificationMini-Imagenet (test)
1-shot Accuracy63.85
141
Few-Shot Class-Incremental LearningCIFAR100 (test)
Session 4 Top-1 Acc55.32
122
Few-shot Image ClassificationminiImageNet (test)
Accuracy80.94
111
Few-shot classificationMiniImagenet
5-way 5-shot Accuracy81.57
98
Few-shot classificationCUB--
96
5-way Few-shot ClassificationCUB
5-shot Acc89.4
95
Showing 10 of 21 rows

Other info

Code

Follow for update