arXiv:2502.04573cs.LGcs.AI2025-02ICML被引 9

用对抗生成数据训练表格模型,零样本预测更准更快。

Zero-shot Meta-learning for Tabular Prediction Tasks with Adversarially Pre-trained Transformer

  • 用对抗性合成数据预训练,无需真实数据
  • 在小规模分类任务上达顶尖性能,单次推理<1秒
  • 支持任意类别数,适合快速部署的表格预测

我们提出一种对抗性预训练变换器(APT),可在无需任何真实世界数据预训练的情况下,对表格预测任务实现零样本元学习,扩展了先前的先验数据适配网络(PFNs)和TabPFN。APT通过对抗性合成数据代理进行预训练,这些代理持续改变数据生成分布,并主动挑战模型以不同合成数据集。此外,我们提出一种混合块架构,可处理任意类别数的分类任务,解决了先前深度表格零样本学习者的关键缺陷——类别数量限制。实验表明,我们的框架在无数据集特征筛选(如类别数、缺失值数量)的情况下,于小型分类任务中达到当前最优性能,平均运行时间低于1秒。在常见的分类与回归基准数据集上,对抗性预训练显著提升了TabPFN的表现。分析显示,对抗性合成数据代理生成的数据多样性优于TabPFN中普通随机生成器;同时,我们的混合块设计增强了泛化能力并大幅加速了预训练过程。

原文摘要 · Abstract (English)

We present an Adversarially Pre-trained Transformer (APT) that is able to perform zero-shot meta-learning on tabular prediction tasks without pre-training on any real-world dataset, extending on the recent development of Prior-Data Fitted Networks (PFNs) and TabPFN. Specifically, APT is pre-trained with adversarial synthetic data agents, who continue to shift their underlying data generating distribution and deliberately challenge the model with different synthetic datasets. In addition, we propose a mixture block architecture that is able to handle classification tasks with arbitrary number of classes, addressing the class size limitation -- a crucial weakness of prior deep tabular zero-shot learners. In experiments, we show that our framework matches state-of-the-art performance on small classification tasks without filtering on dataset characteristics such as number of classes and number of missing values, while maintaining an average runtime under one second. On common benchmark dataset suites in both classification and regression, we show that adversarial pre-training was able to enhance TabPFN's performance. In our analysis, we demonstrate that the adversarial synthetic data agents were able to generate a more diverse collection of data compared to the ordinary random generator in TabPFN. In addition, we demonstrate that our mixture block neural design has improved generalizability and greatly accelerated pre-training.

表格预测零样本学习对抗训练Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。