Oct, 2023

基于异质性的图神经网络用于不平衡分类

TL;DR在这篇论文中,我们介绍了一种独特的方法,通过考虑图异质性来解决图中的不平衡分类问题。我们研究了类别不平衡和图异质性之间的错综复杂关系,发现少数类别不仅样本稀缺,而且表现出较低的同质性水平,这导致错误信息在相邻节点之间传播。基于这一洞察,我们提出了一种名为Fast Im-GBK的高效方法,将不平衡分类策略与意识到异质性的图神经网络相结合,以有效解决类别不平衡问题,并显著减少训练时间。我们在实际图数据上的实验证明了我们的模型在节点分类任务中相比现有基线方法具有更好的分类性能和效率。