中文

GANN:用于半监督学习的图对齐神经网络

机器学习 2023-03-15 v1 人工智能

摘要

图神经网络(GNNs)在半监督图机器学习领域已被广泛研究。多数方法在标记数据有限时未能充分利用图信息,导致过平滑问题。为克服该问题,我们提出图对齐神经网络(GANN),一种简单而有效的图神经架构。我们提出了一种带有三条对齐规则的独特学习算法,以充分挖掘标签不足下的隐藏信息。首先,为更好地探究属性特性,我们提出特征对齐规则,将属性矩阵与嵌入矩阵的内积对齐。其次,为恰当利用高阶邻居信息,我们提出簇中心对齐规则,涉及将簇中心矩阵与单位矩阵的内积对齐。最后,为在少标签下获得可靠预测结果,我们通过将预测概率矩阵与其锐化结果对齐,建立最小熵对齐规则。在图基准数据集上的大量研究表明,GANN 在半监督节点分类中能取得显著收益,并优于最先进的竞争对手。

关键词

引用

@article{arxiv.2303.07778,
  title  = {GANN: Graph Alignment Neural Network for Semi-Supervised Learning},
  author = {Linxuan Song and Wenxuan Tu and Sihang Zhou and Xinwang Liu and En Zhu},
  journal= {arXiv preprint arXiv:2303.07778},
  year   = {2023}
}