用对抗性合成数据训练表格大模型,提升鲁棒性和性能
Robust Tabular Foundation Models
- 通过优化生成器让模型面对更具挑战性的合成数据
- 在基准测试中平均AUC提升6%,仅需少于10万条合成数据
- 适合想用纯合成数据提升表格模型性能的研究者
近年来,表格基础模型(TFM)发展迅速,展现出超越传统机器学习方法的潜力。关键发现是,TFM可在纯合成数据上进行预训练,这为设计能引导理想模型特性的数据生成器提供了可能。以往工作主要聚焦于构建高质量生成器先验以提升预训练性能。本文提出新视角:参数化生成器分布后,可从对抗鲁棒性角度出发——训练时调整生成器,使其突出对模型更具挑战性的数据。为此,我们引入最优性差距度量,即模型性能与强基线(如XGBoost、CatBoost、Random Forests)所能达到的最佳性能之差。基于此,提出模型无关的对抗训练框架RTFM。应用于TabPFN V2分类器,RTFM在基准测试中表现更优,平均标准化AUC相比原版提升达6%,且仅需不到10万条额外合成数据。结果表明,仅用合成数据即可实现目标导向的对抗训练与微调,开辟新方向。
原文摘要 · Abstract (English)
The development of tabular foundation models (TFMs) has accelerated in recent years, showing strong potential to outperform traditional ML methods for structured data. A key finding is that TFMs can be pretrained entirely on synthetic datasets, opening opportunities to design data generators that encourage desirable model properties. Prior work has mainly focused on crafting high-quality priors over generators to improve overall pretraining performance. Our insight is that parameterizing the generator distribution enables an adversarial robustness perspective: during training, we can adapt the generator to emphasize datasets that are particularly challenging for the model. We formalize this by introducing an optimality gap measure, given by the difference between TFM performance and the best achievable performance as estimated by strong baselines such as XGBoost, CatBoost, and Random Forests. Building on this idea, we propose Robust Tabular Foundation Models (RTFM), a model-agnostic adversarial training framework. Applied to the TabPFN V2 classifier, RTFM improves benchmark performance, with up to a 6% increase in mean normalized AUC over the original TabPFN and other baseline algorithms, while requiring less than 100k additional synthetic datasets. These results highlight a promising new direction for targeted adversarial training and fine-tuning of TFMs using synthetic data alone.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。