Tackling prediction tasks in relational databases with LLMs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wydmuch, Marek, Borchmann, Łukasz, Graliński, Filip
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913580165627904
author Wydmuch, Marek
Borchmann, Łukasz
Graliński, Filip
author_facet Wydmuch, Marek
Borchmann, Łukasz
Graliński, Filip
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2411_11829
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Tackling prediction tasks in relational databases with LLMs
Wydmuch, Marek
Borchmann, Łukasz
Graliński, Filip
Machine Learning
Computation and Language
Databases
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.
title Tackling prediction tasks in relational databases with LLMs
topic Machine Learning
Computation and Language
Databases
url https://arxiv.org/abs/2411.11829