中文

图神经网络的能量基视角

机器学习 2021-10-06 v2

摘要

图神经网络是一类处理图结构数据的流行神经网络变体。本工作中,我们考虑将图神经网络与Grathwohl等人(2019)的能量基视角相结合,旨在获得更鲁棒的分类器。我们提出了一种确保对特征及邻接矩阵生成的新方法,成功实现了该框架,并与标准图卷积网络(GCN)架构(Kipf & Welling(2016))进行了对比评估。我们的方法在取得可比判别性能的同时提升了鲁棒性,为能量基图神经网络的未来研究开辟了有前景的新方向。

关键词

引用

@article{arxiv.2104.13492,
  title  = {An Energy-Based View of Graph Neural Networks},
  author = {John Y. Shin and Prathamesh Dharangutte},
  journal= {arXiv preprint arXiv:2104.13492},
  year   = {2021}
}

备注

-Updated with new references. -Accepted to the ICLR2021 EBM Workshop