4 个月前

学习比较:用于少样本学习的关系网络

学习比较:用于少样本学习的关系网络

摘要

我们提出了一种概念上简单、灵活且通用的少样本学习框架,该框架要求分类器在每个类别仅有少量样本的情况下学会识别新类别。我们的方法称为关系网络(Relation Network, RN),是从零开始端到端训练的。在元学习过程中,它学会了学习一种深度距离度量,用于在每个模拟少样本设置的episode中比较少量图像。一旦训练完成,关系网络能够通过计算查询图像与每个新类别的少量示例之间的关系分数来对新类别的图像进行分类,而无需进一步更新网络。除了在少样本学习任务中提供改进的性能外,我们的框架还容易扩展到零样本学习任务。广泛的实验表明,在五个基准数据集上的测试结果证明了我们这一简单方法在这两个任务中均具有统一且有效的性能。

代码仓库

基准测试

基准方法指标
few-shot-image-classification-on-cifar-fs-5-1Relation Networks*
Accuracy: 69.3
few-shot-image-classification-on-cub-200-5Relation Net
Accuracy: 65.32
few-shot-image-classification-on-cub-200-5-1Relation Net
Accuracy: 50.44
few-shot-image-classification-on-meta-datasetRelation Networks
Accuracy: 53.315
few-shot-image-classification-on-meta-dataset-1Relation Networks
Mean Rank: 11.8
few-shot-image-classification-on-mini-12Relation Networks
Accuracy: 34.9
few-shot-image-classification-on-mini-13Relation Networks
Accuracy: 47.9
few-shot-image-classification-on-mini-2Relation Net (Sung et al., 2018)
Accuracy: 50.4
few-shot-image-classification-on-mini-5RelationNet (Sung et al., 2018)
Accuracy: 42.91
few-shot-image-classification-on-omniglot-1-1Relation Net
Accuracy: 97.6%
few-shot-image-classification-on-omniglot-1-2Relation Net
Accuracy: 99.6
few-shot-image-classification-on-omniglot-5-1Relation Net
Accuracy: 99.1%
few-shot-image-classification-on-omniglot-5-2Relation Net
Accuracy: 99.8
few-shot-image-classification-on-tiered-2Relation Networks
Accuracy: 36.3
few-shot-image-classification-on-tiered-3Relation Networks
Accuracy: 58.0
image-classification-on-tiered-imagenet-5-wayRelation Net
Accuracy: 71.31

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
学习比较:用于少样本学习的关系网络 | 论文 | HyperAI超神经