Skip to content

autogluon/tabarena

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

1,346 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
    TabArena Logo

A Living Benchmark for Machine Learning on Tabular Data πŸ’«


πŸš€ Leaderboard πŸ“‚ Example Scripts πŸ“Š Dataset Curation πŸ“„ Papers: TabArena-v0.1 Β· BeyondArena

TabArena is a living benchmarking system that makes benchmarking tabular machine learning models a reliable experience. TabArena implements best practices to ensure methods are represented at their peak potential, including cross-validated ensembles, strong hyperparameter search spaces contributed by the method authors, early stopping, model refitting, parallel bagging, memory usage estimation, and more. Explore the latest results on the live leaderboard.

This single codebase powers two complementary benchmarks that share the same fitting, runner, and evaluation code:

  • 🏟️ TabArena-v0.1 β€” the living benchmark on curated, IID tabular datasets.
  • 🌍 BeyondArena β€” a holistic, beyond-IID benchmark spanning IID, temporal, and grouped tasks across a wide range of dataset sizes and feature dimensionalities. BeyondArena will superseed TabArena-v0.1 in the future.

Tip

New here? Start with TabArena, then graduate to BeyondArena. Get your model working and competitive on TabArena's curated IID datasets first; once it holds up there, run the same code on BeyondArena to stress-test how well it generalizes beyond IID.

TabArena covers 51 curated datasets (9–30 splits each) and 27+ methods, including 10+ tabular foundation models β€” over 50M trained models, with all validation and test predictions cached for tuning and post-hoc ensembling. BeyondArena extends this to 142 datasets across IID, temporal, and grouped task types, spanning tiny to 1M-row datasets and low- to high-dimensional features.

⚑ Quickstart

Tip

The fastest way to try TabArena end-to-end:

pip install uv
git clone https://github.com/autogluon/tabarena.git && cd tabarena
uv venv --seed --python 3.12 && source .venv/bin/activate
uv pip install --prerelease=allow -e "./packages/tabarena[benchmark]"
python examples/benchmarking/run_quickstart_tabarena.py

For other install paths (eval-only, editable AutoGluon, dependency), see Installation below. To try BeyondArena instead, run python examples/beyondarena/run_quickstart_beyondarena.py with the same install.

πŸ•ΉοΈ Use Cases

We share more details on various use cases of TabArena in our examples:

Datasets

Please refer to our dataset curation repository to learn more about or contributed data!

More Documentation

TabArena code is currently being polished. Detailed Documentation for TabArena will be available soon.

πŸͺ„ Installation

Important

Requires Python 3.11–3.13 and uv.

TabArena is a uv workspace; its installable packages live under packages/ (tabarena, bencheval, tabflow_slurm). Install the tabarena package directly from packages/tabarena with the extras you need. The --prerelease=allow flag is required so uv resolves the pre-release dependency.

First clone the repo and create a virtual environment (one time):

git clone https://github.com/autogluon/tabarena.git
cd tabarena
uv venv --seed --python 3.12
source .venv/bin/activate

Then pick the install path that matches what you want to do:

πŸ“Š Evaluation only β€” leaderboards, metrics & plots, no model fitting

Loads cached results and computes/plots leaderboards & metrics (ELO, win-rates, ranks). Depends on autogluon.tabular (not the full AutoGluon meta-package) β€” no model-fitting libraries and no torch.

uv pip install --prerelease=allow -e "./packages/tabarena[plot]"
πŸš€ Benchmark β€” core set of models for benchmarking

Installs the core models used for standard benchmarking: tabpfn, tabicl, ebm, search_spaces, realmlp, tabdpt, tabm.

uv pip install --prerelease=allow -e "./packages/tabarena[benchmark]"
βž• Benchmark + Extended β€” core models plus the extended model set

The extended extra is experimental and may fail to resolve or install due to incompatible version requirements across model dependencies. Use it only if you specifically need every model in a single environment; otherwise prefer benchmark or benchmark plus one specific model.

Layers the extended model set (modernnca, xrfm, sap-rpt-oss, ...) on top of the core benchmark set.

uv pip install --prerelease=allow -e "./packages/tabarena[benchmark,extended]"

To install only one extended model on top of benchmark (recommended over extended when you only need a single extra model), pass its extra by name β€” for example, just xrfm:

uv pip install --prerelease=allow -e "./packages/tabarena[benchmark,xrfm]"
πŸ› οΈ Developer β€” editable AutoGluon + editable TabArena

Create a virtual environment in your workspace directory (it spans both repos cloned below, so .venv lives at the workspace root rather than inside either repo):

uv venv --seed --python 3.12 .venv
source .venv/bin/activate

Install editable AutoGluon and TabArena:

git clone https://github.com/autogluon/autogluon.git
./autogluon/full_install.sh

git clone https://github.com/autogluon/tabarena.git
uv pip install --prerelease=allow -e "./tabarena/packages/tabarena[benchmark]"

In PyCharm, mark packages/tabarena/src/ and each autogluon/src/ subdirectory as Sources Root so imports resolve.

πŸ“¦ Use TabArena as a dependency

Add the following to your project's dependencies:

# TabArena depends on a pre-release of AutoGluon, so allow pre-releases when installing
# (e.g. `uv pip install --prerelease=allow ...` or `pip install --pre ...`).
# Alternatively, pin AutoGluon to a specific pre-release (an exact `==` pin resolves a
# pre-release without the flag), e.g. add `"autogluon.tabular==1.5.1b20260626"`.
"tabarena @ git+https://github.com/autogluon/tabarena.git#subdirectory=packages/tabarena"

πŸ“¦ TabArena Artifacts

TabArena caches predictions, results, and leaderboards as downloadable artifacts so you can reproduce or extend any analysis without re-running the benchmark.

Artifact tiers, sizes, and examples

Artifacts download to ~/.cache/tabarena/ by default. Override the location with the TABARENA_CACHE environment variable.

Raw data is ~100 GB per method type. Point TABARENA_CACHE at a large disk before downloading it.

Tier Contents Size / method Example
Raw data Per-child test predictions, full metadata, system info ~100 GB inspect_raw_data_and_verify_splits.py
Processed data Minimal data for HPO simulation, portfolios, leaderboards ~10 GB inspect_processed_data.py
Results Per-config / HPO DataFrames (test error, val error, train time, inference time) <1 MB run_generate_main_leaderboard.py
Leaderboards Aggregated ELO, win-rate, average rank, improvability <1 MB β€”
Figures & Plots Generated from results and leaderboards β€” β€”

πŸ“„ Citation

If you use this code in a scientific publication, please cite the relevant paper(s): TabArena for the living IID benchmark, and BeyondArena for the beyond-IID benchmark.

TabArena

TabArena: A Living Benchmark for Machine Learning on Tabular Data Nick Erickson, Lennart Purucker, Andrej Tschalzev, David HolzmΓΌller, Prateek Mutalik Desai, David Salinas, Frank Hutter NeurIPS 2025, Datasets and Benchmarks Track

πŸ“„ arXiv Β· 🎀 NeurIPS poster & video

BibTeX

The entry uses year=2026 because NeurIPS'25 proceedings are published in 2026.

@article{erickson2026tabarena,
  title   = {TabArena: A Living Benchmark for Machine Learning on Tabular Data},
  author  = {Erickson, Nick and Purucker, Lennart and Tschalzev, Andrej and Holzm{\"u}ller, David and Desai, Prateek and Salinas, David and Hutter, Frank},
  journal = {Advances in Neural Information Processing Systems},
  volume  = {38},
  year    = {2026}
}

BeyondArena

Beyond IID: How General Are Tabular Foundation Models, Really? Lennart Purucker, Andrej Tschalzev, Nick Erickson, Gioia Blayer, David HolzmΓΌller, Alan Arazi, Alexander Pfefferle, Mustafa Tajjar, GaΓ«l Varoquaux, Frank Hutter

πŸ“„ arXiv

BibTeX
@misc{purucker2026beyondiid,
  title         = {Beyond IID: How General Are Tabular Foundation Models, Really?},
  author        = {Purucker, Lennart and Tschalzev, Andrej and Erickson, Nick and Blayer, Gioia and Holzm{\"u}ller, David and Arazi, Alan and Pfefferle, Alexander and Tajjar, Mustafa and Varoquaux, Ga{\"e}l and Hutter, Frank},
  year          = {2026},
  eprint        = {2606.30410},
  archivePrefix = {arXiv},
  primaryClass  = {cs.LG},
  url           = {https://arxiv.org/abs/2606.30410}
}

Relation to TabRepo

TabArena was built upon and now replaces TabRepo. To see details about TabRepo, the portfolio simulation repository, refer to tabrepo.md.

About

A Living Benchmark for Machine Learning on Tabular Data

Resources

License

Code of conduct

Contributing

Security policy

Stars

261 stars

Watchers

11 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages