国内刊号:11-5602/TP
国际刊号:1673-9418
发布日期:
作者:任博, 董明刚, 于扬, 卢贤睿
单位:1. 桂林理工大学 物理与电子信息工程学院,广西 桂林 5410062. 广西嵌入式技术与智能系统重点实验室,广西 桂林 5410063. 桂林理工大学 计算机科学与工程学院,广西 桂林 541006
关键词:长尾问题,图变分自编码器,图潜向量,生成对抗网络
基金:国家自然科学基金(62366012);广西自然科学基金(2021GXNSFAA220074)。
现实世界中许多图数据存在类别分布不平衡的问题,其通常表现在节点、边和图三个级别。常用的基于过采样的图级不平衡处理方法,因样本缺乏多样性,会导致模型过拟合。针对该问题,提出一种图潜向量分布学习的图过采样方法(GLRD-GAN)。提出一种图潜向量分布学习方法,利用预训练的图变分自编码器(VGAE)和全连接神经网络学习少数类图样本在低维空间内的潜向量分布,在该分布上随机采样潜向量信息并与原少数类潜向量融合,保证了少数类潜向量的多样性。设计了一种基于双解码器的图样本生成器,经预训练的内积解码器和图卷积解码器充分利用采样的潜向量来分别生成图数据的拓扑结构和节点特征。通过GAN判别器检测生成样本的真伪和类别,监督生成样本的有效性,实现多样性的少数类图样本生成。在5个具有代表性的长尾图数据集上进行了对比实验和可视化观察,结果表明提出的基于图潜向量分布学习的图过采样方法在Acc和F1值上较其他方法平均高出1%~4%,且能够生成有效的少数类图样本。
来源:2025年第7期
《计算机科学与探索》期刊编辑部