Dec, 2023

基于子图采样的异构图节点嵌入的 GAN 方法

TL;DR我们提出了一个新方法来解决异构图中的类不平衡问题,该方法结合了生成对抗网络(GANs)和图神经网络(GNNs)的优点,通过创建合成节点和边来有效地平衡数据集。该方法直接针对数据级别的不平衡问题,并解决了数据生成过程中忽略图结构以及在下游任务中使用基于 GNN 的分类器时创建可用的合成结构的问题。同时处理节点和边的信息,通过节点增强和子图采样改善边的平衡。另外,我们的框架还整合了一个阈值策略,在训练过程中帮助确定最佳的边界阈值,避免了耗时的参数调整。在 Amazon 和 Yelp 评论数据集上的实验验证了我们提出的框架的有效性,特别是在少数节点识别方面,它在关键性能指标上始终优于基线模型,展示了其在该领域的潜力。