4 个月前

基于全局类别表示的少样本学习

基于全局类别表示的少样本学习

摘要

在本文中,我们提出了一种通过同时利用基础类和新类训练样本学习全局类别表示的方法来解决具有挑战性的少样本学习(Few-Shot Learning, FSL)问题。在每个训练轮次中,支持集计算出的类别均值通过注册模块与全局表示进行注册,从而生成用于基于查询集计算分类损失的已注册全局类别表示。尽管采用了与现有基于元学习的方法相似的轮次训练流程,但我们的方法从一开始就引入了新类别的训练样本,这一点显著不同。为了弥补新类别训练样本不足的问题,我们开发了一种有效的样本合成策略以避免过拟合。重要的是,通过基础-新类联合训练,我们的方法可以轻松扩展到更具实际意义但也更具挑战性的FSL场景,即广义FSL(Generalized Few-Shot Learning),在这种场景下测试数据的标签空间扩展到了基础类和新类。大量实验表明,我们的方法在这两种FSL设置中都是有效的。

代码仓库

tiangeluo/fsl-global
官方
pytorch
GitHub 中提及
huang-research-group/globalfsl2019
pytorch
GitHub 中提及

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于全局类别表示的少样本学习 | 论文 | HyperAI超神经