用真实数据训练表格大模型,提升泛化能力与训练效率
TabDPT: Scaling Tabular Foundation Models on Real Data
- 结合上下文学习与自监督学习,用真实数据预训练表格模型
- 真实数据预训练使模型更快收敛且在新数据上表现更好
- 模型规模和数据量扩大均带来持续提升,符合幂律规律
表格数据是全球最普遍的信息来源之一,涵盖广泛领域。其固有的异质性阻碍了可快速泛化的表格基础模型(TFM)的发展。上下文学习(ICL)近年来成为解决该问题的有前景方案,支持无需额外调优即可动态适应新任务。尽管已有研究尝试复用大语言模型实现表格ICL,但效果有限,因此近期工作转向构建专用表格基础模型。本文提出一种结合ICL检索与自监督学习的训练方法,并探讨真实数据与合成数据在预训练中的价值。结果表明,真实数据包含合成数据难以捕捉的有效信号。具体而言,在预训练中引入真实数据可显著加速训练并提升对未见数据的下游泛化性能。所提出的模型TabDPT在回归(CTR23)和分类(CC18)基准上表现优异。更重要的是,我们验证了模型规模与数据量扩大均带来一致性能提升,符合幂律规律,暗示大规模表格基础模型具备可行性。代码与模型权重已开源:推理代码与模型位于github.com/layer6ai-labs/TabDPT-inference,训练代码可于github.com/layer6ai-labs/TabDPT-training获取。
原文摘要 · Abstract (English)
Tabular data is one of the most ubiquitous sources of information worldwide, spanning a wide variety of domains. This inherent heterogeneity has slowed the development of Tabular Foundation Models (TFMs) capable of fast generalization to unseen datasets. In-Context Learning (ICL) has recently emerged as a promising solution for TFMs, enabling dynamic adaptation to new tasks without additional tuning. While many studies have attempted to re-purpose large language models for tabular ICL, they have had limited success, so recent works have focused on developing tabular-specific foundation models. In this work, we propose an approach to combine ICL-based retrieval with self supervised learning to train tabular foundation models. We also investigate the utility of real vs. synthetic data for model pre-training, and show that real data can contain useful signal not easily captured in synthetic training. Specifically, we show that incorporating real data during the pre-training phase can lead to significantly faster training and better downstream generalization to unseen data. Our resulting model, TabDPT, achieves strong performance on both regression (CTR23) and classification (CC18) benchmarks. Importantly, we also demonstrate that with our pre-training procedure, scaling both model and data size leads to consistent performance improvements that follow power laws. This echoes scaling laws in LLMs and other foundation models, and suggests that large-scale TFMs can be achievable. We open-source our full pipeline: inference code including trained model weights can be found at github.com/layer6ai-labs/TabDPT-inference, and the training code to reproduce experiments can be found at github.com/layer6ai-labs/TabDPT-training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。