[发明专利]推荐模型的训练方法、推荐方法、装置、设备及存储介质在审
| 申请号: | 202011508002.3 | 申请日: | 2020-12-18 |
| 公开(公告)号: | CN112633356A | 公开(公告)日: | 2021-04-09 |
| 发明(设计)人: | 王健宗;李泽远;何安珣 | 申请(专利权)人: | 平安科技(深圳)有限公司 |
| 主分类号: | G06K9/62 | 分类号: | G06K9/62;G06Q30/06 |
| 代理公司: | 深圳市力道知识产权代理事务所(普通合伙) 44507 | 代理人: | 张传义 |
| 地址: | 518000 广东省深圳市福田区福*** | 国省代码: | 广东;44 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 推荐 模型 训练 方法 装置 设备 存储 介质 | ||
1.一种推荐模型的训练方法,其特征在于,包括:
获取多个参与方的共同本地数据,所述共同本地数据包括用户信息和所述用户信息对应的物品评分信息,并基于所述用户信息和所述用户信息对应的物品评分信息构建多个物品评分矩阵;
对构建的多个所述物品评分矩阵进行矩阵分解,得到用户隐矩阵和多个物品隐矩阵;
各个参与方分别在本地对所述用户隐矩阵以及各自的所述物品隐矩阵进行迭代训练,并基于所述用户隐矩阵和多个所述物品隐矩阵构建矩阵模型,计算所述矩阵模型的损失函数的损失值;
当所述矩阵模型的损失函数的损失值达到预设值时,将所述矩阵模型作为推荐模型,完成模型训练。
2.根据权利要求1所述的推荐模型的训练方法,其特征在于,多个参与方中包括推荐参与方和其他参与方,所述其他参与方为多个参与方中除所述推荐参与方以外的其他训练参与方,多个所述物品隐矩阵包括推荐物品隐矩阵和其他物品隐矩阵;所述各个参与方分别在本地对所述用户隐矩阵以及各自的所述物品隐矩阵进行迭代训练,包括:
分别对所述用户隐矩阵和多个所述物品隐矩阵进行初始化,得到初始化用户隐矩阵、初始化推荐物品隐矩阵和初始化其他物品隐矩阵;
将所述初始化用户隐矩阵发送至其他参与方,使其他参与方在本地基于所述初始化其他物品隐矩阵和所述初始化用户隐矩阵计算中间值,将所述中间值发送给所述推荐参与方;
基于所述中间值对所述初始化用户隐矩阵进行更新,得到更新用户隐矩阵和更新推荐物品隐矩阵;
将所述更新用户隐矩阵发送至其他参与方,使其他参与方基于所述初始化用户隐矩阵对所述初始化其他物品隐矩阵进行更新,得到更新其他物品隐矩阵。
3.根据权利要求2所述的推荐模型的训练方法,其特征在于,所述计算所述矩阵模型的损失函数的损失值,包括:
基于所述特征矩阵的损失函数公式,根据所述更新用户隐矩阵、所述更新推荐物品隐矩阵和所述更新其他物品隐矩阵计算所述特征矩阵的损失函数的损失值;
所述特征矩阵的损失函数公式为:
其中,LJ表示特征矩阵J的损失函数;ui为用户隐矩阵U的第i行向量,代表用户i的特征向量;vAj为推荐物品隐矩阵VA的第j行向量,代表推荐参与方中第j种物品的特征向量;vBj为其他物品隐矩阵VB的第j行向量,代表其他参与方中第j种物品的特征向量;λ1,λ2为正则化系数。
4.根据权利要求2所述的推荐模型的训练方法,其特征在于,所述方法包括:
根据所述更新用户隐矩阵、所述更新推荐物品隐矩阵和所述更新其他物品隐矩阵计算所述特征矩阵的损失函数的损失值,并采用梯度下降法最小化所述损失函数的损失值。
5.根据权利要求2所述的推荐模型的训练方法,其特征在于,所述将所述中间值发送给所述推荐参与方,包括:
对所述中间值进行加密,并将加密后的中间值发送给所述推荐参与方;
所述将所述更新用户隐矩阵发送至其他参与方,包括:
对所述更新用户矩阵进行加密,并将加密后的更新用户矩阵发送至其他参与方。
6.一种用户推荐方法,其特征在于,所述方法包括:
获取新用户的用户信息,并基于所述用户信息获取所述新用户在其他参与方的物品评分信息;
基于所述评分信息和预先训练的推荐模型确定所述新用户在推荐参与方的物品评分信息;
根据所述新用户在所述推荐参与方的物品评分信息对所述新用户进行推荐;
其中,所述预先训练的推荐模型是采用如权利要求1至5中任一项所述的推荐模型的训练方法训练的到的。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于平安科技(深圳)有限公司,未经平安科技(深圳)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202011508002.3/1.html,转载请声明来源钻瓜专利网。





