[发明专利]图像数据处理方法、装置、计算机可读介质及电子设备

权利要求书

1.一种图像数据处理方法，其特征在于，包括：

获取图像样本数据集，所述图像样本数据集中包括带有类别标签的训练样本；

在结构参数的更新轮次下，通过剪枝网络单元对所述训练样本进行映射处理以得到与所述训练样本相关的损失误差；所述剪枝网络单元是根据掩膜参数对初始网络单元进行剪枝处理得到的，所述初始网络单元包括至少两个网络节点；

根据所述损失误差确定各个网络节点之间的结构参数的结构参数误差梯度；

获取当前训练轮次的学习率，并根据学习率以及结构参数误差梯度更新各个网络节点之间的结构参数；

在达到预设的训练终止条件时，根据更新得到的结构参数选取目标操作分支，并基于目标操作分支确定目标网络单元，以通过所述目标网络单元对图像数据集执行分类任务。

2.根据权利要求1所述的图像数据处理方法，其特征在于，所述方法还包括：

获取当前训练轮次的轮次信息，并根据所述轮次信息确定当前训练轮次是否为掩膜参数更新轮次；

若当前训练轮次不是所述掩膜参数更新轮次，则将上一训练轮次的掩膜参数保持为当前训练轮次的掩膜参数；

若当前训练轮次是所述掩膜参数更新轮次，则根据当前训练轮次的剪枝率以及各个网络节点之间的结构参数确定当前训练轮次的掩膜参数。

3.根据权利要求2所述的图像数据处理方法，其特征在于，所述根据当前训练轮次下的剪枝率以及各个网络节点之间的结构参数确定当前训练轮次的掩膜参数，包括：

获取当前训练轮次的轮次信息，并根据所述轮次信息确定当前训练轮次的剪枝率；

根据当前训练轮次的剪枝率以及各个网络节点之间的操作分支的分支数量确定所述操作分支的剪枝数量和保留数量；

根据各个网络节点之间的结构参数选取所述剪枝数量的操作分支作为剪枝分支并选取所述保留数量的操作分支作为保留分支；

为所述剪枝分支分配第一掩膜数值并为所述保留分支分配第二掩膜数值，以得到由所述第一掩膜数值和所述第二掩膜数值组成的当前训练轮次的掩膜参数。

4.根据权利要求3所述的图像数据处理方法，其特征在于，所述根据所述轮次信息确定当前训练轮次下的剪枝率，包括：

获取预设的迭代训练次数，并对所述迭代训练次数进行划分以得到多个剪枝率更新阶段，其中各个剪枝率更新阶段具有分段增大的剪枝率；

根据所述轮次信息确定当前训练轮次所在的剪枝率更新阶段，并将该剪枝率更新阶段的剪枝率确定为当前训练轮次的剪枝率。

5.根据权利要求3所述的图像数据处理方法，其特征在于，所述根据所述轮次信息确定当前训练轮次的剪枝率，包括：

获取用于在不同训练轮次下连续增大剪枝率的剪枝率更新模型；

通过所述剪枝率更新模型对所述轮次信息进行映射处理以得到当前训练轮次下的剪枝率。

6.根据权利要求3所述的图像数据处理方法，其特征在于，所述根据各个网络节点之间的结构参数选取所述剪枝数量的操作分支作为剪枝分支并选取所述保留数量的操作分支作为保留分支，包括：

对各个网络节点之间的结构参数进行归一化映射以得到每个操作分支的选取概率；

按照所述选取概率将概率最低的所述剪枝数量的操作分支选取为剪枝分支并将概率最高的所述保留数量的操作分支选取为保留分支。

7.根据权利要求1所述的图像数据处理方法，其特征在于，所述方法还包括：

根据掩膜参数将初始网络单元中的部分操作分支确定为可更新分支，并将除所述可更新分支以外的其他操作分支确定为不可更新分支；

将所述可更新分支的网络参数和结构参数配置为可更新参数并将所述不可更新分支的网络参数和结构参数配置为不可更新参数；

将所述可更新分支以及所述不可更新分支组成的网络单元确定为剪枝网络单元。