arXiv:2602.22552cs.LG2026-02被引 2

Relatron自动选择最适合关系型数据库的预测模型,提升效率与效果。

Relatron: Automating Relational Machine Learning over Relational Databases

  • 构建统一框架比较RDL与DFS,发现性能依赖具体任务
  • 提出任务信号识别模型优劣,实现精准模型路由
  • 轻量级元选择器降低调参成本,提升稳定性

基于关系型数据库(RDB)的预测建模支撑诸多应用,但需同时捕捉跨表依赖与复杂特征交互,挑战重重。关系型深度学习(RDL)通过消息传递自动特征工程,而经典方法如深度特征合成(DFS)依赖预定义非参数聚合器。尽管性能有所提升,但RDL与DFS的相对优势及架构选择原则仍不清晰。本文通过统一两者的设计空间,在多种RDB任务中开展架构导向搜索,得出三项关键发现:(1)RDL并非始终优于DFS,性能高度依赖任务;(2)无单一架构在所有任务上占优,需任务感知选型;(3)验证准确率无法可靠指导架构选择。由此建立模型性能库,关联架构配置与表现;利用该库分析RDL-DFS性能差距成因,引入两个任务信号——数据库任务同质性与融合规模、路径、特征、时序结构的亲和嵌入,其与性能差的相关性支持合理路由。基于此,提出Relatron:一种基于任务嵌入的元选择器,可在RDL与DFS间决策并压缩同类内搜索空间。轻量级损失曲面度量进一步通过偏好平坦最优解,避免脆弱检查点。实验表明,Relatron缓解了‘更多调参,更差性能’现象,在联合超参数-架构优化中相较强基线最高提升18.5%,成本仅为基于费舍尔信息方法的1/10。

原文摘要 · Abstract (English)

Predictive modeling over relational databases (RDBs) powers applications, yet remains challenging due to capturing both cross-table dependencies and complex feature interactions. Relational Deep Learning (RDL) methods automate feature engineering via message passing, while classical approaches like Deep Feature Synthesis (DFS) rely on predefined non-parametric aggregators. Despite performance gains, the comparative advantages of RDL over DFS and the design principles for selecting effective architectures remain poorly understood. We present a comprehensive study that unifies RDL and DFS in a shared design space and conducts architecture-centric searches across diverse RDB tasks. Our analysis yields three key findings: (1) RDL does not consistently outperform DFS, with performance being highly task-dependent; (2) no single architecture dominates across tasks, underscoring the need for task-aware model selection; and (3) validation accuracy is an unreliable guide for architecture choice. This search yields a model performance bank that links architecture configurations to their performance; leveraging this bank, we analyze the drivers of the RDL-DFS performance gap and introduce two task signals -- RDB task homophily and an affinity embedding that captures size, path, feature, and temporal structure -- whose correlation with the gap enables principled routing. Guided by these signals, we propose Relatron, a task embedding-based meta-selector that chooses between RDL and DFS and prunes the within-family search. Lightweight loss-landscape metrics further guard against brittle checkpoints by preferring flatter optima. In experiments, Relatron resolves the "more tuning, worse performance" effect and, in joint hyperparameter-architecture optimization, achieves up to 18.5% improvement over strong baselines with 10x lower cost than Fisher information-based alternatives.

关系型学习模型选择自动化数据库

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