首个表格数据特征漂移基准,揭示模型在真实环境下的脆弱性
TabFSBench: Tabular Benchmark for Feature Shifts in Open Environments
- 构建首个表格数据特征漂移评测基准,覆盖四类场景
- 发现多数模型在特征漂移下性能显著下降,与特征重要性线性相关
- 首次评估大模型在表格任务中对特征漂移的鲁棒性,适合关注真实场景泛化的研究者
表格数据广泛应用于各类机器学习任务。当前研究多聚焦封闭环境,而现实应用常面临开放环境中的分布与特征漂移,导致模型性能严重下降。以往研究主要关注分布漂移,对表格数据特有的特征漂移问题关注不足。本文首次系统研究表格数据中的特征漂移,提出首个表格特征漂移基准(TabFSBench)。该基准评估四种不同特征漂移场景对四类表格模型的影响,并首次在表格基准上测试大语言模型(LLMs)和表格专用大模型的表现。研究发现:(1)多数表格模型在特征漂移场景下适用性有限;(2)特征集偏移程度与模型性能下降呈线性关系;(3)封闭环境下表现良好的模型,在特征漂移场景中也更稳健。同时探讨了未来研究方向。基准代码已开源:https://github.com/LAMDASZ-ML/TabFSBench。
原文摘要 · Abstract (English)
Tabular data is widely utilized in various machine learning tasks. Current tabular learning research predominantly focuses on closed environments, while in real-world applications, open environments are often encountered, where distribution and feature shifts occur, leading to significant degradation in model performance. Previous research has primarily concentrated on mitigating distribution shifts, whereas feature shifts, a distinctive and unexplored challenge of tabular data, have garnered limited attention. To this end, this paper conducts the first comprehensive study on feature shifts in tabular data and introduces the first tabular feature-shift benchmark (TabFSBench). TabFSBench evaluates impacts of four distinct feature-shift scenarios on four tabular model categories across various datasets and assesses the performance of large language models (LLMs) and tabular LLMs in the tabular benchmark for the first time. Our study demonstrates three main observations: (1) most tabular models have the limited applicability in feature-shift scenarios; (2) the shifted feature set importance has a linear relationship with model performance degradation; (3) model performance in closed environments correlates with feature-shift performance. Future research direction is also explored for each observation. Benchmark: https://github.com/LAMDASZ-ML/TabFSBench.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。