4 个月前

f-VAEGAN-D2:一种适用于任意样本学习的特征生成框架

f-VAEGAN-D2:一种适用于任意样本学习的特征生成框架

摘要

当标记训练数据稀缺时,一种有前景的数据增强方法是利用未知类别的属性生成其视觉特征。为了学习卷积神经网络(CNN)特征的类别条件分布,这些模型依赖于图像特征和类别属性的配对。因此,它们无法利用大量未标记的数据样本。在本文中,我们提出了一种统一的特征生成框架来解决任意样本学习问题,即零样本和少样本学习,并且该框架既适用于归纳学习也适用于转换学习设置。我们开发了一种条件生成模型,结合了变分自编码器(VAE)和生成对抗网络(GANs)的优势,并通过一个无条件判别器学习未标记图像的边缘特征分布。实验结果表明,我们的模型为五个数据集(CUB、SUN、AWA 和 ImageNet)生成了高度区分性的 CNN 特征,并在任意样本学习中建立了新的最先进水平,即在归纳和转换(广义)零样本和少样本学习设置中。此外,我们还展示了所学特征的可解释性:通过将其逆变换回像素空间进行可视化,并通过生成文本论据来解释为何这些特征与特定标签相关联。

基准测试

基准方法指标
generalized-zero-shot-learning-on-sunf-VAEGAN
Harmonic mean: 41.3
zero-shot-learning-on-cub-200-2011f-VAEGAN-D2
average top-1 classification accuracy: 61.0
zero-shot-learning-on-sun-attributef-VAEGAN
average top-1 classification accuracy: 64.7

用 AI 构建 AI

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

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

Hyper Newsletters

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