arXiv · 1603.02754
XGBoost: A Scalable Tree Boosting System
Abstract
Tree boosting is a highly effective and widely used machine learning method. In this paper, we describe a scalable end-to-end tree boosting system called XGBoost, which is used widely by data scientists to achieve state-of-the-art results on many machine learning challenges. We propose a novel sparsity-aware algorithm for sparse data and weighted quantile sketch for approximate tree learning. More importantly, we provide insights on cache access patterns, data compression and sharding to build a scalable tree boosting system. By combining these insights, XGBoost scales beyond billions of examples using far fewer resources than existing systems.
Explore related subjects
Keep this discovery
Tianqi Chen, Carlos Guestrin. 2016-03-09. XGBoost: A Scalable Tree Boosting System. https://doi.org/10.1145/2939672.2939785
Cite the original work for its findings. Save a collection to share your selection of sources.