arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.02071cs.LGcs.AIstat.ML

针对具有高基数分类变量的数据的机器学习方法比较

A Comparison of Machine Learning Methods for Data with High-Cardinality Categorical Variables

Lucerne University of Applied Sciences and Arts(卢塞恩应用科学与艺术大学)

Fabio Sigrist

AI总结:

本文针对高基数分类变量问题,实证比较树提升、深度神经网络与线性混合效应模型,发现带随机效应的机器学习模型预测准确率更高,且带随机效应的树提升优于带随机效应的深度神经网络。

摘要:

高基数分类变量是指其不同水平的数量相对于数据集的样本量较大,或者说每个水平只有很少数据点的变量。机器学习方法在处理高基数变量时可能会遇到困难。在本文中,我们使用多个包含高基数分类变量的表格数据集,对两种最成功的机器学习方法——树提升(tree-boosting)和深度神经网络——以及线性混合效应模型的若干版本进行了实证比较。我们发现:第一,带随机效应的机器学习模型比不带随机效应的经典对应模型具有更高的预测准确率;第二,带随机效应的树提升方法优于带随机效应的深度神经网络。

英文摘要:

High-cardinality categorical variables are variables for which the number of different levels is large relative to the sample size of a data set, or in other words, there are few data points per level. Machine learning methods can have difficulties with high-cardinality variables. In this article, we empirically compare several versions of two of the most successful machine learning methods, tree-boosting and deep neural networks, and linear mixed effects models using multiple tabular data sets with high-cardinality categorical variables. We find that, first, machine learning models with random effects have higher prediction accuracy than their classical counterparts without random effects, and, second, tree-boosting with random effects outperforms deep neural networks with random effects.

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

↑