摘要
图神经网络 (GNN) 在处理类别不平衡问题时面临巨大挑战,导致推理结果存在偏差。为了解决异构图中的这个问题,我们提出了一种新颖的框架,该框架结合了图神经网络 (GNN) 和生成对抗网络 (GAN) 来增强对欠表示节点类别的分类。该框架包含一个高级的边生成和选择模块,能够通过对抗学习同时创建合成节点和边。与以往主要关注同构图的方法不同(由于难以以矩阵形式表示异构图结构),这种方法专门针对异构数据设计。现有的解决方案通常依赖于预训练模型来整合合成节点,这可能导致优化不一致和数据表示不匹配。我们的框架避免了这些缺陷,因为它生成的与固有图拓扑和属性紧密匹配的数据,从而确保更紧密的集成。对多个真实世界数据集的评估表明,该方法优于基线模型,尤其是在识别少数节点类别的任务中,性能指标(如 F 分数和 AUC-PRC 分数)均有显著提高。这些发现突出了这种方法在解决该领域关键挑战方面的潜力。