arXiv:2506.08982cs.LG2025-06被引 14

微调让表格基础模型更准,尤其在小数据上表现卓越。

On Finetuning Tabular Foundation Models

  • 全量微调是TabPFNv2最高效实用的适配方法。
  • 微调后模型能更准确地捕捉测试样本与训练样本的相似性。
  • 适合追求高精度的表格数据任务,尤其小数据场景。

基础模型是表格深度学习的新兴方向。近期,TabPFNv2在小规模数据集上以零参数调整的上下文学习范式超越了传统基于GBDT的方法。然而,如何最优地微调此类模型及其对内部机制的影响仍不明确。我们系统评估了多种微调策略,发现全量微调在效率与效果上表现最佳。进一步分析显示,微调使测试样本的查询表示与上下文训练样本的键表示之间的点积更准确反映其目标相似性,从而改进基于检索的预测逻辑。实践中,我们在最大5万样本的数据集上成功微调,几乎所有任务均获性能提升;在独立同分布(I.I.D.)学术数据集上达到当前最优结果,但在存在渐进时间偏移和丰富特征的数据集上稳定性较差,此时传统方法仍更优。

原文摘要 · Abstract (English)

Foundation models are an emerging research direction in tabular deep learning. Notably, TabPFNv2 recently claimed superior performance over traditional GBDT-based methods on small-scale datasets using an in-context learning paradigm, which does not adapt model parameters to target datasets. However, the optimal finetuning approach for adapting tabular foundational models, and how this adaptation reshapes their internal mechanisms, remains underexplored. While prior works studied finetuning for earlier foundational models, inconsistent findings and TabPFNv2's unique architecture necessitate fresh investigation. To address these questions, we first systematically evaluate various finetuning strategies on diverse datasets. Our findings establish full finetuning as the most practical solution for TabPFNv2 in terms of time-efficiency and effectiveness. We then investigate how finetuning alters TabPFNv2's inner mechanisms, drawing an analogy to retrieval-augmented models. We reveal that the success of finetuning stems from the fact that after gradient-based adaptation, the dot products of the query-representations of test objects and the key-representations of in-context training objects more accurately reflect their target similarity. This improved similarity allows finetuned TabPFNv2 to better approximate target dependency by appropriately weighting relevant in-context samples, improving the retrieval-based prediction logic. From the practical perspective, we managed to finetune TabPFNv2 on datasets with up to 50K objects, observing performance improvements on almost all tasks. More precisely, on academic datasets with I.I.D. splits, finetuning allows TabPFNv2 to achieve state-of-the-art results, while on datasets with gradual temporal shifts and rich feature sets, TabPFNv2 is less stable and prior methods remain better.

表格模型微调基础模型机器学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。