可信的图神经网络节点分类方法

  • 打印
  • 收藏
收藏成功


打开文本图片集

径向基函数计算节点间距离,得到各类节点质心后,根据距离分配与未标记节点最近质心的类别标签提高节点分类性能,同时定义未标记节点和质心之间的距离为模型输出的不确定性,并使用梯度惩罚损失加强输入变化的可检测性,可以有效地检测分布外节点样本。在Cora、Citeseer 和Pubmed 这3 个公开网络数据集上的结果表明:模型在分类任务的AUROC 指标分别达到81.5%、76.2%和74.6%,在分布外样本检测任务中AUROC 指标分别达到83.6%、72.8%和70.6%,证明了所提算法在提高节点分类性能的同时,可以有效检测分布外的节点样本,提高了节点分类的可信性。(剩余13200字)

试读结束

monitor