文献类型：期刊文章

作　　者：段萌[1] 王功鹏[1] 牛常勇[1]

机构地区：[1]郑州大学信息工程学院,河南郑州450001

出　　处：《计算机工程与设计》

基　　金：国家自然科学基金项目(60901078);郑州市科技计划基金项目(10LJRC189)

年　　份：2018

卷　　号：39

期　　号：1

起止页码：224-229

语　　种：中文

收录情况：BDHX、BDHX2017、CSA、CSA-PROQEUST、IC、INSPEC、JST、RCCSE、ZGKJHX、核心刊

摘　　要：为提高仅包含少量训练样本的图像识别准确率,利用卷积神经网络作为图像的特征提取器,提出一种基于卷积神经网络的小样本图像识别方法。在原始小数据集中引入数据增强变换,扩充数据样本的范围;在此基础上将大规模数据集上的源预训练模型在目标小数据集上进行迁移训练,提取除最后全连接层之外的模型权重和图像特征;结合源预训练模型提取的特征,采用层冻结方法,微调目标小规模数据集上的卷积模型,得到最终分类识别结果。实验结果表明,该方法在小规模图像数据集的识别问题中具有较高的准确率和鲁棒性。

关 键 词：卷积神经网络 模型预训练  模型微调  迁移学习  小样本

分 类 号：TP391.41]