计算机技术与发展2022,Vol.32Issue(9) :8-13.DOI:10.3969/j.issn.1673-629X.2022.09.002

基于密度加权原型网络的小样本学习算法

Few-shot Learning Based on Density-weighted Prototypical Network

华超 刘向阳
计算机技术与发展2022,Vol.32Issue(9) :8-13.DOI:10.3969/j.issn.1673-629X.2022.09.002

基于密度加权原型网络的小样本学习算法

Few-shot Learning Based on Density-weighted Prototypical Network

华超 1刘向阳1
扫码查看

作者信息

  • 1. 河海大学 理学院,江苏 南京 211100
  • 折叠

摘要

小样本学习主要研究从少量的样本快速学习和归纳,原型网络对该问题提出了解决方法.原型网络算法将样本通过网络框架嵌入到低维空间,在低维空间中将支持集的每个类别的样本均值作为各自的原型,计算查询集样本到每个类别原型的距离,进而得到损失函数,通过迭代更新损失函数来优化网络框架,使得经过网络框架后的样本类内的差异变小,类间的差异变大.在计算每个类别原型的过程中,将每个类的样本均值作为各自的原型,由于每个类别的样本数较少,在计算原型时会有不确定性,当样本中存在离群点时,会使计算出的原型偏离真实的原型,在判别新样本属于哪一类别时,很容易产生误差.针对该问题,提出密度加权原型网络,利用密度加权算法计算每个类别中样本的密度,对密度大的样本赋予较大的权值,密度小的样本赋予较小的权值,进行加权得到原型,来缓解以上的缺陷.在miniImageNet数据集上设置5-way 5-shot、20-way 5-shot实验,结果表明所提出密度加权原型网络相对于原型网络算法在识别正确率方面有所提升.

关键词

小样本学习/情节/密度加权/临界距离/距离矩阵

引用本文复制引用

基金项目

出版年

2022
计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子:0.621
ISSN:1673-629X
参考文献量2
段落导航相关论文