中文

Label-GCN:一种在图卷积网络中有效加入标签传播的方法

机器学习 2021-04-07 v1

摘要

我们展示了通过对图卷积网络(GCN)第一层的修改,可用于在邻居节点间有效传播标签信息,适用于二分类与多分类问题。这是在GCN训练阶段对标签特征选择性地消除自环来实现的。GCN架构其余部分保持不变,无任何额外超参数,并可用于直推式与归纳式设定。我们通过若干实验表明,根据推理阶段可用标签数量的不同,该策略相比标准GCN方法能带来模型性能的显著提升,包括在类别不平衡数据集上。

关键词

引用

@article{arxiv.2104.02153,
  title  = {Label-GCN: An Effective Method for Adding Label Propagation to Graph Convolutional Networks},
  author = {Claudio Bellei and Hussain Alattas and Nesrine Kaaniche},
  journal= {arXiv preprint arXiv:2104.02153},
  year   = {2021}
}