关键词:
小样本学习
数据增强
类内变化
类辨别特征
个体特征
摘要:
小样本学习致力于通过极少数量的训练样本,甚至一个样本来实现对新类数据的分类。面对这种挑战,数据增强成为小样本学习中一种直接而有效的解决方法,但是确保增强数据的多样性和可辨别性是数据增强的关键。为此,提出一种基于迁移基类类内变化的两阶段数据增强方法,分为特征学习和小样本学习阶段。在特征学习阶段,模型通过自监督任务学习基类数据的个体特征表达,有监督任务则学习类辨别特征,模型通过这两种特征获得基类数据的类内变化并建模基类的类内变化分布。在小样本学习阶段,模型从基类的类内变化分布中采样与任务相关的类内变化信息并添加到小样本特征中,以实现增强小样本数据的目的。实验结果表明,在5-way 1-shot情况下,所提方法在miniImageNet、tieredImageNet和CUB数据集上的分类性能相较于基线模型提升了4~7百分点,在5-way 5-shot情况下提升了3~7百分点,相较于其他数据增强方法,也展现了具有竞争力的性能,这表明生成的增强数据在保持可辨别性的同时增强小样本数据的多样性,并验证了该方法的可行性和有效性。