表格基础模型泛化特性研究

Understanding the Surprising Generalization Properties of Tabular Foundation Models

精选理由

这篇论文揭示了表格基础模型如何通过单表格预训练实现强迁移,颠覆了传统大规模数据训练的认知。

AI 摘要

研究显示,表格基础模型(TFMs)通过单表格自监督预训练即可实现强迁移能力。表格的有用性取决于特征数量而非实例数量,且与下游任务无关。列级预处理能持续提升下游性能,而数据集级别的过滤或去重则无改善。研究认为TFMs的泛化主要基于检索机制,模型需学会识别相关示例并进行有效聚合。

原文 · arXiv cs.LG

Understanding the Surprising Generalization Properties of Tabular Foundation Models

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.