打造可持续更新的表格数据机器学习基准测试系统
TabArena: A Living Benchmark for Machine Learning on Tabular Data
- 构建动态维护的表格数据基准,自动更新模型与数据集
- 深度学习在大时间预算下表现追平梯度提升树,小数据上基础模型优势明显
- 鼓励开发者避免验证集过拟合,适合关注表格学习前沿的研究者
随着深度学习和基础模型在表格数据上的普及,标准化可靠基准的需求愈发迫切。然而现有基准多为静态设计,即使发现缺陷或新模型发布也难以更新。为此,我们提出首个持续维护的活体基准系统TabArena。通过人工精选代表性数据集与高质量实现模型,开展大规模基准测试以建立公开排行榜,并组建专业维护团队。结果表明,验证方法与超参数集成对模型性能有显著影响。梯度提升树在实际表格数据中仍具竞争力,但深度学习在更大计算预算下通过集成已迎头赶上。基础模型在小数据集上表现优异。此外,跨模型集成进一步推动了表格学习的性能上限。我们发现部分深度学习模型因验证集过拟合而在集成中过度代表,呼吁开发者改进。目前TabArena已上线公开排行榜、可复现代码及维护规程,网址为https://tabarena.ai。
原文摘要 · Abstract (English)
With the growing popularity of deep learning and foundation models for tabular data, the need for standardized and reliable benchmarks is higher than ever. However, current benchmarks are static. Their design is not updated even if flaws are discovered, model versions are updated, or new models are released. To address this, we introduce TabArena, the first continuously maintained living tabular benchmarking system. To launch TabArena, we manually curate a representative collection of datasets and well-implemented models, conduct a large-scale benchmarking study to initialize a public leaderboard, and assemble a team of experienced maintainers. Our results highlight the influence of validation method and ensembling of hyperparameter configurations to benchmark models at their full potential. While gradient-boosted trees are still strong contenders on practical tabular datasets, we observe that deep learning methods have caught up under larger time budgets with ensembling. At the same time, foundation models excel on smaller datasets. Finally, we show that ensembles across models advance the state-of-the-art in tabular machine learning. We observe that some deep learning models are overrepresented in cross-model ensembles due to validation set overfitting, and we encourage model developers to address this issue. We launch TabArena with a public leaderboard, reproducible code, and maintenance protocols to create a living benchmark available at https://tabarena.ai.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。