Folia
← 返回头版

TabICL在表格分类任务上全面超越调优XGBoost

研究者针对表格基础模型TabICL与TabPFN和调优XGBoost的性能进行了系统基准测试。1在14个数据集的对比中,TabICL在精确度指标上赢得12场并打成1平1负,在曲线下面积(AUC)指标上实现14连胜。1两个模型的平均精确度差异为0.01,TabICL平均精确度提升达+0.0106。1

推理效率方面,TabICL展现出明显优势。1TabICL的推理时间普遍在1秒以内,而调优XGBoost需耗时0.7至27.2秒。1在32000行数据规模下,TabICL最差种子得分(0.8216)仍优于XGBoost最佳种子(0.7950),两者无重叠区间。1此外,TabICL无需进行超参数调优,进一步简化了使用流程。1

然而,性能优势并非在所有场景下都保持一致。1在列数众多的宽表上,模型性能出现下降,Bioresponse数据集(419列)尤为典型,TabPFN 2.2.1版本在该数据集上的得分跌至0.7567,耗时18.1秒。1

测试使用了14个截至3000行的数据集,采用5个随机种子取中位数的方式,硬件配置为RTX 4070 Ti SUPER GPU。1在许可证方面,TabICL采用三条款BSD许可证可无障碍下载,而TabPFN 8.3.0版本因许可证障碍无法获取,测试采用了2.2.1版本。1


评论