English

Tackling prediction tasks in relational databases with LLMs

Machine Learning 2024-11-19 v1 Computation and Language Databases

Abstract

Though large language models (LLMs) have demonstrated exceptional performance across numerous problems, their application to predictive tasks in relational databases remains largely unexplored. In this work, we address the notion that LLMs cannot yield satisfactory results on relational databases due to their interconnected tables, complex relationships, and heterogeneous data types. Using the recently introduced RelBench benchmark, we demonstrate that even a straightforward application of LLMs achieves competitive performance on these tasks. These findings establish LLMs as a promising new baseline for ML on relational databases and encourage further research in this direction.

Keywords

Cite

@article{arxiv.2411.11829,
  title  = {Tackling prediction tasks in relational databases with LLMs},
  author = {Marek Wydmuch and Łukasz Borchmann and Filip Graliński},
  journal= {arXiv preprint arXiv:2411.11829},
  year   = {2024}
}
R2 v1 2026-06-28T20:03:56.464Z