Boosting Relational Deep Learning with Pretrained Tabular Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lachi, Veronica, Longa, Antonio, Bevilacqua, Beatrice, Lepri, Bruno, Passerini, Andrea, Ribeiro, Bruno
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908305041915904
author Lachi, Veronica
Longa, Antonio
Bevilacqua, Beatrice
Lepri, Bruno
Passerini, Andrea
Ribeiro, Bruno
author_facet Lachi, Veronica
Longa, Antonio
Bevilacqua, Beatrice
Lepri, Bruno
Passerini, Andrea
Ribeiro, Bruno
contents Relational databases, organized into tables connected by primary-foreign key relationships, are a common format for organizing data. Making predictions on relational data often involves transforming them into a flat tabular format through table joins and feature engineering, which serve as input to tabular methods. However, designing features that fully capture complex relational patterns remains challenging. Graph Neural Networks (GNNs) offer a compelling alternative by inherently modeling these relationships, but their time overhead during inference limits their applicability for real-time scenarios. In this work, we aim to bridge this gap by leveraging existing feature engineering efforts to enhance the efficiency of GNNs in relational databases. Specifically, we use GNNs to capture complex relationships within relational databases, patterns that are difficult to featurize, while employing engineered features to encode temporal information, thereby avoiding the need to retain the entire historical graph and enabling the use of smaller, more efficient graphs. Our \textsc{LightRDL} approach not only improves efficiency, but also outperforms existing models. Experimental results on the RelBench benchmark demonstrate that our framework achieves up to $33\%$ performance improvement and a $526\times$ inference speedup compared to GNNs, making it highly suitable for real-time inference.
format Preprint
id arxiv_https___arxiv_org_abs_2504_04934
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Boosting Relational Deep Learning with Pretrained Tabular Models
Lachi, Veronica
Longa, Antonio
Bevilacqua, Beatrice
Lepri, Bruno
Passerini, Andrea
Ribeiro, Bruno
Databases
Artificial Intelligence
Machine Learning
Relational databases, organized into tables connected by primary-foreign key relationships, are a common format for organizing data. Making predictions on relational data often involves transforming them into a flat tabular format through table joins and feature engineering, which serve as input to tabular methods. However, designing features that fully capture complex relational patterns remains challenging. Graph Neural Networks (GNNs) offer a compelling alternative by inherently modeling these relationships, but their time overhead during inference limits their applicability for real-time scenarios. In this work, we aim to bridge this gap by leveraging existing feature engineering efforts to enhance the efficiency of GNNs in relational databases. Specifically, we use GNNs to capture complex relationships within relational databases, patterns that are difficult to featurize, while employing engineered features to encode temporal information, thereby avoiding the need to retain the entire historical graph and enabling the use of smaller, more efficient graphs. Our \textsc{LightRDL} approach not only improves efficiency, but also outperforms existing models. Experimental results on the RelBench benchmark demonstrate that our framework achieves up to $33\%$ performance improvement and a $526\times$ inference speedup compared to GNNs, making it highly suitable for real-time inference.
title Boosting Relational Deep Learning with Pretrained Tabular Models
topic Databases
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2504.04934