(19)中华 人民共和国 国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202111244280.7 (22)申请日 2021.10.25 (71)申请人 中国科学技术大学 地址 230026 安徽省合肥市包河区金寨路 96号 (72)发明人 何向南　王艳芳　隋勇铎　王翔　 刘振广　 (74)专利代理 机构 中科专利商标代理有限责任 公司 11021 代理人 樊晓 (51)Int.Cl. G06F 30/27(2020.01) G06F 17/16(2006.01) G06N 20/00(2019.01) (54)发明名称 基于矩阵分解模 型的剪枝 方法、 装置、 设备、 存储介质 (57)摘要 本发明公开了一种基于矩阵分解模型的剪 枝方法， 包括： 获取目标矩阵分解模型的模型结 构数据， 其中， 模型结构数据包括多个矩阵参数； 多个矩阵参数的每一行分别对应一个目标用户 或一个目标物品； 根据多个矩阵参数确定目标矩 阵分解模型中可被剪枝的第一矩阵参数； 利用基 于第一矩 阵参数生成的第二矩 阵参数更新矩 阵 集合； 以及通过评价指标确定矩阵集合中的目标 矩阵参数。 本公开提供的基于矩阵分解模型的剪 枝方法能够在保证模型性能的同时， 减少表征矩 阵的参数量， 从而有效地降低模型能耗， 并且加 快推理速度。 权利要求书2页 说明书12页 附图4页 CN 113962156 A 2022.01.21 CN 113962156 A 1.一种基于矩阵分解模型的剪枝方法， 包括： 获取目标矩阵分解模型的模型结构数据， 其中， 所述模型结构数据包括多个矩阵参数； 所述多个矩阵参数的每一行分别对应一个目标用户或一个目标物品； 根据所述多个矩阵参数确定所述目标矩阵分解模型中可被剪枝的第一矩阵参数； 利用基于所述第一矩阵参数生成的第二矩阵参数 更新矩阵集 合； 以及 通过评价指标确定所述矩阵集 合中的目标矩阵参数。 2.根据权利要求1所述的方法， 其中， 所述根据所述多个矩阵参数确定所述目标矩阵分 解模型中可被剪枝的第一矩阵参数包括： 根据预设规则将所述多个矩阵参数进行排序， 其中， 所述预设规则为所述多个矩阵参 数的绝对值； 确定所述多个矩阵参数的绝对值最小的矩阵参数为所述第一矩阵参数， 其中， 所述第 一矩阵参数包括第一掩蔽矩阵和第一表征矩阵。 3.根据权利要求2所述的方法， 其中， 所述利用基于所述第 一矩阵参数生成的第 二矩阵 参数更新矩阵集 合包括： 将所述第一掩蔽矩阵对应位置的元 素设置为0， 得到第二掩蔽矩阵； 利用所述第二掩蔽矩阵和表征矩阵生成所述第二矩阵参数； 利用所述第二矩阵参数 更新矩阵集 合。 4.根据权利要求3 中任一项所述的方法， 其中， 所述通过评价指标确定所述矩阵集合中 的目标矩阵参数包括： 根据所述第二矩阵参数， 采用第一关系模型， 得到所述第二矩阵参数的评价指标， 其 中， 所述第一关系模型为： 其中， T(u)表示用户在测试集上的行为列表； R(u)表示根据用户在训练集上的行为给 用户作出的推荐列表； | ·|表示列表长度。 5.根据权利要求3或4中任一项所述的方法， 其中， 所述通过评价指标确定所述矩阵集 合中的目标矩阵参数还 包括： 根据所述第二矩阵参数， 采用第二关系模型， 得到所述第二矩阵参数的评价指标， 其 中， 所述第二关系模型为： 其中， |REL|表示测试集物品列表的前K个； rel_i表示K个物品中第i个物品在物品列表 中的排序。 6.根据权利要求1至5中任一项所述的方法， 其中， 在所述通过评价指标确定所述矩阵 集合中的目标矩阵参数之后， 所述方法还 包括： 将所述目标矩阵参数分解为目标用户矩阵和目标物品矩阵的乘积， 以便根据 所述目标 用户矩阵和所述目标物品矩阵为所述目标用户推荐所述目标物品。权　利　要　求　书 1/2 页 2 CN 113962156 A 27.根据权利要求1所述的方法， 其中， 在所述获取目标矩阵分解模型的模型结构数据之 前， 所述方法还 包括： 获取用户与物品交 互关系的二部图、 包 含用户和物品的所述表征矩阵； 利用所述表征矩阵参数构建初始掩蔽矩阵， 其中， 所述初始掩蔽矩阵的维度与所述表 征矩阵参数相同； 所述初始掩蔽矩阵中的各个元 素均为1； 利用所述用户与物品交互关系的二部图、 所述表征矩阵参数和所述初始掩蔽矩阵构建 待训练的矩阵分解模型； 利用所述包含用户与物品交互关系的二部图训练所述待训练的矩阵分解模型， 得到目 标矩阵分解模型。 8.一种基于矩阵分解模型的剪枝装置， 包括： 第一获取模块， 用于获取目标矩阵分解模型的模型结构数据， 其中， 所述模型结构数据 包括多个矩阵参数； 所述多个矩阵参数的每一行分别对应一个目标用户或一个目标物品； 第一确定模块， 用于根据所述多个矩阵参数确定所述目标矩阵分解模型中可被剪枝的 第一矩阵参数； 更新模块， 用于利用基于所述第一矩阵参数生成的第二矩阵参数 更新矩阵集 合； 以及 第二确定模块， 用于通过评价指标确定所述矩阵集 合中的目标矩阵参数。 9.一种电子设备， 包括： 一个或多个处 理器； 存储器， 用于存 储一个或多个程序， 其中， 当所述一个或多个程序被所述一个或多个处理器执行时， 使得所述一个或多个 处理器执行根据权利要求1～7中任一项所述的方法。 10.一种计算机可读存储介质， 其上存储有可执行指令， 该指令被处理器执行时使处理 器执行根据权利要求1～7中任一项所述的方法。权　利　要　求　书 2/2 页 3 CN 113962156 A 3