通过图序列化将头数据特征注入尾数据,缓解图学习中的类别不平衡问题。
Graffin: Stand for Tails in Imbalanced Node Classification
- 利用图序列化技术,将头类特征流动到尾类节点增强其表示
- 在4个真实数据集上提升尾类节点分类准确率,整体性能无明显下降
- 模块可插拔,适用于各类图神经网络,尤其适合长尾分布场景
图表示学习(GRL)在众多场景中表现优异。然而现实图数据普遍存在节点标签和度数分布不均的问题,导致模型对尾部类别表现不佳。现有方法多假设输入图分布均衡,忽略实际不平衡性,造成尾部数据在训练中被忽视。为此,本文提出Graffin,一个可插拔的尾部数据增强模块。受循环神经网络启发,Graffin通过图序列化技术将头部特征传递至尾部节点,融合局部与全局结构信息,丰富尾部节点的语义表征。在四个真实世界数据集上的节点分类实验表明,Graffin能有效提升对尾部数据的适应能力,且不会显著损害整体性能。
原文摘要 · Abstract (English)
Graph representation learning (GRL) models have succeeded in many scenarios. Real-world graphs have imbalanced distribution, such as node labels and degrees, which leaves a critical challenge to GRL. Imbalanced inputs can lead to imbalanced outputs. However, most existing works ignore it and assume that the distribution of input graphs is balanced, which cannot align with real situations, resulting in worse model performance on tail data. The domination of head data makes tail data underrepresented when training graph neural networks (GNNs). Thus, we propose Graffin, a pluggable tail data augmentation module, to address the above issues. Inspired by recurrent neural networks (RNNs), Graffin flows head features into tail data through graph serialization techniques to alleviate the imbalance of tail representation. The local and global structures are fused to form the node representation under the combined effect of neighborhood and sequence information, which enriches the semantics of tail data. We validate the performance of Graffin on four real-world datasets in node classification tasks. Results show that Graffin can improve the adaptation to tail data without significantly degrading the overall model performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。