arXiv:2502.02527cs.LG2025-02ICML被引 45

改进TabPFN,让表格分类模型更准更快、适应更大数据

TabPFN Unleashed: A Scalable and Effective Solution to Tabular Classification Problems

  • 用轻量编码器对齐下游任务,降低偏差
  • 结合自助采样和多编码器,显著减少方差
  • 保持高效推理,适合大规模表格数据

TabPFN是一种有前景的上下文学习模型,可直接根据标注的训练样本预测测试样本标签,在小规模分类任务中表现优异。然而,其在高维特征处理、下游数据对齐和大规模数据扩展方面仍有提升空间。本文提出Beta方法,通过轻量级编码器增强任务对齐以降低偏差,并通过多编码器结构与自助采样有效缓解方差问题,同时保持推理效率。该方法显著提升了TabPFN在高维数据和大样本场景下的性能。在超过200个基准分类数据集上的实验表明,Beta方法要么超越、要么匹配现有最佳模型。

原文摘要 · Abstract (English)

TabPFN has emerged as a promising in-context learning model for tabular data, capable of directly predicting the labels of test samples given labeled training examples. It has demonstrated competitive performance, particularly on small-scale classification tasks. However, despite its effectiveness, TabPFN still requires further refinement in several areas, including handling high-dimensional features, aligning with downstream datasets, and scaling to larger datasets. In this paper, we revisit existing variants of TabPFN and observe that most approaches focus either on reducing bias or variance, often neglecting the need to address the other side, while also increasing inference overhead. To fill this gap, we propose Beta (Bagging and Encoder-based Fine-tuning for TabPFN Adaptation), a novel and effective method designed to minimize both bias and variance. To reduce bias, we introduce a lightweight encoder to better align downstream tasks with the pre-trained TabPFN. By increasing the number of encoders in a lightweight manner, Beta mitigate variance, thereby further improving the model's performance. Additionally, bootstrapped sampling is employed to further reduce the impact of data perturbations on the model, all while maintaining computational efficiency during inference. Our approach enhances TabPFN's ability to handle high-dimensional data and scale to larger datasets. Experimental results on over 200 benchmark classification datasets demonstrate that Beta either outperforms or matches state-of-the-art methods.

表格分类上下文学习模型优化高效推理

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