解决关系数据库中的少数类数据不平衡问题,提升分类准确性。
Rel-MOSS: Towards Imbalanced Relational Deep Learning on Relational Databases
- 按关系类型设计门控机制,筛选关键邻居信息。
- 基于关系签名合成少数类样本,保持数据一致性。
- 在12个数据集上平均提升4.00%的G-Mean,适合处理不平衡关系数据。
近年来,为在关系型数据库(RDB)上实现全数据驱动的学习范式,关系深度学习(RDL)被提出,将RDB结构化为异构实体图,并采用图神经网络(GNN)作为预测模型。然而,现有RDL方法忽视了RDB中关系数据的不平衡问题,导致少数类实体被过度忽略,实际应用中模型失效。本文首次研究了RDB实体分类中的类别不平衡问题,提出关系中心的少数类过采样GNN(Rel-MOSS),填补了该领域空白。为缓解少数类信息被多数类淹没的问题,设计了关系级门控控制器,调节每种关系类型的邻域消息;在此基础上,提出关系引导的少数类合成器,结合实体关系签名以保持关系一致性。在12个实体分类数据集上的实验表明,Rel-MOSS相比SOTA RDL方法和经典不平衡处理方法,在平衡准确率和G-Mean上分别平均提升最高2.46%和4.00%。
原文摘要 · Abstract (English)
In recent advances, to enable a fully data-driven learning paradigm on relational databases (RDB), relational deep learning (RDL) is proposed to structure the RDB as a heterogeneous entity graph and adopt the graph neural network (GNN) as the predictive model. However, existing RDL methods neglect the imbalance problem of relational data in RDBs and risk under-representing the minority entities, leading to an unusable model in practice. In this work, we investigate, for the first time, class imbalance problem in RDB entity classification and design the relation-centric minority synthetic over-sampling GNN (Rel-MOSS), in order to fill a critical void in the current literature. Specifically, to mitigate the issue of minority-related information being submerged by majority counterparts, we design the relation-wise gating controller to modulate neighborhood messages from each individual relation type. Based on the relational-gated representations, we further propose the relation-guided minority synthesizer for over-sampling, which integrates the entity relational signatures to maintain relational consistency. Extensive experiments on 12 entity classification datasets provide compelling evidence for the superiority of Rel-MOSS, yielding an average improvement of up to 2.46% and 4.00% in terms of Balanced Accuracy and G-Mean, compared with SOTA RDL methods and classic methods for handling class imbalance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。