这个例子展示了如何训练一个使用自定义学习率计划分类手写数字的网络。
如果trainingOptions
不提供您需要的选项(例如,自定义学习速率计划),那么您可以使用自动区分定义自己的自定义训练循环。
此示例训练网络对手写数字进行分类基于时间的衰减学习率计划:对于每次迭代,求解器使用
,在那里t是迭代数,
初始学习率,和k就是衰变。
负荷训练数据
方法将数字数据加载为图像数据存储imageDatastore
函数并指定包含图像数据的文件夹。
将数据划分为训练集和验证集。方法留出10%的数据用于验证splitEachLabel
函数。
本例中使用的网络需要大小为28 × 28 × 1的输入图像。要自动调整训练图像的大小,请使用增强图像数据存储。指定要对训练图像执行的附加增强操作:在水平轴和垂直轴上随机将图像平移到5个像素。数据增强有助于防止网络过度拟合和记忆训练图像的确切细节。
若要自动调整验证图像的大小,而不执行进一步的数据增强,请使用增强图像数据存储,而不指定任何额外的预处理操作。
确定训练数据中的类数。
定义网络
定义图像分类的网络。
创建一个dlnetwork
对象从图层图。
dlnet = dlnetwork with properties: Layers: [12×1 nnet.cnn.layer.Layer] Connections: [11×2 table] Learnables: [14×3 table] State: [6×3 table] InputNames: {'input'} OutputNames: {'softmax'}
定义模型梯度函数
创建函数modelGradients
,列在示例末尾,它需要adlnetwork
对象,一个带有相应标签的小批量输入数据,并返回相对于网络中可学习参数的损失梯度和相应的损失。
指定培训项目
用128个小批量训练10个epoch。
指定用于SGDM优化的选项。指定初始学习速率为0.01,衰减为0.01,动量为0.9。
火车模型
创建一个minibatchqueue
对象,该对象在训练期间处理和管理小批量图像。对于每个小批量:
使用自定义小批量预处理功能preprocessMiniBatch
(在本例结束时定义)将标签转换为单热编码变量。
用尺寸标签格式化图像数据“SSCB”
(空间,空间,通道,批次)。默认情况下,minibatchqueue
对象将数据转换为dlarray
具有基础类型的对象单
.不要向类标签添加格式。
如果有GPU,可以在GPU上进行训练。默认情况下,minibatchqueue
对象将每个输出转换为gpuArray
如果GPU可用。使用GPU需要并行计算工具箱™和受支持的GPU设备。金宝app有关受支持设备的信息,请参见金宝appGPU支金宝app持版本(并行计算工具箱).
初始化培训进度图。
初始化SGDM求解器的速度参数。
使用自定义训练循环训练网络。对于每个纪元,洗牌数据并在小批量数据上循环。对于每个小批量:
测试模型
通过将验证集上的预测结果与真实标签进行比较,检验模型的分类精度。
经过训练后,对新数据进行预测不需要标签。创建minibatchqueue
对象,只包含测试数据的预测器:
为了忽略标签进行测试,将迷你批处理队列的输出数量设置为1。
指定用于训练的相同的小批大小。
属性预处理预测器preprocessMiniBatchPredictors
函数,在示例末尾列出。
对于数据存储的单个输出,指定迷你批处理格式“SSCB”
(空间,空间,通道,批次)。
循环小批和分类图像使用modelPredictions
函数,在示例末尾列出。
评估分类准确率。
模型梯度函数
的modelGradients
函数的参数为dlnetwork
对象dlnet
,一小批输入数据dlX
有相应的标签Y
并返回损失相对于中可学习参数的梯度dlnet
、网络状态、丢失。要自动计算梯度,请使用dlgradient
函数。
模型预测函数
的modelPredictions
函数的参数为dlnetwork
对象dlnet
,一个minibatchqueue
输入数据的兆贝可
和网络类,并通过迭代中所有数据来计算模型预测minibatchqueue
对象。函数使用onehotdecode
函数来查找得分最高的预测班级。
迷你批量预处理功能
的preprocessMiniBatch
函数使用以下步骤预处理一小批预测器和标签:
对图像进行预处理preprocessMiniBatchPredictors
函数。
从传入的单元格数组中提取标签数据,并沿着第二次维度连接到分类数组中。
One-hot将分类标签编码为数字数组。编码到第一个维度会产生一个与网络输出形状匹配的编码数组。
小批量预测预处理函数
的preprocessMiniBatchPredictors
函数通过从输入单元格数组中提取图像数据并将其连接到数值数组来预处理一小批预测器。对于灰度输入,在第四个维度上的连接将为每个图像添加第三个维度,以用作单个通道维度。