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.


翻译:近期研究为在关系数据库(RDB)上实现完全数据驱动的学习范式,提出关系深度学习(RDL)方法,将RDB结构化为异质实体图,并采用图神经网络(GNN)作为预测模型。然而现有RDL方法忽略了RDB中关系数据的非平衡问题,存在少数类实体表征不足的风险,导致模型实际应用中失效。本文首次系统研究RDB实体分类中的类别非平衡问题,并提出面向关系中心的少数类合成过采样GNN模型(Rel-MOSS),以填补当前文献的关键空白。具体而言,为缓解少数类相关信息被多数类信息淹没的问题,我们设计了关系维度的门控控制器,对来自各关系类型的邻域消息进行调节。基于关系门控表征,我们进一步提出关系引导的少数类合成过采样方法,通过整合实体关系签名保持关系一致性。在12个实体分类数据集上的大量实验验证了Rel-MOSS的优越性,相比处理类别非平衡的最先进RDL方法及经典方法,其在均衡准确率和G-Mean指标上分别实现了平均2.46%和4.00%的提升。

0
下载
关闭预览

相关内容

不平衡数据学习的全面综述
专知会员服务
44+阅读 · 2025年2月15日
【NTU博士论文】图结构数据的深度学习
专知会员服务
36+阅读 · 2024年9月18日
当深度强化学习遇见图神经网络
专知
227+阅读 · 2019年10月21日
用深度学习揭示数据的因果关系
专知
28+阅读 · 2019年5月18日
Github热门图深度学习(GraphDL)源码与框架
新智元
21+阅读 · 2019年3月19日
SFFAI报告 | 常建龙 :深度卷积网络中的卷积算子研究进展
人工智能前沿讲习班
11+阅读 · 2018年10月22日
深度学习中的五大正则化方法和七大优化策略
全球人工智能
11+阅读 · 2017年12月25日
国家自然科学基金
5+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
17+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
11+阅读 · 2012年12月31日
VIP会员
最新内容
致命七类无人机:无人机时代的演进型合成兵种
专知会员服务
1+阅读 · 19分钟前
《异构无人水面艇集群作战自主制导算法》130页
《人工智能能通过美国陆军战争学院吗?》报告
军事域人工智能驱动系统的治理
专知会员服务
4+阅读 · 9月14日
相关VIP内容
不平衡数据学习的全面综述
专知会员服务
44+阅读 · 2025年2月15日
【NTU博士论文】图结构数据的深度学习
专知会员服务
36+阅读 · 2024年9月18日
相关基金
国家自然科学基金
5+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
17+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
国家自然科学基金
11+阅读 · 2012年12月31日
Top
微信扫码咨询专知VIP会员