Command Palette
Search for a command to run...

摘要
本文提出了一种针对基于度量学习的少样本学习方法的负边距损失(negative margin loss)。该损失函数显著优于传统的Softmax损失,在三个标准的少样本分类基准上取得了当前最优的准确率,且仅需极少的额外设计技巧。这一结果与度量学习领域普遍采用的零边距或正边距的做法形成了鲜明对比。为了探究负边距损失在少样本分类任务中表现优异的原因,本文从实验和理论两个层面,系统分析了不同边距设置下所学习特征在训练类别与新类别上的可区分性。研究发现,尽管负边距会降低对训练类别的特征可区分性,但它能够有效避免同一新类别样本被错误地映射至多个峰值或聚类中心,从而提升对新类别的判别能力。相关代码已开源,地址为:https://github.com/bl0/negative-margin.few-shot。
代码仓库
bl0/negative-margin.few-shot
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 | 
|---|---|---|
| few-shot-image-classification-on-cub-200-5 | Neg-Margin | Accuracy: 89.40  | 
| few-shot-image-classification-on-cub-200-5-1 | Neg-Margin | Accuracy: 72.66  | 
| few-shot-image-classification-on-mini-1 | Neg-Margin | Accuracy: 63.85  | 
| few-shot-image-classification-on-mini-10 | Neg-Margin | Accuracy: 69.30  |