cs.LGFeb 26, 2026

Relatron: Automating Relational Machine Learning over Relational Databases

Authors: Zhikai Chen, Han Xie, Jian Zhang, Jiliang Tang, Xiang Song, Huzefa Rangwala

Organizations: Michigan State University · Amazon

Abstract

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.

Figures & tables

Appendix figures & tables9 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. A Fair Benchmarking of Deep Relational Database Learning Models

    Jul 4, 2026Kazi F. Akhter, Bharath Ajendla, Manar D. SamadTransformerGraph Representation Learning

  2. Towards Foundation Models for Relational Databases with Language Models and Graph Neural Networks

    May 15, 2026Jingcheng Wu, Ratan Bahadur Thapa, Mojtaba Nayyeri +4Relational Foundation ModelsTabular Foundation Models

  3. What Makes a Desired Graph for Relational Deep Learning?

    Jun 7, 2026Yao Cheng, Siqiang LuoGraph Structure LearningGraph Neural Networks