TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Salinas, David, Erickson, Nick
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929473406894080
author Salinas, David
Erickson, Nick
author_facet Salinas, David
Erickson, Nick
contents We introduce TabRepo, a new dataset of tabular model evaluations and predictions. TabRepo contains the predictions and metrics of 1310 models evaluated on 200 classification and regression datasets. We illustrate the benefit of our dataset in multiple ways. First, we show that it allows to perform analysis such as comparing Hyperparameter Optimization against current AutoML systems while also considering ensembling at marginal cost by using precomputed model predictions. Second, we show that our dataset can be readily leveraged to perform transfer-learning. In particular, we show that applying standard transfer-learning techniques allows to outperform current state-of-the-art tabular systems in accuracy, runtime and latency.
format Preprint
id arxiv_https___arxiv_org_abs_2311_02971
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications
Salinas, David
Erickson, Nick
Machine Learning
Artificial Intelligence
We introduce TabRepo, a new dataset of tabular model evaluations and predictions. TabRepo contains the predictions and metrics of 1310 models evaluated on 200 classification and regression datasets. We illustrate the benefit of our dataset in multiple ways. First, we show that it allows to perform analysis such as comparing Hyperparameter Optimization against current AutoML systems while also considering ensembling at marginal cost by using precomputed model predictions. Second, we show that our dataset can be readily leveraged to perform transfer-learning. In particular, we show that applying standard transfer-learning techniques allows to outperform current state-of-the-art tabular systems in accuracy, runtime and latency.
title TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applications
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2311.02971