4 个月前

特征生成网络用于零样本学习

特征生成网络用于零样本学习

摘要

由于已见类别和未见类别之间存在极端的训练数据不平衡问题,现有的大多数最先进方法在应对具有挑战性的广义零样本学习任务时无法取得令人满意的结果。为了解决未见类别标注样本的需求,我们提出了一种新颖的生成对抗网络(GAN),该网络能够在类级别的语义信息条件下合成卷积神经网络(CNN)特征,从而提供了一条从类别的语义描述直接到条件特征分布的捷径。我们提出的这种方法结合了Wasserstein GAN和分类损失,能够生成足够有区分度的CNN特征以训练softmax分类器或任何多模态嵌入方法。实验结果表明,在零样本学习和广义零样本学习两种设置下,我们的方法在五个具有挑战性的数据集——CUB、FLO、SUN、AWA和ImageNet上均显著提高了准确性。

代码仓库

基准测试

基准方法指标
generalized-zero-shot-learning-on-sunf-CLSWGAN
Harmonic mean: 39.4
zero-shot-learning-on-cub-200-2011f-CLSWGAN
average top-1 classification accuracy: 57.3
zero-shot-learning-on-sun-attributef-CLSWGAN
average top-1 classification accuracy: 60.8

用 AI 构建 AI

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

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

Hyper Newsletters

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