一种动态异构图注意力神经架构搜索方法及产品技术

本申请实施例涉及数据处理技术领域，公开了一种动态异构图注意力神经架构搜索方法及产品，包括：获取输入的动态异构图的异构信息，根据所述异构信息，构建基于动态异构图注意力的搜索空间；对所述搜索空间进行多阶段可微搜索，输出多个候选神经架构以及每个候选神经架构的性能评分；根据所述每个候选神经架构的性能评分，确定动态异构图注意力神经架构。本申请能够根据动态异构图自动化设计出动态异构图注意力神经架构，且将时间戳信息纳入异构邻域中构建的搜索空间具有通用性和灵活性，可以自动搜索适应于不同的数据集和应用场景的动态异构图注意力神经架构，使得到的动态异构图注意力神经架构在处理下游任务时达到更好的效果。效果。效果。

【技术实现步骤摘要】
一种动态异构图注意力神经架构搜索方法及产品


[0001]本申请实施例涉及数据处理
，具体而言，涉及一种动态异构图注意力神经架构搜索方法及产品。

技术介绍

[0002]动态异构图在现实世界的应用中无处不在，包括社交网络、电子商务网络、学术引文网络等。与静态同质图相比，动态异构图包含更丰富的异构信息，表现为节点和边的类型，以及随时间演变的图结构等动态信息。异质性和时间演化模式的建模对于动态异构图的应用至关重要，包括对未来链接、节点标签和属性的预测。
[0003]然而，虽然动态异构图神经网络最近在挖掘图动态和异构信息方面取得了显著进展，但是它们的架构都是手动设计，这就导致了架构的设计需要大量的人工努力和专家知识，而且由于手工设计的架构是固定的，不能自动适应于不同的数据集和应用场景。因此，如何根据不同应用场景搜索适用的动态异构图模型从而达到更好的效果，是现在亟待解决的问题。

技术实现思路

[0004]本申请实施例在于提供一种动态异构图注意力神经架构搜索方法及产品，旨在解决根据不同应用场景搜索适用的动态异构图模型从而达到更好本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种动态异构图注意力神经架构搜索方法，其特征在于，包括：获取输入的动态异构图的异构信息，根据所述异构信息，构建基于动态异构图注意力的搜索空间，所述搜索空间包括定位空间和参数化空间；对所述搜索空间进行多阶段可微搜索，输出多个候选神经架构以及每个候选神经架构的性能评分；根据所述每个候选神经架构的性能评分，确定动态异构图注意力神经架构，所述动态异构图注意力神经架构用于对动态异构图进行处理，并输出所述动态异构图的节点表征。2.根据权利要求1所述的动态异构图注意力神经架构搜索方法，其特征在于，获取输入的动态异构图的异构信息，根据所述异构信息，构建基于动态异构图注意力的搜索空间，包括：按照时间戳对输入的动态异构图进行分解，得到多张不同时间戳对应的静态异构图；获取每张静态异构图的异构信息，所述异构信息包括节点类型、边类型和时间戳；对所述异构信息按照注意力进行组合，得到基于动态异构图注意力的搜索空间。3.根据权利要求1所述的动态异构图注意力神经架构搜索方法，其特征在于，对所述搜索空间进行多阶段可微搜索，输出多个候选神经架构以及每个候选神经架构的性能评分，包括：将定位空间和参数化空间中对应异构信息的动态异构邻域的连接定位和动态异构邻域的映射函数进行组合，得到候选神经架构；将所述候选神经架构以可微的方式进行多阶段训练，并在验证集上进行验证，得到所述每个候选神经架构的性能评分。4.根据权利要求1所述的动态异构图注意力神经架构搜索方法，其特征在于，在确定动态异构图注意力神经架构之后，包括：将动态异构图输入所述动态异构图注意力神经架构，输出动态异构图上所有节点的表征；将所述动态异构图上所有节点的表征输入动态异构图注意力神经架构的下游任务，输出结果准确率；根据所述结果准确率，在所述动态异构图注意力架构搜索空间进行搜索，得到最优动态异构图注意力神经架构。5.根据权利要求2所述的动态异构图注意力神经架构搜索方法，其特征在于，对所述异构信息按照注意力进行组合，得到基于动态异构图注意力的搜索空间，包括：将所述异构信息按照注意力进行连接，得到多种动态异构邻域的连接定位，作为输入的动态异构图对应的定位空间；将所述异构信息与预设注意力计算函数进行映射，得到多种动态异构邻域的映射函数，作为输入的动态异构图...

【专利技术属性】
技术研发人员：朱文武，王鑫，张子威，张泽阳，
申请(专利权)人：清华大学，
类型：发明
国别省市：