论文标题
交替优化的图神经网络
Alternately Optimized Graph Neural Networks
论文作者
论文摘要
图形神经网络(GNN)在图形上大大推进了半监督节点分类任务。现有的大多数GNN以端到端的方式进行了培训,可以看作是解决双层优化问题。此过程通常在计算和内存使用方面效率低下。在这项工作中,我们为在图表上半监督学习提供了一个新的优化框架。所提出的框架可以通过交替的优化算法来方便地解决,从而显着提高了效率。广泛的实验表明,所提出的方法可以在最先进的基准中实现可比或更好的性能,而它具有更好的计算和记忆效率。
Graph Neural Networks (GNNs) have greatly advanced the semi-supervised node classification task on graphs. The majority of existing GNNs are trained in an end-to-end manner that can be viewed as tackling a bi-level optimization problem. This process is often inefficient in computation and memory usage. In this work, we propose a new optimization framework for semi-supervised learning on graphs. The proposed framework can be conveniently solved by the alternating optimization algorithms, resulting in significantly improved efficiency. Extensive experiments demonstrate that the proposed method can achieve comparable or better performance with state-of-the-art baselines while it has significantly better computation and memory efficiency.