arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2108.08790cs.LG
收藏

简单更好:使用随机抽样使决策树更快

Simple is better: Making Decision Trees faster using random sampling

AI Labs, American Express(美国运通AI实验室)

Vignesh Nanda Kumar, Narayanan U Edakunni

摘要:

近年来,梯度提升的决策树在大数据上构建鲁棒的机器学习模型中变得流行。使这些算法成功的主要技术是在构建决策树的同时分布计算。分布式决策树的构建,反过来,已经通过构建大数据集的分位数和从这些分位数集中选择候选分裂点来实现。例如,在XGBoost中,采用复杂的分位数构建算法来识别决策树的候选分裂点。当计算是分布式的时,这种方法通常会产生更好的结果。在本文中,我们消除的概念,这些方法提供了更准确和可扩展的方法,以分布式的方式建立决策树。在一个显着的贡献,我们从理论上和经验表明,随机选择均匀的分裂点提供了相同的,甚至更好的性能方面的准确性和计算效率。因此,与更复杂的方法相比,点的简单随机选择足以用于决策树构建。

英文摘要:

In recent years, gradient boosted decision trees have become popular in building robust machine learning models on big data. The primary technique that has enabled these algorithms success has been distributing the computation while building the decision trees. A distributed decision tree building, in turn, has been enabled by building quantiles of the big datasets and choosing the candidate split points from these quantile sets. In XGBoost, for instance, a sophisticated quantile building algorithm is employed to identify the candidate split points for the decision trees. This method is often projected to yield better results when the computation is distributed. In this paper, we dispel the notion that these methods provide more accurate and scalable methods for building decision trees in a distributed manner. In a significant contribution, we show theoretically and empirically that choosing the split points uniformly at random provides the same or even better performance in terms of accuracy and computational efficiency. Hence, a simple random selection of points suffices for decision tree building compared to more sophisticated methods.

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

↑