卷积神经网络CNN代码解析-matlab

整理文档很辛苦,赏杯茶钱您下走!

免费阅读已结束,点击下载阅读编辑剩下 ...

阅读已结束,您可以下载文档离线阅读编辑

资源描述

内蒙古大学计算机学院模式识别小组小强459125872@qq.com1卷积神经网络CNN代码解析deepLearnToolbox-master是一个深度学习matlab包,里面含有很多机器学习算法,如卷积神经网络CNN,深度信念网络DBN,自动编码AutoEncoder(堆栈SAE,卷积CAE)的作者是RasmusBergPalm(rasmusbergpalm@gmail.com)代码下载:这里我们介绍deepLearnToolbox-master中的CNN部分。DeepLearnToolbox-master中CNN内的函数:调用关系为:该模型使用了mnist的数字mnist_uint8.mat作为训练样本,作为cnn的一个使用样例,每个样本特征为一个28*28=的向量。内蒙古大学计算机学院模式识别小组小强459125872@qq.com2网络结构为:让我们来看看各个函数:一、Test_example_CNN:.................................................................................................................................................2三、cnntrain.m..................................................................................................................................................................5四、cnnff.m.......................................................................................................................................................................6五、cnnbp.m.....................................................................................................................................................................7五、cnnapplygrads.m......................................................................................................................................................10六、cnntest.m...................................................................................................................................................................11一、Test_example_CNN:Test_example_CNN:1设置CNN的基本参数规格,如卷积、降采样层的数量,卷积核的大小、降采样的降幅2cnnsetup函数初始化卷积核、偏置等3cnntrain函数训练cnn,把训练数据分成batch,然后调用3.1cnnff完成训练的前向过程,内蒙古大学计算机学院模式识别小组小强459125872@qq.com33.2cnnbp计算并传递神经网络的error,并计算梯度(权重的修改量)3.3cnnapplygrads把计算出来的梯度加到原始模型上去4cnntest函数,测试当前模型的准确率该模型采用的数据为mnist_uint8.mat,含有70000个手写数字样本其中60000作为训练样本,10000作为测试样本。把数据转成相应的格式,并归一化。设置网络结构及训练参数初始化网络,对数据进行批训练,验证模型准确率绘制均方误差曲线二、Cnnsetup.m该函数你用于初始化CNN的参数。设置各层的mapsize大小,初始化卷积层的卷积核、bias尾部单层感知机的参数设置*bias统一设置为0内蒙古大学计算机学院模式识别小组小强459125872@qq.com4权重设置为:-1~1之间的随机数/sqrt(6/(输入神经元数量+输出神经元数量))对于卷积核权重,输入输出为fan_in,fan_outfan_out=net.layers{l}.outputmaps*net.layers{l}.kernelsize^2;%卷积核初始化,1层卷积为1*6个卷积核,2层卷积一共6*12=72个卷积核。对于每个卷积输出featuremap,%fan_in=表示该层的一个输出map,所对应的所有卷积核,包含的神经元的总数。1*25,6*25fan_in=numInputmaps*net.layers{l}.kernelsize^2;fin=1*25or6*25fout=1*6*25or6*12*25net.layers{l}.k{i}{j}=(rand(net.layers{l}.kernelsize)-0.5)*2*sqrt(6/(fan_in+fan_out));1卷积降采样的参数初始化内蒙古大学计算机学院模式识别小组小强459125872@qq.com52尾部单层感知机的参数(权重和偏量)设置:三、cnntrain.m该函数用于训练CNN。生成随机序列,每次选取一个batch(50)个样本进行训练。批训练:计算50个随机样本的梯度,求和之后一次性更新到模型权重中。在批训练过程中调用:Cnnff.m完成前向过程Cnnbp.m完成误差传导和梯度计算过程Cnnapplygrads.m把计算出来的梯度加到原始模型上去内蒙古大学计算机学院模式识别小组小强459125872@qq.com6四、cnnff.m1、取得CNN的输入2、两次卷积核降采样层处理3、尾部单层感知机的数据处理,需要把subFeatureMap2连接成为一个(4*4)*12=192的向量,但是由于采用了50样本批训练的方法,subFeatureMap2被拼合成为一个192*50的特征向量fv;Fv作为单层感知机的输入,全连接的方式得到输出层内蒙古大学计算机学院模式识别小组小强459125872@qq.com7五、cnnbp.m该函数实现2部分功能,计算并传递误差,计算梯度1、计算误差和LossFunction2、计算尾部单层感知机的误差3、把单层感知机的输入层featureVector的误差矩阵,恢复为subFeatureMap2的4*4二维矩阵形式内蒙古大学计算机学院模式识别小组小强459125872@qq.com8插播一张图片:4、误差在特征提取网络【卷积降采样层】的传播如果本层是卷积层,它的误差是从后一层(降采样层)传过来,误差传播实际上是用降采样的反向过程,也就是降采样层的误差复制为2*2=4份。卷积层的输入是经过sigmoid处理的,所以,从降采样层扩充来的误差要经过sigmoid求导处理。如果本层是降采样层,他的误差是从后一层(卷积层)传过来,误差传播实际是用卷积的反向过程,也就是卷积层的误差,反卷积(卷积核转180度)卷积层的误差,原理参看插图。内蒙古大学计算机学院模式识别小组小强459125872@qq.com95、计算特征抽取层和尾部单层感知机的梯度内蒙古大学计算机学院模式识别小组小强459125872@qq.com10五、cnnapplygrads.m该函数完成权重修改,更新模型的功能1更新特征抽取层的权重weight+bias2更新末尾单层感知机的权重weight+bias内蒙古大学计算机学院模式识别小组小强459125872@qq.com11六、cnntest.m验证测试样本的准确率

1 / 11
下载文档,编辑使用

©2015-2020 m.777doc.com 三七文档.

备案号:鲁ICP备2024069028号-1 客服联系 QQ:2149211541

×
保存成功