中文

弹性图神经网络

机器学习 2021-07-16 v1 人工智能 神经与进化计算

摘要

尽管许多现有的图神经网络(GNNs)已被证明执行基于 2\ell_2 的图平滑以全局强制平滑性,但在本工作中,我们旨在通过基于 1\ell_1 的图平滑进一步增强 GNNs 的局部平滑适应性。为此,我们引入了一类基于 1\ell_12\ell_2 图平滑的 GNNs(弹性 GNNs)。特别地,我们提出了一种新颖且通用的消息传递方案到 GNNs 中。该消息传递算法不仅利于反向传播训练,而且以理论收敛保证实现了所需的平滑特性。在半监督学习任务上的实验表明,所提出的弹性 GNNs 在基准数据集上获得了更好的适应性,并且对图对抗攻击具有显著鲁棒性。弹性 GNNs 的实现可在 \url{https://github.com/lxiaorui/ElasticGNN} 获取。

关键词

引用

@article{arxiv.2107.06996,
  title  = {Elastic Graph Neural Networks},
  author = {Xiaorui Liu and Wei Jin and Yao Ma and Yaxin Li and Hua Liu and Yiqi Wang and Ming Yan and Jiliang Tang},
  journal= {arXiv preprint arXiv:2107.06996},
  year   = {2021}
}

备注

ICML 2021 (International Conference on Machine Learning)