arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2405.03401cs.LGcs.AI

E2GNN:用于半监督分类的高效图神经网络集成方法

E2GNN: Efficient Graph Neural Network Ensembles for Semi-Supervised Classification

The Hong Kong Polytechnic University(香港理工大学);Rice University(莱斯大学);New York University Shanghai(上海纽约大学)

Xin Zhang, Daochen Zha, Qiaoyu Tan

AI总结:

本文提出E2GNN,通过训练MLP模仿多个GNN在带标签与无标签节点上的预测,并利用强化判别器过滤错误预测节点,以可学习方式高效集成GNN,解决半监督分类中集成推理效率低和基模型性能受限的问题。

摘要:

本文研究在流行的半监督设置下图神经网络(GNN)的集成学习。集成学习通过组合多个弱学习器的输出,在提升传统机器学习的准确性和鲁棒性方面已展现出优越性。然而,采用类似思路来集成不同的GNN模型具有挑战性,原因有二。第一,GNN因其推理能力较差而广为人知,因此简单地将多个GNN模型组装在一起会降低推理效率。第二,当GNN模型仅用少量带标签节点训练时,其性能有限。在这种情况下,普通的集成方法(例如多数投票)可能是次优的,因为大多数基模型(即GNN)可能做出错误预测。为此,本文提出一种高效的集成学习器——E2GNN,通过同时利用带标签节点和无标签节点,以可学习的方式组装多个GNN。具体而言,首先根据带标签节点在给定数据场景下预训练不同的GNN模型。接着,不是直接组合它们的输出进行标签推断,而是训练一个简单的多层感知机——MLP模型,在带标签节点和无标签节点上模仿它们的预测。然后部署统一的MLP模型来推断无标签节点或新节点的标签。由于不同GNN模型对无标签节点的预测可能不正确,我们开发了一个强化判别器,以有效过滤掉那些被错误预测的节点,从而提升MLP的性能。通过这种方式,我们提出了一种原则性的方法来解决GNN集成的推理问题,同时保持集成学习的优点:性能提升。在直推式和归纳式两种设置下,跨不同GNN骨干网络和8个基准数据集的综合实验证明了E2GNN的优越性。

英文摘要:

This work studies ensemble learning for graph neural networks (GNNs) under the popular semi-supervised setting. Ensemble learning has shown superiority in improving the accuracy and robustness of traditional machine learning by combining the outputs of multiple weak learners. However, adopting a similar idea to integrate different GNN models is challenging because of two reasons. First, GNN is notorious for its poor inference ability, so naively assembling multiple GNN models would deteriorate the inference efficiency. Second, when GNN models are trained with few labeled nodes, their performance are limited. In this case, the vanilla ensemble approach, e.g., majority vote, may be sub-optimal since most base models, i.e., GNNs, may make the wrong predictions. To this end, in this paper, we propose an efficient ensemble learner--E2GNN to assemble multiple GNNs in a learnable way by leveraging both labeled and unlabeled nodes. Specifically, we first pre-train different GNN models on a given data scenario according to the labeled nodes. Next, instead of directly combing their outputs for label inference, we train a simple multi-layer perceptron--MLP model to mimic their predictions on both labeled and unlabeled nodes. Then the unified MLP model is deployed to infer labels for unlabeled or new nodes. Since the predictions of unlabeled nodes from different GNN models may be incorrect, we develop a reinforced discriminator to effectively filter out those wrongly predicted nodes to boost the performance of MLP. By doing this, we suggest a principled approach to tackle the inference issues of GNN ensembles and maintain the merit of ensemble learning: improved performance. Comprehensive experiments over both transductive and inductive settings, across different GNN backbones and 8 benchmark datasets, demonstrate the superiority of E2GNN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑