基于信息论设计预训练框架,让数据库模型更懂任务上下文。
A Pre-training Framework for Relational Data with Information-theoretic Principles
- 用图遍历聚合生成预测信号,显式建模关系数据的时间动态
- 在RelBench上显著优于传统预训练方法,提升任务感知表征能力
- 适合需要理解复杂查询逻辑与时间依赖的数据库智能场景
关系数据库支撑着众多关键领域基础设施,但如何设计可泛化的预训练策略以从关系数据库中学习仍面临挑战,主要源于任务异质性——下游任务由关系模式图、时间依赖和SQL定义的标签逻辑决定。为弥合这一差距,我们提出任务向量估计(TVE)框架,通过基于集合的模式遍历图聚合构建预测监督信号,显式建模下一个时间窗口的关系动态。我们从信息论角度形式化该方法,证明引入任务先验的表征能保留更多相关信号。在RelBench基准上的大量实验表明,TVE始终优于传统预训练基线。研究结果支持将任务异质性和时间结构纳入预训练目标的设计原则,用于关系数据库上的预测建模。代码已公开于https://github.com/quang-truong/task-vector-estimation。
原文摘要 · Abstract (English)
Relational databases underpin critical infrastructure across a wide range of domains, yet the design of generalizable pre-training strategies for learning from relational databases remains an open challenge due to task heterogeneity. Specifically, there exist many possible downstream tasks, as tasks are defined based on relational schema graphs, temporal dependencies, and SQL-defined label logics. An effective pre-training framework is desired to take these factors into account in order to obtain task-aware representations. By incorporating knowledge of the underlying distribution that drives label generation, downstream tasks can benefit from relevant side-channel information. To bridge this gap, we introduce Task Vector Estimation (TVE), a novel pre-training framework that constructs predictive supervisory signals via set-based aggregation over schema traversal graphs, explicitly modeling next-window relational dynamics. We formalize our approach through an information-theoretic lens, demonstrating that task-informed representations retain more relevant signals than those obtained without task priors. Extensive experiments on the RelBench benchmark show that TVE consistently outperforms traditional pre-training baselines. Our findings advocate for pre-training objectives that encode task heterogeneity and temporal structure as design principles for predictive modeling on relational databases. Our code is publicly available at https://github.com/quang-truong/task-vector-estimation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。