单张真实表格也能训练出强迁移能力的表格基础模型。
Understanding the Surprising Generalization Properties of Tabular Foundation Models

- 用单个真实表格自监督预训练,即可实现强大迁移性能。
- 表格实用性主要由特征数决定,而非样本数量。
- 模型泛化依赖于检索和聚合相关示例的能力,适合任务驱动设计。
表格基础模型(TFMs)越来越多地依赖上下文学习,在推理时接收带标签示例并预测新输入,不更新权重。现有TFMs通常在大规模合成数据或真实数据集集合上训练。本文发现,仅用单个真实表格进行自监督预训练,也能产生显著的迁移能力。在此设定下,我们观察到表格要么普遍有用,要么普遍无用,且其有用性主要取决于特征数量而非实例数量。这揭示了任务中心的预训练视角:任务的数量与质量对TFM预训练至关重要。该视角还可指导大规模语料库设计:细粒度的列级预处理持续提升下游性能,而数据集级别的过滤或去重则无益。最后,我们提出新的泛化理解:表格上下文泛化本质上是基于检索的,优秀模型能准确识别上下文中的相关示例并有效聚合。这一任务中心、检索驱动的视角为未来模型与语料设计提供了新框架。
原文摘要 · Abstract (English)
Tabular Foundation Models (TFMs) increasingly rely on in-context learning, where a model receives labelled examples at inference time and predicts labels for new inputs without updating its weights. Existing TFMs are typically trained on either massive synthetic corpora or very large collections of real datasets. In contrast, we show that surprisingly strong transfer can emerge from self-supervised pre-training on just a single real table. In this setting, we also find that tables tend to be either broadly useful or broadly poor regardless of downstream prediction task, and that the strongest predictor of usefulness is the number of features rather than the number of instances. This leads to a task-centric interpretation of tabular pre-training: the number and the quality of tasks are essential for the pre-training of TFMs. We show that the same task-centric perspective can help corpus design at scale: fine-grained column-level pre-processing consistently improves downstream performance, while no improvements are observed when we filter or deduplicate at the dataset level. Finally, we offer a new perspective for how TFMs generalize: we believe that tabular in-context generalization is largely retrieval-based, and good models are those that learn to identify relevant examples in the provided context and aggregate them well. The mechanics of TFMs have been relatively understudied; our task-centric, retrieval-based perspective offers a new framework to guide future model and corpus design.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。