English

Faster Boosting with Smaller Memory

Machine Learning 2019-10-29 v3 Machine Learning

Abstract

State-of-the-art implementations of boosting, such as XGBoost and LightGBM, can process large training sets extremely fast. However, this performance requires that the memory size is sufficient to hold a 2-3 multiple of the training set size. This paper presents an alternative approach to implementing the boosted trees, which achieves a significant speedup over XGBoost and LightGBM, especially when the memory size is small. This is achieved using a combination of three techniques: early stopping, effective sample size, and stratified sampling. Our experiments demonstrate a 10-100 speedup over XGBoost when the training data is too large to fit in memory.

Keywords

Cite

@article{arxiv.1901.09047,
  title  = {Faster Boosting with Smaller Memory},
  author = {Julaiti Alafate and Yoav Freund},
  journal= {arXiv preprint arXiv:1901.09047},
  year   = {2019}
}

Comments

NeurIPS 2019

R2 v1 2026-06-23T07:22:36.494Z