arXiv:2410.09348cs.LGcs.SI2024-10ICLR被引 5

用博弈论选节点,让图自训练更稳更准

BANGS: Game-Theoretic Node Selection for Graph Self-Training

  • 将节点选择视为集体博弈,考虑节点间依赖关系
  • 在多个数据集上超越现有方法,抗噪声能力强
  • 适合追求高鲁棒性的图学习研究者使用

图自训练是一种半监督学习方法,通过迭代选择未标记数据来重新训练图神经网络(GNN)模型以提升预测性能。虽然高置信度节点选择策略有效,但其忽略了节点间的组合依赖关系,且仅关注局部分布。为此,我们提出BANGS框架,将标签策略与条件互信息统一为节点选择的目标。该方法基于博弈论,以组合方式选择节点,并在噪声目标下提供理论鲁棒性保证。不同于传统方法独立排序并选择节点,BANGS将节点视为自训练过程中的整体集合。实验表明,BANGS在多种数据集、基础模型和超参数设置下均表现更优,显著超越现有技术。代码已开源:https://github.com/fangxin-wang/BANGS。

原文摘要 · Abstract (English)

Graph self-training is a semi-supervised learning method that iteratively selects a set of unlabeled data to retrain the underlying graph neural network (GNN) model and improve its prediction performance. While selecting highly confident nodes has proven effective for self-training, this pseudo-labeling strategy ignores the combinatorial dependencies between nodes and suffers from a local view of the distribution. To overcome these issues, we propose BANGS, a novel framework that unifies the labeling strategy with conditional mutual information as the objective of node selection. Our approach -- grounded in game theory -- selects nodes in a combinatorial fashion and provides theoretical guarantees for robustness under noisy objective. More specifically, unlike traditional methods that rank and select nodes independently, BANGS considers nodes as a collective set in the self-training process. Our method demonstrates superior performance and robustness across various datasets, base models, and hyperparameter settings, outperforming existing techniques. The codebase is available on https://github.com/fangxin-wang/BANGS .

图神经网络自训练博弈论节点选择

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。