中文

PiNet:用于图分类的注意力池化

机器学习 2020-08-12 v1 计算机视觉与模式识别 机器学习

摘要

我们提出 PiNet,一种广义的可微分基于注意力的池化机制,用于利用图卷积操作进行图级分类。我们展示了在区分同构图类时相较其他图神经网络的高采样效率与更优性能,并在标准化学信息学数据集上取得了与最先进方法具竞争力的结果。

关键词

引用

@article{arxiv.2008.04575,
  title  = {PiNet: Attention Pooling for Graph Classification},
  author = {Peter Meltzer and Marcelo Daniel Gutierrez Mallea and Peter J. Bentley},
  journal= {arXiv preprint arXiv:2008.04575},
  year   = {2020}
}

备注

4 pages, 3 figures 1 table