一种基于原型校正的小样本学习算法

摘要

现有的少样本学习(Few-Shot Learning)方法通常使用基于任务(episode)的训练策略训练模型，但在随机采样构建任务的过程中不可避免地存在采样偏差问题，质量较差的支持样本一定程度上导致模型难以收敛.本文提出了交互视图原型校正网络通过两种策略缓解这个问题，首先，模型利用全局信息来纠正每个任务中的支持样本特征，从而缓解随机采样导致的原型偏移问题；其次，文中提出了实例原型生成算法，该算法利用支持样本特征的局部视图原型集根据不同的查询样本实例生成对应的实例级原型，从而优化了使用支持样本全局特征作为类原型区分度不足的问题.模型在多个基准数据集上的进行了充分实验，实验结果验证了方法的有效性及鲁棒性.

关键词

小样本学习 / 原型校正 / 图像分类 / 原型集生成

基金资助

国家自然科学基金项目(62371235,62072246)资助