用Transformer提升决策树在各种数据量下的表现,融合语言理解与预训练优势。
Transformers Boost the Performance of Decision Trees on Tabular Data across Sample Sizes
- 将LLM或TabPFN的语义理解能力融合进梯度提升树,实现轻量级增强。
- 在中小数据集上性能超越纯变压器模型,在大样本下优于纯决策树。
- 适合需要高精度且数据量变化大的表格数据分析场景。
大型语言模型(LLMs)在零样本和少样本设置下对表格数据表现出色,因其能从描述特征和标签的自然语言列名中提取语义信息。类似地,最近提出的非LLM型变换器TabPFN在数千样本以内的数据集上也展现了优异性能。相比之下,梯度提升决策树(GBDTs)通常从头训练,缺乏预训练数据支持,且无法理解自然语言,仅依赖数据条目学习列间关系。尽管LLMs和TabPFN在小规模数据集上因强先验而表现优异,但在中等或大规模数据集上不如GBDTs,因其上下文长度受限。本文提出一种简单轻量的融合方法,将大型语言模型与TabPFN与梯度提升决策树结合,使可扩展的GBDT能受益于变换器的语言理解和预训练优势。我们分别命名为LLM-Boost和PFN-Boost。在足够小的数据集上,其性能可匹敌或超越变换器;在足够大的数据集上,则优于原始决策树。在中等范围数据集上,两者均显著优于单一组件。实验表明,在除极小数据集外的所有情况下,PFN-Boost在所有测试方法中平均表现最佳。代码已开源:http://github.com/MayukaJ/LLM-Boost。
原文摘要 · Abstract (English)
Large language models (LLMs) perform remarkably well on tabular datasets in zero- and few-shot settings, since they can extract meaning from natural language column headers that describe features and labels. Similarly, TabPFN, a recent non-LLM transformer pretrained on numerous tables for in-context learning, has demonstrated excellent performance for dataset sizes up to a thousand samples. In contrast, gradient-boosted decision trees (GBDTs) are typically trained from scratch on each dataset without benefiting from pretraining data and must learn the relationships between columns from their entries alone since they lack natural language understanding. LLMs and TabPFN excel on small tabular datasets where a strong prior is essential, yet they are not competitive with GBDTs on medium or large datasets, since their context lengths are limited. In this paper, we propose a simple and lightweight approach for fusing large language models and TabPFN with gradient-boosted decision trees, which allows scalable GBDTs to benefit from the natural language capabilities and pretraining of transformers. We name our fusion methods LLM-Boost and PFN-Boost, respectively. While matching or surpassing the performance of the transformer at sufficiently small dataset sizes and GBDTs at sufficiently large sizes, LLM-Boost and PFN-Boost outperform both standalone components on a wide range of dataset sizes in between. We demonstrate state-of-the-art performance against numerous baselines and ensembling algorithms. We find that PFN-Boost achieves the best average performance among all methods we test for all but very small dataset sizes. We release our code at http://github.com/MayukaJ/LLM-Boost .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。