[发明专利]模型训练方法、装置、存储介质及终端在审
申请号: | 202210975521.3 | 申请日: | 2022-08-15 |
公开(公告)号: | CN115345234A | 公开(公告)日: | 2022-11-15 |
发明(设计)人: | 崔正文;何珊珊;吴杰;杜冲 | 申请(专利权)人: | OPPO广东移动通信有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N20/00;G06Q30/06;G06F16/9535;G06F16/958 |
代理公司: | 深圳翼盛智成知识产权事务所(普通合伙) 44300 | 代理人: | 蒋尧 |
地址: | 523860 广东*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 模型 训练 方法 装置 存储 介质 终端 | ||
1.一种模型训练方法,其特征在于,所述方法包括:
获取训练样本数据集合,所述训练样本数据集合中的训练样本数据包括用户数据、对象数据以及对应的标签数据,所述标签数据指示所述训练样本数据的正负样本状态;
在模型训练过程中,基于计数损失函数,对所述训练样本数据中正样本数据的用户数据以及对象数据进行实时计数;
基于对所述用户数据以及所述对象数据的实时计数数值计算训练样本数据的权重系数;
根据所述权重系数调整所述模型的训练过程中的损失函数,并基于调整后的损失函数对所述模型进行训练。
2.根据权利要求1所述的方法,其特征在于,所述基于计数损失函数,对所述训练样本数据中正样本数据的用户数据以及对象数据进行实时计数,包括:
确定用户数据对应的用户计数特征以及对象数据对应的对象计数特征;
构建关于用户数据和对象数据的计数损失函数;
基于预设学习率的梯度下降优化器、所述计数损失函数以及所述标签数据,对所述用户计数特征以及所述对象计数特征进行更新,以实现对所述训练样本数据中正样本数据的用户数据以及对象数据进行实时计数。
3.根据权利要求1所述的方法,其特征在于,所述基于对所述用户数据以及所述对象数据的实时计数数值计算训练样本数据的权重系数,包括:
计算所述用户数据的实时计数数值与一之和的对数值,得到第一对数值;
计算所述对象数据的实时计数数值与一之和的对数值,得到第二对数值;
基于所述第一对数值和所述第二对数值计算训练样本数据的权重系数。
4.根据权利要求3所述的方法,其特征在于,所述基于所述第一对数值和所述第二对数值计算训练样本数据的权重系数,包括:
基于激活函数对所述第一对数值和所述第二对数值进行映射处理,得到第一映射数值和第二映射数值;
计算所述第一映射数值和所述第二映射数值的乘积,得到训练样本数据的权重系数。
5.根据权利要求1所述的方法,其特征在于,所述根据所述权重系数调整所述模型的训练过程中的损失函数,并基于调整后的损失函数对所述模型进行训练,包括:
基于所述模型对训练样本数据的预测结果和所述训练样本数据的标签数据计算分类损失函数值;
根据所述权重系数以及所述分类损失函数值计算目标损失函数值;
根据所述目标损失函数值进行梯度反传处理,以对所述模型的模型参数进行更新。
6.根据权利要求5所述的方法,其特征在于,所述基于所述模型对训练样本数据预测结果和所述训练样本的标签数据计算分类损失函数值,包括:
基于所述模型对训练样本数据进行点击率预测,得到预测结果;
计算所述预测结果与所述标签数据之间的交叉熵,得到分类损失函数值。
7.根据权利要求5所述的方法,其特征在于,所述根据所述权重系数以及所述分类损失函数值计算目标损失函数值,包括:
计算所述权重系数和所述分类损失函数值的乘积数值;
根据所述乘积数值确定目标损失函数值。
8.一种模型训练装置,其特征在于,所述装置包括:
获取模块,用于获取训练样本数据集合,所述训练样本数据集合中的训练样本数据包括用户数据、对象数据以及对应的标签数据,所述标签数据指示所述训练样本数据的正负样本状态;
计数模块,用于在模型训练过程中,基于计数损失函数,对所述训练样本数据中正样本数据的用户数据以及对象数据进行实时计数;
计算模块,用于基于对所述用户数据以及所述对象数据的实时计数数值计算训练样本数据的权重系数;
训练模块,用于根据所述权重系数调整所述模型的训练过程中的损失函数,并基于调整后的损失函数对所述模型进行训练。
9.一种存储介质,其上存储有计算机程序,其特征在于,当所述计算机程序被电子设备的处理器加载时执行如权利要求1-7中任一项所述的模型训练方法中的步骤。
10.一种终端,所述终端包括处理器和存储器,所述存储器储存有计算机程序,其特征在于,所述处理器通过加载所述计算机程序执行如权利要求1-7中任一项所述的模型训练方法中的步骤。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于OPPO广东移动通信有限公司,未经OPPO广东移动通信有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202210975521.3/1.html,转载请声明来源钻瓜专利网。