TabPFN用Transformer实现快速高精度表格式数据预测,无需调参。
TabPFN: One Model to Rule Them All?
- 将TabPFN视为近似贝叶斯推断,统一处理参数与非参数结构。
- 在1万样本内数据上性能超越多数现有方法,训练时间大幅缩短。
- 适合统计学研究者、需要快速建模的工业场景使用。
Hollmann等人(Nature 637 (2025) 319-326)最近提出了TabPFN,一种基于Transformer的深度学习模型,用于表格式数据的回归与分类,声称其在不超过10,000样本的数据集上性能远超此前所有方法,且训练时间显著更少。他们还将TabPFN称为表格式数据的‘基础模型’,因其可支持数据生成、密度估计、可复用嵌入学习与微调。本文面向统计学受众,通过强调其作为近似贝叶斯推断的视角,深入解释了TabPFN的工作机制。我们进一步探讨了其对统计学领域的影响:结果显示,无需调参的TabPFN在半监督参数估计、协变量偏移下的预测以及异质性处理效应估计任务中,有时能超越专门设计的先进方法。部分原因在于,该模型可同时适应非参数与参数结构,例如在假设正确设定时,仍可能优于LASSO。所有实验均可通过提供的代码复现(https://github.com/qinglong-tian/tabpfn_study)。
原文摘要 · Abstract (English)
Hollmann et al. (Nature 637 (2025) 319-326) recently introduced TabPFN, a transformer-based deep learning model for regression and classification on tabular data, which they claim "outperforms all previous methods on datasets with up to 10,000 samples by a wide margin, using substantially less training time." Furthermore, they have called TabPFN a "foundation model" for tabular data, as it can support "data generation, density estimation, learning reusable embeddings and fine-tuning". In this paper, we provide a tailored explanation of how TabPFN works for a statistics audience, by emphasizing its interpretation as approximate Bayesian inference. We then explore the significance of TabPFN to the field of statistics: We show that an out-of-the-box application of TabPFN can sometimes outperform specialized state-of-the-art methods for semi-supervised parameter estimation, prediction under covariate shift, and heterogeneous treatment effect estimation. As a partial explanation for the predictive effectiveness of TabPFN, we show that it can simultaneously adapt to both nonparametric structure and parametric structure, for instance, sometimes outperforming LASSO even when assumptions are correctly specified. All experiments can be reproduced using the code provided at https://github.com/qinglong-tian/tabpfn_study (https://github.com/qinglong-tian/tabpfn_study).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。