基于改进匹配网络的单样本学习

摘要

当前深度学习是基于大量标注数据样本通过多层网络实现模型自动识别.然而,在很多特殊场景下,难以获取大量标注样本数据,小样本物体识别仍是深度学习下关键性的难题.针对这一问题,首先利用4层深度卷积神经网络(deep convolution neural network,DCNN)提取训练样本和测试样本的高层语义特征,然后基于改进的匹配网络分别采用双向LSTM和attLSTM算法对训练样本和测试样本深入提取更加关键和有用特征并进行编码,最后在平方欧氏距离上利用softmax非线性分类器对测试样本进行分类识别.实验通过Omniglot数据集对提出的改进模型进行测试,取得了非常好的效果.改进的模型即使在最复杂的20-way 1-shot情况下,依然能够达到93.2％的识别率,Vinyals的原创匹配网络模型在20-way 1-shot的情况下只能达到88.2％的识别率,与原创匹配网络模型相比,改进的模型在类别数更多而样本数较少的复杂场景下具有更好的识别效果.