模型训练方法、装置、设备、存储介质和计算机程序产品制造方法及图纸

本申请涉及一种模型训练方法、装置、计算机设备、存储介质和计算机程序产品。该方法涉及人工智能技术，该方法包括：对于每个节点，通过节点表示网络，按至少一种元路径，采样邻居节点，得到各元路径的采样子图，对采样子图进行图编码，得到节点对应各元路径的子图表示，聚合节点对应各种元路径的子图表示，得到节点的节点表示；对于每个对象节点，根据对象节点分别与第一资源节点、第二资源节点各自节点表示间的相似度的差异，构建目标域损失；根据对象节点分别与第三资源节点、第四资源节点各自节点表示间的相似度的差异，构建辅助域损失；联合目标域损失和辅助域损失进行模型训练，得到训练好的节点表示网络。这样，解决了用户冷启动问题。启动问题。启动问题。

【技术实现步骤摘要】
模型训练方法、装置、设备、存储介质和计算机程序产品


[0001]本申请涉及人工智能
，特别是涉及一种模型训练方法、装置、计算机设备、存储介质和计算机程序产品。

技术介绍

[0002]随着人工智能技术的发展，为了对用户进行资源推送，常常通过个性化推送系统生成用户感兴趣的资源，以此，实现资源推送。其中，个性化推送系统是基于用户的大量的历史行为数据，通过推荐算法来捕捉用户的兴趣。
[0003]然而，由于个性化推送系统常常存在新用户(即冷用户)的进入，而新用户的历史行为数据十分稀少，因此，个性化推送系统无法准确捕捉新用户的兴趣，即存在对新用户推送资源存在准确率较低的问题。

技术实现思路

[0004]基于此，有必要针对上述技术问题，提供一种能够提高对新用户推送资源准确率的模型训练方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。
[0005]本申请提供了一种模型训练方法。所述方法包括：
[0006]获取基于对象与资源交互形成的异构图，所述异构图中的节点包括对象节点与资源节点，所述资源节点具有领域属性；
[0007]对于所述异构图中的每个节点，通过节点表示网络，按至少一种元路径，从所述异构图中采样所述节点的邻居节点，得到所述节点对应各所述元路径的采样子图，对所述采样子图进行图编码，得到所述节点对应各所述元路径的子图表示，聚合所述节点对应各种元路径的子图表示，得到所述节点的节点表示；
[0008]对于所述异构图中的每个对象节点，根据所述对象节点分别与第一资源节点、第二资源节点各自节点表示间的相似度的差异，构建目标域损失；根据所述对象节点分别与第三资源节点、第四资源节点各自节点表示间的相似度的差异，构建辅助域损失；联合所述目标域损失和所述辅助域损失进行模型训练，得到训练好的用于资源推送的节点表示网络；
[0009]其中，所述第一资源节点与所述第二资源节点的领域属性为目标域，且所述第一资源节点与所述对象节点存在连边、所述第二资源节点与所述对象节点不存在连边，所述第三资源节点与所述第四资源节点的领域属性为辅助域，且所述第三资源节点与所述对象节点存在连边、所述第四资源节点与所述对象节点不存在连边。
[0010]本申请还提供了一种模型训练装置。所述装置包括：
[0011]获取模块，用于获取基于对象与资源交互形成的异构图，所述异构图中的节点包括对象节点与资源节点，所述资源节点具有领域属性；
[0012]聚合模块，用于对于所述异构图中的每个节点，通过节点表示网络，按至少一种元路径，从所述异构图中采样所述节点的邻居节点，得到所述节点对应各所述元路径的采样
子图，对所述采样子图进行图编码，得到所述节点对应各所述元路径的子图表示，聚合所述节点对应各种元路径的子图表示，得到所述节点的节点表示；
[0013]训练模块，用于对于所述异构图中的每个对象节点，根据所述对象节点分别与第一资源节点、第二资源节点各自节点表示间的相似度的差异，构建目标域损失；根据所述对象节点分别与第三资源节点、第四资源节点各自节点表示间的相似度的差异，构建辅助域损失；联合所述目标域损失和所述辅助域损失进行模型训练，得到训练好的用于资源推送的节点表示网络；
[0014]其中，所述第一资源节点与所述第二资源节点的领域属性为目标域，且所述第一资源节点与所述对象节点存在连边、所述第二资源节点与所述对象节点不存在连边，所述第三资源节点与所述第四资源节点的领域属性为辅助域，且所述第三资源节点与所述对象节点存在连边、所述第四资源节点与所述对象节点不存在连边。
[0015]在一个实施例中，所述获取模块，用于基于对象与目标域资源之间的历史交互数据，得到表征所述对象的对象节点与表征所述目标域资源的资源节点之间的交互二部图；基于对象与辅助域资源之间的历史交互数据，得到表征所述对象的对象节点与表征所述辅助域资源的资源节点之间的交互二部图；基于对象与对象之间的社交关系数据，得到表征对象的对象节点之间的社交关系图；基于资源与资源之间的种属关系数据，得到表征资源的资源节点之间的资源关系图；根据所述交互二部图、所述社交关系图以及所述资源关系图，构建异构图。
[0016]在一个实施例中，所述聚合模块，用于当所述元路径指示从所述异构图中采样一阶邻居节点时，按所述元路径所指示的一阶邻居节点的节点类型，采样所述节点的邻居节点，得到至少一个一阶邻居节点，根据所述节点与采样的所述一阶邻居节点，以及从所述一阶邻居节点指向所述节点的路径，得到所述节点对应所述元路径的采样子图。
[0017]在一个实施例中，所述聚合模块，用于当所述元路径指示从所述异构图中采样一阶邻居节点与二阶邻居节点时，按所述元路径所指示的一阶邻居节点的节点类型，采样所述节点的邻居节点，得到至少一个一阶邻居节点；对于每个所述一阶邻居节点，按所述元路径所指示的二阶邻居节点的节点类型，采样所述一阶邻居节点的邻居节点，得到至少一个二阶邻居节点；根据所述节点、采样的所述一阶邻居节点与所述二阶邻居节点，以及至少一条从所述二阶邻居节点指向所述一阶邻居节点、从所述一阶邻居节点指向所述节点的路径，得到所述节点对应所述元路径的采样子图。
[0018]在一个实施例中，所述聚合模块，用于获取所述节点对应所述元路径的采样子图中每个节点的节点特征；对于所述采样子图中的每个一阶邻居节点，通过图编码器，对指向所述一阶邻居节点的二阶邻居节点的节点特征进行融合，得到所述一阶邻居节点对应的融合特征；通过所述图编码器，对所述采样子图中各一阶邻居节点对应的融合特征进行融合，得到所述节点对应的融合特征；将所述节点对应的融合特征，作为所述节点对应所述元路径的子图表示。
[0019]在一个实施例中，所述聚合模块，用于拼接所述节点对应的各种元路径的子图表示，得到拼接表示；通过图编码器的映射函数，根据所述拼接表示确定各种元路径的子图表示分别对应的权重；按所述权重，将所述节点对应各种元路径的子图表示进行加权求和，得到所述节点的节点表示。
[0020]在一个实施例中，所述训练模块，用于确定所述异构图中与所述对象节点存在连边、且领域属性为目标域的第一资源节点，确定所述异构图中与所述对象节点不存在连边、且领域属性为目标域的第二资源节点；计算所述对象节点与所述第一资源节点各自节点表示间的第一相似度；计算所述对象节点与所述第二资源节点各自节点表示间的第二相似度；根据所述第一相似度与所述第二相似度的差异，构建目标域损失。
[0021]在一个实施例中，所述训练模块，用于确定所述异构图中与所述对象节点存在连边、且领域属性为辅助域的第三资源节点，确定所述异构图中与所述对象节点不存在连边、且领域属性为辅助域的第四资源节点；计算所述对象节点与所述第三资源节点各自节点表示间的第三相似度；计算对象节点与所述第四资源节点各自节点表示间的第四相似度；根据所述第三相似度与所述第四相似度的差异，构建辅助域损失。
[0022]在一个实施例中，所述训练模块，用于将所述目标域损失和所述辅助域损失相加本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种模型训练方法，其特征在于，所述方法包括：获取基于对象与资源交互形成的异构图，所述异构图中的节点包括对象节点与资源节点，所述资源节点具有领域属性；对于所述异构图中的每个节点，通过节点表示网络，按至少一种元路径，从所述异构图中采样所述节点的邻居节点，得到所述节点对应各所述元路径的采样子图，对所述采样子图进行图编码，得到所述节点对应各所述元路径的子图表示，聚合所述节点对应各种元路径的子图表示，得到所述节点的节点表示；对于所述异构图中的每个对象节点，根据所述对象节点分别与第一资源节点、第二资源节点各自节点表示间的相似度的差异，构建目标域损失；根据所述对象节点分别与第三资源节点、第四资源节点各自节点表示间的相似度的差异，构建辅助域损失；联合所述目标域损失和所述辅助域损失进行模型训练，得到训练好的用于资源推送的节点表示网络；其中，所述第一资源节点与所述第二资源节点的领域属性为目标域，且所述第一资源节点与所述对象节点存在连边、所述第二资源节点与所述对象节点不存在连边，所述第三资源节点与所述第四资源节点的领域属性为辅助域，且所述第三资源节点与所述对象节点存在连边、所述第四资源节点与所述对象节点不存在连边。2.根据权利要求1所述的方法，其特征在于，所述获取基于对象与资源交互形成的异构图，包括：基于对象与目标域资源之间的历史交互数据，得到表征所述对象的对象节点与表征所述目标域资源的资源节点之间的交互二部图；基于对象与辅助域资源之间的历史交互数据，得到表征所述对象的对象节点与表征所述辅助域资源的资源节点之间的交互二部图；基于对象与对象之间的社交关系数据，得到表征对象的对象节点之间的社交关系图；基于资源与资源之间的种属关系数据，得到表征资源的资源节点之间的资源关系图；根据所述交互二部图、所述社交关系图以及所述资源关系图，构建异构图。3.根据权利要求1所述的方法，其特征在于，所述按至少一种元路径，从所述异构图中采样所述节点的邻居节点，得到所述节点对应各所述元路径的采样子图，包括：当所述元路径指示从所述异构图中采样一阶邻居节点时，按所述元路径所指示的一阶邻居节点的节点类型，采样所述节点的邻居节点，得到至少一个一阶邻居节点，根据所述节点与采样的所述一阶邻居节点，以及从所述一阶邻居节点指向所述节点的路径，得到所述节点对应所述元路径的采样子图。4.根据权利要求1所述的方法，其特征在于，所述按至少一种元路径，从所述异构图中采样所述节点的邻居节点，得到所述节点对应各所述元路径的采样子图，包括：当所述元路径指示从所述异构图中采样一阶邻居节点与二阶邻居节点时，按所述元路径所指示的一阶邻居节点的节点类型，采样所述节点的邻居节点，得到至少一个一阶邻居节点；对于每个所述一阶邻居节点，按所述元路径所指示的二阶邻居节点的节点类型，采样所述一阶邻居节点的邻居节点，得到至少一个二阶邻居节点；根据所述节点、采样的所述一阶邻居节点与所述二阶邻居节点，以及至少一条从所述二阶邻居节点指向所述一阶邻居节点、从所述一阶邻居节点指向所述节点的路径，得到所
述节点对应所述元路径的采样子图。5.根据权利要求4所述的方法，其特征在于，所述对所述采样子图进行图编码，得到所述节点对应各所述元路径的子图表示，包括：获取所述节点对应所述元路径的采样子图中每个节点的节点特征；对于所述采样子图中的每个一阶邻居节点，通过图编码器，对指向所述一阶邻居节点的二阶邻居节点的节点特征进行融合，得到所述一阶邻居节点对应的融合特征；通过所述图编码器，对所述采样子图中各一阶邻居节点对应的融合特征进行融合，得到所述节点对应的融合特征；将所述节点对应的融合特征，作为所述节点对应所述元路径的子图表示。6.根据权利要求1所述的方法，其特征在于，所述聚合所述节点对应各种元路径的子图表示，得到所述节点的节点表示，包括：拼接所述节点对应的各种元路径的子图表示，得到拼接表示；通过图编码器的映射函数，根据所述拼接表示确定各种元路径的子图表示分别对应的权重；按所述权重，将所述节点对应各种元路径的子图表示进行加权求和，得到所述节点的节点表示。7.根据权利要求1所述的方法，其特征在于，所述根据所述对象节点分别与第一资源节点、第二资源节点各自节点表示间的相似度的差异，构建目标域损失，包括：确定所述异构图中与所述对象节点存在连边、且领域属性为目标域的第一资源节点，确定所述异构图中与所述对象节点不存在连边、且领域属性为目标域的第二资源节点；计算所述对象节点与所述第一资源节点各自节点表示间的第一相似度；计算所述对象节点与所述第二资源节点各自节点表示间的第二相似度；根据所述第一相似度与所述第二相似度的差异，构建目标域损失。8.根据权利要求1所述的方法，其特征在于，所述根据所述对象节点分别与...

【专利技术属性】
技术研发人员：郑佳炜，
申请(专利权)人：腾讯科技深圳有限公司，
类型：发明
国别省市：