arXiv:2510.06980cs.DBcs.LG2025-10被引 2

将大型数据库压缩成小图,还能保持预测精度。

Relational Database Distillation: From Structured Tables to Condensed Graph Data

  • 用异构图压缩多表数据库,保留字段和关联关系。
  • 压缩后数据量显著减小,分类与回归任务表现接近原数据。
  • 无需复杂训练流程,适合各类下游任务快速部署。

关系型数据库(RDBs)支撑全球多数数据管理系统,信息以多张相互依赖的表格形式组织。为将其中知识用于预测任务,近期方法利用图表示学习捕捉跨表的多跳依赖关系,虽达当前最优性能,但仍受限于数据库规模带来的存储开销和消息传递计算负担。为此,本文提出关系型数据库蒸馏(RDD)问题,旨在将大规模RDBs压缩为紧凑的异构图,同时保留训练图模型所需的预测能力。通过节点特征保留多模态列信息,通过异构边编码主键-外键关系,实现数据保真度与结构完整性。为避免传统低效的两阶段蒸馏框架,设计基于核岭回归与伪标签的目标函数,生成高质量的蒸馏图特征。在多个真实世界RDBs上的实验表明,该方法显著减少数据规模,且在分类与回归任务中保持竞争力,为基于RDBs的可扩展学习提供有效路径。

原文摘要 · Abstract (English)

Relational databases (RDBs) underpin the majority of global data management systems, where information is structured into multiple interdependent tables. To effectively use the knowledge within RDBs for predictive tasks, recent advances leverage graph representation learning to capture complex inter-table relations as multi-hop dependencies. Despite achieving state-of-the-art performance, these methods remain hindered by the prohibitive storage overhead and excessive training time, due to the massive scale of the database and the computational burden of intensive message passing across interconnected tables. To alleviate these concerns, we propose and study the problem of Relational Database Distillation (RDD). Specifically, we aim to distill large-scale RDBs into compact heterogeneous graphs while retaining the predictive power (i.e., utility) required for training graph-based models. Multi-modal column information is preserved through node features, and primary-foreign key relations are encoded via heterogeneous edges, thereby maintaining both data fidelity and relational structure. To ensure adaptability across diverse downstream tasks without engaging the traditional, inefficient bi-level distillation framework, we further design a kernel ridge regression-guided objective with pseudo-labels, which produces quality features for the distilled graph. Extensive experiments on multiple real-world RDBs demonstrate that our solution substantially reduces the data size while maintaining competitive performance on classification and regression tasks, creating an effective pathway for scalable learning with RDBs.

数据库蒸馏图神经网络数据压缩异构图

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