arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02112cs.LGcs.AI
收藏

评估表格数据机器学习模型的数据中心视角

A Data-Centric Perspective on Evaluating Machine Learning Models for Tabular Data

University of Mannheim(曼海姆大学);University of Rostock(罗斯托克大学)

Andrej Tschalzev, Sascha Marton, Stefan Lüdtke, Christian Bartelt, Heiner Stuckenschmidt

AI总结:

针对表格数据模型评估中过度标准化预处理的偏差问题,提出数据中心评估框架,发现针对特定数据集的特征工程会显著改变模型排名并降低模型选择重要性,强调表格数据需要特征工程并具有时间特征。

摘要:

表格数据在现实世界的机器学习应用中非常普遍,针对表格数据监督学习的新模型也经常被提出。评估模型性能的比较研究通常采用以模型为中心的评估设置,并使用过度标准化的数据预处理。本文表明,这种以模型为中心的评估存在偏差,因为现实世界的建模流程通常需要针对特定数据集的预处理和特征工程。因此,我们提出了一个数据中心评估框架。我们从 Kaggle 竞赛中选择了 10 个相关数据集,并为每个数据集实现了专家级的预处理流程。我们使用不同的预处理流程和超参数优化(HPO)机制进行实验,以量化模型选择、HPO、特征工程和测试时适应的影响。我们的主要发现是:1. 在进行针对特定数据集的特征工程后,模型排名发生显著变化,性能差异减小,模型选择的重要性降低。2. 最近的模型尽管取得了可衡量的进展,但仍然显著受益于手动特征工程。这对于基于树的模型和神经网络都适用。3. 虽然表格数据通常被认为是静态的,但样本往往是随时间收集的,适应分布偏移即使在所谓的静态数据中也可能很重要。这些见解表明,研究工作应转向数据中心视角,承认表格数据需要特征工程并且通常表现出时间特征。我们的框架可在以下网址获取:https://github.com/atschalz/dc_tabeval。

英文摘要:

Tabular data is prevalent in real-world machine learning applications, and new models for supervised learning of tabular data are frequently proposed. Comparative studies assessing the performance of models typically consist of model-centric evaluation setups with overly standardized data preprocessing. This paper demonstrates that such model-centric evaluations are biased, as real-world modeling pipelines often require dataset-specific preprocessing and feature engineering. Therefore, we propose a data-centric evaluation framework. We select 10 relevant datasets from Kaggle competitions and implement expert-level preprocessing pipelines for each dataset. We conduct experiments with different preprocessing pipelines and hyperparameter optimization (HPO) regimes to quantify the impact of model selection, HPO, feature engineering, and test-time adaptation. Our main findings are: 1. After dataset-specific feature engineering, model rankings change considerably, performance differences decrease, and the importance of model selection reduces. 2. Recent models, despite their measurable progress, still significantly benefit from manual feature engineering. This holds true for both tree-based models and neural networks. 3. While tabular data is typically considered static, samples are often collected over time, and adapting to distribution shifts can be important even in supposedly static data. These insights suggest that research efforts should be directed toward a data-centric perspective, acknowledging that tabular data requires feature engineering and often exhibits temporal characteristics. Our framework is available under: https://github.com/atschalz/dc_tabeval.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑