• Cite Icon45
  • https://doi.org/10.1145/3183713.3196892Copy DOI Icon

DimBoost

  • May 27, 2018
  • Jiawei Jiang +3 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Gradient boosting decision tree (GBDT) is one of the most popular machine learning models widely used in both academia and industry. Although GBDT has been widely supported by existing systems such as XGBoost, LightGBM, and MLlib, one system bottleneck appears when the dimensionality of the data becomes high. As a result, when we tried to support our industrial partner on datasets of the dimension up to 330K, we observed suboptimal performance for all these aforementioned systems. In this paper, we ask Can we build a scalable GBDT training system whose performance scales better with respect to dimensionality of the data? The first contribution of this paper is a careful investigation of existing systems by developing a performance model with respect to the dimensionality of the data. We find that the collective communication operations in many existing systems only implement the algorithm designed for small messages. By just fixing this problem, we are able to speed up these systems by up to 2X. Our second contribution is a series of optimizations to further optimize the performance of collective communications. These optimizations include a task scheduler, a two-phase split finding method, and low-precision gradient histograms. Our third contribution is a sparsity-aware algorithm to build gradient histograms and a novel index structure to build histograms in parallel. We implement these optimizations in DimBoost and show that it can be 2-9X faster than existing systems.

Similar Papers
  • Research Article
  • Citations16

Improving the capability of water vapor retrieval from Landsat 8 using ensemble machine learning

  • Jun 28, 2023
  • International Journal of Applied Earth Observation and Geoinformation
  • Yuhao Wu +6
  • Research Article
  • Citations154

Machine learning-based landslide susceptibility assessment with optimized ratio of landslide to non-landslide samples

  • May 25, 2022
  • Gondwana Research
  • Can Yang +4
  • PDF
  • Research Article
  • Citations50

Wind Speed Forecast Based on Post-Processing of Numerical Weather Predictions Using a Gradient Boosting Decision Tree Algorithm

  • Jul 12, 2020
  • Atmosphere
  • Wenqing Xu +2
  • Conference Article

Collaborative Implementation of Hardware-Oriented GBDT Compress Algorithm Based on DSP+FPGA

  • Oct 01, 2019
  • Yafei Li +4
  • Book Chapter
  • Citations17

ACCT: Automatic Collective Communications Tuning

  • Jan 01, 2000
  • Graham E Fagg +2
  • Research Article
  • Citations19

Distribution network topology identification based on gradient boosting decision tree and attribute weighted naive Bayes

  • May 03, 2023
  • Energy Reports
  • Wenkai Guo +3
  • Book Chapter
  • Citations2

Terminal Access Data Anomaly Detection Based on GBDT for Power User Electric Energy Data Acquisition System

  • Jan 01, 2019
  • Qian Ma +4
  • Research Article

DF-BETA: An FPGA-based Memory Locality Aware Decision Forest Accelerator via Bit-Level Early Termination

  • Mar 21, 2025
  • ACM Transactions on Reconfigurable Technology and Systems
  • Daichi Tokuda +1
  • Research Article
  • Citations50

Prediction model for daily reference crop evapotranspiration based on hybrid algorithm and principal components analysis in Southwest China

  • Sep 04, 2021
  • Computers and Electronics in Agriculture
  • Long Zhao +4
  • Research Article
  • Citations2

Topological embedding and directional feature importance in ensemble classifiers for multi-class classification

  • Nov 13, 2024
  • Computational and Structural Biotechnology Journal
  • Eloisa Rocha Liedl +3
  • PDF
  • Research Article
  • Citations41

Artificial Intelligence–Based Multimodal Risk Assessment Model for Surgical Site Infection (AMRAMS): Development and Validation Study

  • Jun 15, 2020
  • JMIR Medical Informatics
  • Weijia Chen +5
  • Research Article
  • Citations4

Glass forming ability prediction of bulk metallic glasses based on fused strategy

  • May 01, 2024
  • Transactions of Nonferrous Metals Society of China
  • Ting Zhang +2
  • Research Article
  • Citations12

Develop a radiomics-based machine learning model to predict the stone-free rate post-percutaneous nephrolithotomy.

  • Apr 13, 2024
  • Urolithiasis
  • Xin Chang Zou +5
  • Research Article

Machine Learning Prediction Models for Colorectal Cancer Based on the Novel Ensemble Framework

  • Jun 09, 2025
  • Applied and Computational Engineering
  • Qing Mi
  • Book Chapter

Entity Relation Extraction of Traditional Chinese Medicine Influenza Based on Bi-GRU+GBDT

  • Jan 01, 2023
  • Lecture notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering
  • Yanhua Zhao +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.