论文标题
ASGNN:具有自适应结构的图形神经网络
ASGNN: Graph Neural Networks with Adaptive Structure
论文作者
论文摘要
图形神经网络(GNN)模型在众多机器学习任务中取得了令人印象深刻的成就。但是,许多现有的GNN模型被证明容易受到对抗性攻击的影响,这很难构建强大的GNN体系结构。在这项工作中,我们提出了一种具有自适应结构(ASMP)的新型可解释的消息传递方案,以防止对图形结构的对抗性攻击。 ASMP中的层是基于优化步骤得出的,该步骤最小化了同时学习节点特征和图形结构的目标函数。 ASMP是自适应的,因为可以通过动态调整的图进行不同层中的消息传递过程。这种属性允许对嘈杂(或扰动)图结构进行更细粒度的处理,从而改善了稳健性。理论上建立了ASMP方案的收敛属性。将ASMP与神经网络集成可以导致具有自适应结构(ASGNN)的新型GNN模型家族。对半监督节点分类任务的广泛实验表明,在各种对抗性攻击下,提议的ASGNN在分类性能方面优于最先进的GNN架构。
The graph neural network (GNN) models have presented impressive achievements in numerous machine learning tasks. However, many existing GNN models are shown to be vulnerable to adversarial attacks, which creates a stringent need to build robust GNN architectures. In this work, we propose a novel interpretable message passing scheme with adaptive structure (ASMP) to defend against adversarial attacks on graph structure. Layers in ASMP are derived based on optimization steps that minimize an objective function that learns the node feature and the graph structure simultaneously. ASMP is adaptive in the sense that the message passing process in different layers is able to be carried out over dynamically adjusted graphs. Such property allows more fine-grained handling of the noisy (or perturbed) graph structure and hence improves the robustness. Convergence properties of the ASMP scheme are theoretically established. Integrating ASMP with neural networks can lead to a new family of GNN models with adaptive structure (ASGNN). Extensive experiments on semi-supervised node classification tasks demonstrate that the proposed ASGNN outperforms the state-of-the-art GNN architectures in terms of classification performance under various adversarial attacks.