主要内容

rmspropupdate

使用均方根传播(RMSProp)更新参数

描述

使用均方根传播(RMSProp)算法更新自定义训练循环中的网络可学习参数。

请注意

该函数应用RMSProp优化算法来更新自定义训练循环中的网络参数,该循环使用定义为的网络dlnetwork对象或模型函数。如果你想训练一个定义为a的网络数组或作为LayerGraph,使用以下函数:

例子

netUpdatedaverageSqGrad= rmspropupdate(研究生averageSqGrad更新网络的可学习参数使用RMSProp算法。在训练循环中使用此语法可迭代更新定义为的网络dlnetwork对象。

例子

参数个数averageSqGrad= rmspropupdate(参数个数研究生averageSqGrad中更新可学习参数参数个数使用RMSProp算法。在训练循环中使用此语法迭代更新使用函数定义的网络的可学习参数。

例子

___= rmspropupdate(___learnRatesqGradDecayε还指定除了前面语法中的输入参数外,用于全局学习率、平方梯度衰减和小常数epsilon的值。

例子

全部折叠

执行一个全局学习率为的均方根传播更新步骤0.05的平方梯度衰减因子0.95

将参数和参数梯度创建为数值数组。

Params = rand(3,3,4);Grad = ones(3,3,4);

初始化第一次迭代的平均梯度平方。

averageSqGrad = [];

指定全局学习率和平方梯度衰减因子的自定义值。

learnRate = 0.05;sqGradDecay = 0.95;

使用更新可学习参数rmspropupdate

[params,averageSqGrad] = rmspropupdate(params,grad,averageSqGrad,learnRate,sqGradDecay);

使用rmspropupdate使用均方根传播(RMSProp)算法训练网络。

负荷训练数据

加载数字训练数据。

[XTrain,TTrain] = digitTrain4DArrayData;类=类别(TTrain);numClasses = nummel(类);

定义网络

属性定义网络体系结构并指定平均图像值的意思是选项在图像输入层。

图层= [imageInputLayer([28 28 1],“的意思是”reluLayer卷积2dlayer (3,20),“填充”,1) relullayer卷积2dlayer (3,20,“填充”,1) reluLayer fullyConnectedLayer(numClasses) softmaxLayer];

创建一个dlnetwork对象。

Net = dlnetwork(layers);

定义模型损失函数

创建helper函数modelLoss,在示例的末尾列出。函数的参数为dlnetwork对象和带有相应标签的小批输入数据,并返回损失和损失相对于可学习参数的梯度。

指定培训项目

指定在培训期间使用的选项。

miniBatchSize = 128;numEpochs = 20;numObservations = numel(TTrain);numIterationsPerEpoch = floor(numObservations./miniBatchSize);

列车网络的

初始化平均梯度的平方。

averageSqGrad = [];

计算训练进度监控器的总迭代次数。

numIterations = nummepochs * numIterationsPerEpoch;

初始化TrainingProgressMonitor对象。因为计时器在创建监视器对象时开始,所以请确保创建的对象接近训练循环。

monitor = trainingProgressMonitor(指标=“损失”信息=“时代”包含=“迭代”);

使用自定义训练循环训练模型。对于每个纪元,洗牌数据并在小批量数据上循环。方法更新网络参数rmspropupdate函数。在每次迭代结束时,显示训练进度。

在GPU上训练(如果有的话)。使用GPU需要并行计算工具箱™和受支持的GPU设备。万博1manbetx有关受支持设备的信息,请参见万博1manbetxGPU计算要求(并行计算工具箱)

培训网络。

迭代= 0;Epoch = 0;epoch < numEpochs && ~monitor。停止epoch = epoch + 1;% Shuffle数据。idx = randperm(数字(TTrain));XTrain = XTrain(:,:,:,idx);TTrain = TTrain(idx);I = 0;i < numIterationsPerEpoch && ~monitor。Stop i = i + 1;迭代=迭代+ 1;读取小批数据并将标签转换为虚拟标签%变量。idx = (i-1)*miniBatchSize+1:i*miniBatchSize;X = XTrain(:,:,:,idx);T = 0 (numClasses,miniBatchSize,“单身”);T(c,TTrain(idx)==classes(c)) = 1;结束将小批数据转换为大数组。X = dlarray(single(X),“SSCB”);如果在GPU上训练,则将数据转换为gpuArray。如果canUseGPU X = gpuArray(X);结束计算模型损失和梯度使用dlfeval和% modelLoss函数。[loss,gradients] = dlfeval(@modelLoss,net,X,T);使用RMSProp优化器更新网络参数。[net,averageSqGrad] = rmspropupdate(net,gradients,averageSqGrad);更新培训进度监视器。recordMetrics(监控、迭代损失=损失);updateInfo(监视、时代=时代+“的”+ numEpochs);班长。进度= 100 * iteration/numIterations;结束结束

测试网络

通过比较测试集上的预测与真实标签来测试模型的分类准确性。

[XTest,TTest] = digitTest4DArrayData;

将数据转换为adlarray带尺寸格式“SSCB”.对于GPU预测,也将数据转换为agpuArray

XTest = dlarray(XTest,“SSCB”);如果canUseGPU XTest = gpuArray(XTest);结束

对图像进行分类dlnetwork对象时,使用预测计算并找出得分最高的课程。

YTest =预测(net,XTest);[~,idx] = max(extractdata(YTest),[],1);YTest = classes(idx);

评估分类准确率。

精度=平均值(YTest==TTest)
准确度= 0.9902

模型损失函数

辅助函数modelLoss需要一个dlnetwork对象和一小批输入数据X有相应的标签T,并返回损失以及损失相对于中可学习参数的梯度.要自动计算梯度,请使用dlgradient函数。

函数[loss,gradients] = modelLoss(net,X,T) Y = forward(net,X);损失=交叉熵(Y,T);gradients = dlgradient(loss,net.Learnables);结束

输入参数

全部折叠

网络,指定为dlnetwork对象。

函数更新可学的的属性dlnetwork对象。网可学的是一个包含三个变量的表:

  • -层名,指定为字符串标量。

  • 参数—参数名称,指定为字符串标量。

  • 价值参数的值,指定为包含dlarray

输入参数研究生一定是和?一样形式的表网可学的

网络可学习参数,指定为dlarray、数字数组、单元格数组、结构体或表。

如果你指定参数个数作为一个表,它必须包含以下三个变量。

  • -层名,指定为字符串标量。

  • 参数—参数名称,指定为字符串标量。

  • 价值参数的值,指定为包含dlarray

你可以指定参数个数作为使用单元格数组、结构、表或嵌套单元格数组或结构的网络可学习参数的容器。单元格数组、结构或表中的可学习参数必须为dlarray或数据类型的数值

输入参数研究生必须提供与?完全相同的数据类型、顺序和字段(用于结构)或变量(用于表)参数个数

数据类型:||结构体|表格|细胞

损耗的梯度,指定为adlarray、数字数组、单元格数组、结构体或表。

确切的形式研究生取决于输入网络或可学习参数。下表显示了所需的格式研究生可能的输入rmspropupdate

输入 可学的参数 梯度
表格网可学的包含参数,价值变量。的价值变量由单元格数组组成,单元格数组包含每个可学习参数dlarray 表具有相同的数据类型、变量和排序网可学的研究生必须有一个价值由包含每个可学习参数梯度的单元格数组组成的变量。
参数个数 dlarray dlarray使用相同的数据类型和顺序参数个数
数字数组 具有相同数据类型和顺序的数值数组参数个数
单元阵列 单元格数组,具有相同的数据类型、结构和顺序参数个数
结构 结构,具有相同的数据类型、字段和排序参数个数
参数,价值变量。的价值变量必须由单元格数组组成,其中包含每个可学习参数dlarray 表具有相同的数据类型、变量和排序参数个数研究生必须有一个价值由包含每个可学习参数梯度的单元格数组组成的变量。

你可以获得研究生从电话到dlfeval对包含调用的函数求值dlgradient.有关更多信息,请参见在深度学习工具箱中使用自动区分

参数梯度平方的移动平均,指定为空数组,adlarray、数字数组、单元格数组、结构体或表。

确切的形式averageSqGrad取决于输入网络或可学习参数。下表显示了所需的格式averageSqGrad可能的输入rmspropupdate

输入 可学的参数 平均梯度平方
表格网可学的包含参数,价值变量。的价值变量由单元格数组组成,单元格数组包含每个可学习参数dlarray 表具有相同的数据类型、变量和排序网可学的averageSqGrad必须有一个价值由包含每个可学习参数的平均梯度平方的单元格数组组成的变量。
参数个数 dlarray dlarray使用相同的数据类型和顺序参数个数
数字数组 具有相同数据类型和顺序的数值数组参数个数
单元阵列 单元格数组,具有相同的数据类型、结构和顺序参数个数
结构 结构,具有相同的数据类型、字段和排序参数个数
参数,价值变量。的价值变量必须由单元格数组组成,其中包含每个可学习参数dlarray 表具有相同的数据类型、变量和排序参数个数averageSqGrad必须有一个价值由包含每个可学习参数的平均梯度平方的单元格数组组成的变量。

如果你指定averageSqGrad作为一个空数组,函数假设之前没有渐变,并以与一系列迭代中的第一次更新相同的方式运行。要迭代地更新可学习参数,请使用averageSqGrad的前一次调用的输出rmspropupdate随着averageSqGrad输入。

全局学习率,指定为正标量。的默认值learnRate0.001

如果指定网络参数为adlnetwork,每个参数的学习率为全局学习率乘以网络层中定义的相应学习率因子属性。

梯度衰减因子的平方,指定为之间的正标量0而且1.的默认值sqGradDecay0.9

用于防止被零除错误的小常数,指定为正标量。的默认值ε1 e-8

输出参数

全部折叠

网络,返回为adlnetwork对象。

函数更新可学的的属性dlnetwork对象。

更新网络可学习参数,返回为dlarray类型的数字数组、单元格数组、结构体或表价值变量,包含网络更新后的可学习参数。

更新的参数梯度平方的移动平均值,返回为adlarray、数字数组、单元格数组、结构体或表。

更多关于

全部折叠

RMSProp

该函数使用均方根传播算法来更新可学习参数。有关更多信息,请参阅下面的RMSProp算法的定义随机梯度下降trainingOptions参考页面。

扩展功能

版本历史

R2019b引入

全部展开