2.118

影响因子

    高级检索

    双侧鲁棒增强的图节点分类模型

    Bilateral Robustness Enhanced Node Classification Model

    • 摘要: 图表示学习广泛应用于节点分类任务,但在结构投毒攻击下鲁棒性较差,甚至低于基础图卷积神经网络。现有方法主要关注嵌入侧优化,忽略任务侧鲁棒性。提出基于对比学习的双侧鲁棒增强节点分类模型 。在嵌入侧,引入基于边曲率的图卷积,增强同类节点的聚合能力,并结合局部-全局对比学习获取鲁棒节点表示。利用节点嵌入重构邻接关系,降低攻击边对表示的影响。在任务侧,结合原始特征与重构结构的多视图信息,优化分类器,使不同视图的输出保持一致,从而提升鲁棒性。在 3个基准数据集上,采用无目标攻击、目标攻击及随机攻击进行实验,该模型在对抗攻击下的鲁棒性优于或相当于当前强基线模型,通过嵌入优化与任务增强的双侧协同机制,提高图神经网络在对抗攻击下的稳定性,为图神经网络的安全性研究提供新思路。

       

      Abstract: Graph representation learning, which learns node representations in a self-supervised manner for downstream supervised tasks like node classification, has gained wide attention in the field of graph mining. However, recent studies have shown that graph representa‐tion learning is not robust enough when facing malicious attacks on the graph structure, and its performance in node classification tasks even significantly lower than that of basic graph convolutional neural network models. This paper targets adversarial poisoning attacks and identifies that the vulnerability of graph self-supervised node classification models is influenced not only by the node representation mod‐ule(embedding side)but also by the robustness of the task-side neural network. Therefore, a bilateral robustness enhanced node classifi‐cation model(BREM)based on contrastive learning is proposed. Specifically, a graph convolution based on edge curvature is introduced on the node embedding side to correct messages to increase the likelihood of aggregation within the same category of nodes, and a local-global information contrastive approach is used to obtain robust node embeddings. Node embeddings are used to reconstruct inter-node re‐lationships and reduce the impact of attack edges on node representations. Unlike traditional methods that directly use node embeddings as input to multilayer perceptron(MLP), the task side utilizes the reconstructed structural updates of node features and the original node features to construct multi-view information for nodes, optimizing the task-side model to make outputs from different views more similar, thus enhancing robustness. Experiments on three popular benchmark datasets with different attack types such as no-target attacks, target attacks, and random attacks validate that BREM achieves better or comparable robustness compared to current strong baseline models.

       

    /

    返回文章
    返回