arXiv:2602.05649cs.LG2026-02中稿 · ICML被引 4

用压缩数据提升表格模型推理速度,快94倍且省97%内存。

End-to-End Compression for Tabular Foundation Models

  • 将训练数据压缩到隐空间,实现端到端加速。
  • 在TabArena上推理快94倍,内存减少97%,性能几乎不变。
  • 适合处理大规模表格数据的高效建模场景。

梯度提升树长期主导表格数据建模,但近期基于上下文学习的表格基础模型挑战了这一地位。这类方法通过将训练数据作为上下文,在一次前向传播中完成拟合与预测,无需参数更新。尽管当前表格基础模型表现优异,其基于注意力机制的Transformer架构存在与数据集规模呈二次复杂度的问题,导致训练和推理开销大,难以处理大规模数据。本文提出TACO,一种端到端的表格压缩模型,可将训练数据在隐空间中压缩。在TabArena基准测试中,该方法推理速度最高提升94倍,内存占用减少高达97%,同时保持性能无显著下降。此外,该方法在数据规模增大时仍具更好扩展性,并优于其他基线模型。

原文摘要 · Abstract (English)

The long-standing dominance of gradient-boosted decision trees for tabular data has recently been challenged by in-context learning tabular foundation models. In-context learning methods fit and predict in one forward pass without parameter updates by leveraging the training data as context for predicting on query test points. While recent tabular foundation models achieve state-of-the-art performance, their transformer architecture based on the attention mechanism has quadratic complexity regarding dataset size, which in turn increases the overhead on training and inference time, and limits the capacity of the models to handle large-scale datasets. In this work, we propose TACO, an end-to-end tabular compression model that compresses the training dataset in a latent space. We test our method on the TabArena benchmark, where our proposed method is up to 94x faster in inference time, while consuming up to 97\% less memory compared to the state-of-the-art tabular transformer architecture, all while retaining performance without significant degradation. Lastly, our method not only scales better with increased dataset sizes, but it also achieves better performance compared to other baselines.

表格模型压缩Transformer加速

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