[发明专利]用于训练模型的方法和装置

权利要求书

1.一种用于训练模型的方法，包括：

获取训练数据、以及获取所述训练数据的真实标签；

采用初始视频分类模型预测所述训练数据的预测标签；

采用所述真实标签与所述预测标签构建第一损失函数；

采用所述第一损失函数训练所述初始视频分类模型，并得到训练完成的目标视频分类模型，其中，在第一训练阶段，采用第一学习率训练所述初始视频分类模型，以及在第二训练阶段，采用第二学习率训练所述初始视频分类模型。

2.根据权利要求1所述的方法，其中，所述第一学习率基于训练进程动态确定，所述第二学习率基于训练进程动态确定。

3.根据权利要求1所述的方法，其中，所述初始视频分类模型为卷积神经网络模型，所述初始视频分类模型包括归一化层，所述方法包括：

针对所述初始视频分类模型中的每一层卷积层，将该卷积层的输出数据经过所述归一化层的处理后，输入该卷积层的下一层卷积层进行数据处理。

4.根据权利要求1所述的方法，其中，所述采用所述真实标签与所述预测标签构建第一损失函数，包括：

对所述预测标签进行标签平滑处理；

采用经过标签平滑处理后的预测标签与所述真实标签构建所述第一损失函数。

5.根据权利要求1所述的方法，其中，所述还方法包括：

采用预先训练好的教师模型，预测所述训练数据的实际标签；

采用所述预测标签与所述实际标签构建第二损失函数；

所述采用所述第一损失函数训练所述初始视频分类模型，并得到训练完成的目标视频分类模型，包括：

采用所述第一损失函数以及所述第二损失函数共同训练所述初始视频分类模型，并得到所述目标视频分类模型。

6.一种视频分类方法，包括：

获取待分类的视频；

采用目标视频分类模型对所述待分类的视频进行分类，其中，所述目标视频分类模型基于如权利要求1-5之一所述的方法训练得到。

7.一种用于训练模型的装置，包括：

第一获取单元，被配置为获取训练数据、以及获取所述训练数据的真实标签；

第一预测单元，被配置为采用初始视频分类模型预测所述训练数据的预测标签；

第一构建单元，被配置为采用所述真实标签与所述预测标签构建第一损失函数；

第一训练单元，被配置为采用所述第一损失函数训练所述初始视频分类模型，并得到训练完成的目标视频分类模型，其中，在第一训练阶段，采用第一学习率训练所述初始视频分类模型，以及在第二训练阶段，采用第二学习率训练所述初始视频分类模型。

8.根据权利要求7所述的装置，其中，所述第一学习率基于训练进程动态确定，所述第二学习率基于训练进程动态确定。

9.根据权利要求7所述的装置，其中，所述初始视频分类模型为卷积神经网络模型，所述初始视频分类模型包括归一化层，所述装置包括：

归一化单元，被配置为针对所述初始视频分类模型中的每一层卷积层，将该卷积层的输出数据经过所述归一化层的处理后，输入该卷积层的下一层卷积层进行数据处理。

10.根据权利要求7所述的装置，其中，所述第一构建单元，包括：

标签处理单元，被配置为对所述预测标签进行标签平滑处理；

第一构建模块，被配置为采用经过标签平滑处理后的预测标签与所述真实标签构建所述第一损失函数。

11.根据权利要求7所述的装置，其中，所述还装置包括：

第二预测单元，被配置为采用预先训练好的教师模型，预测所述训练数据的实际标签；

第二构建单元，被配置为采用所述预测标签与所述实际标签构建第二损失函数；

所述第一训练单元，包括：

第二训练单元，被配置为采用所述第一损失函数以及所述第二损失函数共同训练所述初始视频分类模型，并得到所述目标视频分类模型。