摘要
arXiv:2502.06784v1 类型: cross
摘要: 在电子商务、医疗保健和社交媒体等领域广泛存在的关系数据库上的预测任务至关重要。为了有效应对这些任务,关系深度学习(RDL)将关系数据编码为图,从而使图神经网络(GNNs)能够利用关系结构以提高预测效果。然而,现有的异构GNNs往往忽视了关系数据库内在的结构性质,导致建模效率低下。在此,我们引入了RelGNN,这是一种专门设计用于捕捉关系数据库独特特性的全新GNN框架。在我们的方法的核心是引入原子路线,这是一种形成高阶三部分结构的节点序列。基于这些原子路线,RelGNN 设计了新的复合消息传递机制,允许异构节点之间直接的单跳交互。这种方法避免了冗余聚集,并减少了信息纠缠,最终导致了更高效和准确的预测建模。RelGNN 在来自 RelBench(Fey 等人,2024)的 30 个多样化的真实世界任务中进行了评估,并且始终实现了最先进的准确率,最高提高了25%。