少量样本下的图像分类算法研究
2020-02-10林汉斌
中文信息 2020年2期
林汉斌



摘 要:基于深度学习的计算机视觉研究都是基于大量图片数据样本,由于有时在工业上收集目标样本的成本很高,因此这大大限制了深度学习在计算机视觉中的应用。针对少量样本下的图片分类任务,本文提出一种新型的网络模型。并跟目前主流的深度学习网络比较。证明了我们的模型在少量样本下仍然具有很好分类精度。
中图分类号:TP391文献标识码:A文章编号:1003-9082(2020)02-000-01
一、卷积神经网络模型
卷积神经网络是一类特殊的人工神经網络,其主要特点是卷积运
算操作。理论上卷积神经网络的深度越深,模型的拟合能力越强。但研究发现,当随着卷积神经网络深度的加深,模型的拟合性能不升反降,这主要是因为如果网络的深度过深,卷积神经网络在反向传播时容易出现梯度消失或梯度爆炸。因此一些研究人员设计出了一种残差网络可以有效防止网络在反向传播时出现梯度消失或梯度爆炸。残差网络的核心创新点是将前面的输入也与后面的网络层连接。
二、少量样本下卷积神经网络模型的设计
由于少量样本下,样本的数据有限,因此应当设计一个网络使得网
络能有效的提取图片的深层次特征。为此,鉴于对残差网络,我们设计了一个基于残差块的注意力卷积神经网络(Res-attention)模型。结构如图1所示。
与传统卷积神经网络对残差块进行简单堆叠所不同的是,我们还加入了如虚线所示的结构。该结构先对前置网络输出的特征图先进行下采样,以获得更深层次的特征,接着进行上采样,将特征图缩放至原特征图一样大小。……
登录APP查看全文
