about
UniPredict: Large Language Models Are Universal Tabular Predictors (arxiv.org)
1 point by PaulHoule on Oct 14, 2023 | hide | past | pdf | discuss on HN

In plain words: Instead of a separate model for every table and target, this trains one language model on many different tables so it answers prediction questions from instructions. It beat the best tree-boosting model by 5.4% and handled brand-new tables with only a handful of examples.

Abstract · UniPredict: Large Language Models are Universal Tabular Classifiers

Tabular data prediction is a fundamental machine learning task for many applications. Existing methods predominantly employ discriminative modeling and operate under the assumption of a fixed target column, necessitating re-training for every new predictive task. Inspired by the generative power of large language models (LLMs), this paper exploits the idea of building universal tabular data predictors based on generative modeling, namely UniPredict. Here, we demonstrate the scalability of an LLM to extensive tabular datasets, enabling it to comprehend diverse tabular inputs and predict target variables following the provided instructions. Specifically, we train a single LLM on an aggregation of 169 tabular datasets with diverse targets and compare its performance against baselines that are trained on each dataset separately. We observe this versatile UniPredict model demonstrates an advantage over other models, ranging from 5.4% to 13.4%, when compared with the best tree-boosting baseline and the best neural network baseline, respectively. We further test UniPredict in few-shot learning settings on another 62 tabular datasets. Our method achieves strong performance in quickly adapting to new tasks. In low-resource few-shot setup, we observed a 100%+ performance advantage compared with XGBoost, and significant margin over all baselines. We envision that UniPredict sheds light on developing a universal tabular data prediction system that learns from data at scale and serves a wide range of prediction tasks.

Ruiyu Wang, Zifeng Wang, Jimeng Sun
arXiv:2310.03266 · cs.LG · submitted Oct 5, 2023 · updated Jan 16, 2024
abstract · pdf

add comment on HN