【NeurIPS2022】基于结构聚类的自监督异构图预训练

近年来，基于异构信息网络(HINs)的自监督预训练方法与传统的半监督异构图神经网络(HGNNs)相比显示出了良好的竞争力。不幸的是，它们的性能严重依赖于精心定制各种策略，以生成高质量的正例和反例，这明显限制了它们的灵活性和泛化能力。在这项工作中，我们提出了一种新的自监督异构图预训练方法SHGP，它不需要生成任何正例或负例。它由两个共享相同注意力聚合方案的模块组成。在每次迭代中，Att-LPA模块通过结构聚类产生伪标签，这些伪标签作为自我监督信号，指导Att-HGNN模块学习对象嵌入和注意系数。这两个模块可以有效地相互利用和增强，促进模型学习判别嵌入。在四个真实数据集上的大量实验证明了SHGP对最先进的无监督基线甚至半监督基线的优越有效性。我们在https://github.com/kepsail/SHGP发布源代码。

专知便捷查看

便捷下载，请关注专知公众号（点击上方蓝色专知关注）

• 后台回复“SHGP” 就可以获取《【NeurIPS2022】基于结构聚类的自监督异构图预训练》专知下载链接

                       

专知，专业可信的人工智能知识分发 ，让认知协作更快更好！欢迎注册登录专知www.zhuanzhi.ai，获取100000+AI(AI与军事、医药、公安等)主题干货知识资料！

欢迎微信扫一扫加入专知人工智能知识星球群，获取最新AI专业干货知识教程资料和与专家交流咨询！

点击“ 阅读原文 ”，了解使用 专知 ，查看获取100000+AI主题知识资料