arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.01784cs.CVcs.AIcs.LG
收藏

SB-MTL:基于分数的跨领域少样本学习Meta迁移学习

SB-MTL: Score-based Meta Transfer-Learning for Cross-Domain Few-Shot Learning

Princeton University(普林斯顿大学);Massachusetts Institute of Technology(麻省理工学院);Pensees Pte Ltd(Pensees私人有限公司)

John Cai, Bill Cai, Sheng Mei Shen

摘要:

虽然许多深度学习方法在分别解决领域自适应和少样本学习问题方面取得了巨大成功,但能够在跨领域少样本学习(CD-FSL)中联合解决这两个问题的方法要少得多。这个问题在典型的常见计算机视觉应用的急剧域移位下加剧。在本文中,我们提出了一种新的,灵活的和有效的方法来解决CD-FSL问题。我们的方法,称为基于分数的Meta迁移学习(SB-MTL),结合迁移学习和元学习,通过使用MAML优化的特征编码器和基于分数的图神经网络。首先,我们有一个特征编码器,它具有被设计为微调的特定层。为此,我们应用一阶MAML算法来找到好的初始化。其次,我们不是在微调后直接获取分类分数,而是通过将pre-softmax分类分数映射到度量空间来将分数解释为坐标。随后,我们应用图神经网络将标签信息从支持集传播到基于分数的度量空间中的查询集。我们在跨域少样本学习(BSCD-FSL)基准的更广泛的研究上测试我们的模型,该基准包括一系列与miniImagenet源域具有高度不同的差异的目标域。我们观察到在5,20和50杆,并在四个目标域的准确性显着改善。在平均准确率方面,我们的模型比以前的迁移学习方法高出5.93%,比以前的元学习方法高出14.28%。

英文摘要:

While many deep learning methods have seen significant success in tackling the problem of domain adaptation and few-shot learning separately, far fewer methods are able to jointly tackle both problems in Cross-Domain Few-Shot Learning (CD-FSL). This problem is exacerbated under sharp domain shifts that typify common computer vision applications. In this paper, we present a novel, flexible and effective method to address the CD-FSL problem. Our method, called Score-based Meta Transfer-Learning (SB-MTL), combines transfer-learning and meta-learning by using a MAML-optimized feature encoder and a score-based Graph Neural Network. First, we have a feature encoder with specific layers designed to be fine-tuned. To do so, we apply a first-order MAML algorithm to find good initializations. Second, instead of directly taking the classification scores after fine-tuning, we interpret the scores as coordinates by mapping the pre-softmax classification scores onto a metric space. Subsequently, we apply a Graph Neural Network to propagate label information from the support set to the query set in our score-based metric space. We test our model on the Broader Study of Cross-Domain Few-Shot Learning (BSCD-FSL) benchmark, which includes a range of target domains with highly varying dissimilarity to the miniImagenet source domain. We observe significant improvements in accuracy across 5, 20 and 50 shot, and on the four target domains. In terms of average accuracy, our model outperforms previous transfer-learning methods by 5.93% and previous meta-learning methods by 14.28%.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑