arXiv:2508.08327cs.DBcs.LG2025-08被引 1

提出SRP框架,让关系型数据库更好用深度学习预测。

Synthesize, Retrieve, and Propagate: A Unified Predictive Modeling Framework for Relational Databases

  • 用合成、检索、传播三步统一建模表间依赖关系。
  • 在5个真实数据集上显著提升预测性能,超越基线方法。
  • 适合需要处理复杂表结构的工业级数据预测场景。

关系型数据库(RDB)已成为存储海量异构数据的行业标准。然而,其固有结构限制了深度学习方法的应用。以往研究主要利用主键-外键关系挖掘多表间的单一依赖,或通过连接表或构建图来建模,忽略了表间隐含的复合依赖关系,也未充分释放预测建模潜力。本文提出统一的预测建模框架SRP,通过合成特征捕捉单一依赖、检索相关信息捕获复合依赖,并在构建的图上消息传播以学习邻近模式进行预测。引入新检索机制,全面捕获关系数据库中的单一与复合依赖,扩大表格数据预测的感受野。我们对SRP组件进行系统分析,揭示模型行为并提供实用指导。在五个真实数据集上的实验验证了SRP的有效性及其在工业场景中的应用潜力。代码已开源:https://github.com/NingLi670/SRP。

原文摘要 · Abstract (English)

Relational databases (RDBs) have become the industry standard for storing massive and heterogeneous data. However, despite the widespread use of RDBs across various fields, the inherent structure of relational databases hinders their ability to benefit from flourishing deep learning methods. Previous research has primarily focused on exploiting the unary dependency among multiple tables in a relational database using the primary key - foreign key relationships, either joining multiple tables into a single table or constructing a graph among them, which leaves the implicit composite relations among different tables and a substantial potential of improvement for predictive modeling unexplored. In this paper, we propose SRP, a unified predictive modeling framework that synthesizes features using the unary dependency, retrieves related information to capture the composite dependency, and propagates messages across a constructed graph to learn adjacent patterns for prediction on relation databases. By introducing a new retrieval mechanism into RDB, SRP is designed to fully capture both the unary and the composite dependencies within a relational database, thereby enhancing the receptive field of tabular data prediction. In addition, we conduct a comprehensive analysis on the components of SRP, offering a nuanced understanding of model behaviors and practical guidelines for future applications. Extensive experiments on five real-world datasets demonstrate the effectiveness of SRP and its potential applicability in industrial scenarios. The code is released at https://github.com/NingLi670/SRP.

关系数据库预测建模深度学习图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。