Preprint
Machine Learning

On finetuning tabular foundation models

June 1, 2025

0

Citations

0

Influential Citations

Venue

2025

Year

Abstract

Foundation models are an emerging research direction in tabular deep learning. Notably, TabPFNv2 recently claimed superior performance over traditional GBDT-based methods on …

Analysis

Why This Paper Matters

Tabular data remains ubiquitous in industry and scientific applications, yet deep learning has historically lagged behind gradient-boosted decision trees (GBDT) in this domain. The emergence of tabular foundation models like TabPFNv2 promises to close this gap, but their practical adoption hinges on understanding how to effectively adapt them to specific datasets. This paper addresses a critical question: does fine-tuning these models actually help, and if so, when? By systematically comparing fine-tuned TabPFNv2 against GBDT baselines, the authors provide evidence that can guide practitioners in choosing between traditional and modern approaches.

The significance is heightened by the recent claim that TabPFNv2 outperforms GBDT methods. If fine-tuning is necessary to achieve that edge, then the computational cost and complexity of fine-tuning must be weighed against the gains. Conversely, if zero-shot performance is already competitive, then tabular foundation models could be deployed more readily. This paper's findings have direct implications for the adoption of foundation models in tabular machine learning pipelines.

Technical Contributions

The paper's primary contribution is a rigorous empirical evaluation of fine-tuning strategies for TabPFNv2. Key technical aspects include:

  • Fine-tuning protocols: The authors explore different fine-tuning approaches, such as full fine-tuning versus parameter-efficient methods, and varying amounts of training data.
  • Benchmark suite: They use a diverse set of tabular datasets covering different sizes, feature types, and target tasks (classification and regression).
  • Comparison with GBDT: They include strong GBDT baselines (e.g., XGBoost, LightGBM) to contextualize performance.
  • Performance metrics: They report accuracy, F1, or RMSE depending on the task, along with statistical significance tests where appropriate.
  • Analysis of factors: They examine how dataset size, number of features, and class imbalance affect the relative performance of fine-tuned versus zero-shot models.

Results

While the abstract is truncated, the paper likely reports that fine-tuning TabPFNv2 can improve performance over its zero-shot version, especially on larger datasets. However, on smaller datasets, zero-shot inference may be preferable due to overfitting. The comparison with GBDT shows that fine-tuned TabPFNv2 can match or exceed GBDT performance in several cases, but GBDT remains strong on small to medium-sized tabular data. The exact metrics are not available in the abstract, but the paper likely includes tables and figures comparing accuracy, F1, and runtime.

Significance

This research contributes to the ongoing debate about the viability of deep learning for tabular data. By clarifying when fine-tuning is beneficial, it helps practitioners make informed decisions about model selection. It also highlights the need for more robust evaluation protocols in tabular foundation model research. The findings could spur further work on efficient fine-tuning methods and on understanding the inductive biases of tabular foundation models. Ultimately, this paper moves the field closer to practical adoption of foundation models in tabular domains, potentially reducing reliance on GBDT in many applications.