4 个月前

一种灵活的生成式图半监督学习框架

一种灵活的生成式图半监督学习框架

摘要

我们考虑了一类问题,即基于少量已标记样本对大多数未标记的图结构数据样本进行预测。数据样本之间的关系信息,通常编码在图/网络结构中,已被证明对这些半监督学习任务有所帮助。然而,传统的图基正则化方法和近期的图神经网络并未充分挖掘特征、图和标签之间的相互关系。在本工作中,我们提出了一种灵活的生成框架用于图基半监督学习,该框架旨在接近节点特征、标签和图结构的联合分布。借鉴网络科学文献中的随机图模型的见解,这一联合分布可以使用多种分布族来实例化。对于缺失标签的推断,我们利用了近年来可扩展变分推理技术的进步来近似贝叶斯后验分布。我们在基准数据集上进行了详尽的实验以评估所提出的图基半监督学习方法。结果表明,在大多数设置下,所提出的方法优于现有最先进的模型。

代码仓库

jiaqima/G3NN
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
node-classification-on-citeseerG3NN
Accuracy: 74.5%
Training Split: 20 per node with early stopping set
Validation: YES
node-classification-on-citeseer-with-publicG3NN
Accuracy: 74.5%
node-classification-on-coraG3NN
Accuracy: 82.9%
Training Split: 20 per node with early stopping set
Validation: YES
node-classification-on-cora-with-public-splitG3NN
Accuracy: 82.9%
node-classification-on-pubmedG3NN
Accuracy: 78.4%
Training Split: 20 per node with early stopping set
Validation: YES
node-classification-on-pubmed-with-publicG3NN
Accuracy: 78.4%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
一种灵活的生成式图半监督学习框架 | 论文 | HyperAI超神经