Abstract:In this paper, we present a novel parallel implementation for training Gradient Boosting Decision Trees (GBDTs) on Graphics Processing Units (GPUs). Thanks to the excellent results on classification/regression and the open sourced libraries such as XGBoost, GBDTs have become very popular in recent years and won many awards in machine learning and data mining competitions. Although GPUs have demonstrated their success in accelerating many machine learning applications, it is challenging to develop an efficient GPU-based GBDT algorithm. The key challenges include irregular memory accesses, many sorting operations with small inputs and varying data parallel granularities in tree construction. To tackle these challenges on GPUs, we propose various novel techniques including (i) Run-length Encoding compression and thread/block workload dynamic allocation, (ii) data partitioning based on stable sort, and fast and memory efficient attribute ID lookup in node splitting, (iii) finding approximate split points using two-stage histogram building, (iv) building histograms with the aware of sparsity and exploiting histogram subtraction to reduce histogram building workload, (v) reusing intermediate training results for efficient gradient computation, and (vi) exploiting multiple GPUs to handle larger data sets efficiently. Our experimental results show that our algorithm named <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">ThunderGBM</italic> can be 10x times faster than the state-of-the-art libraries (i.e., XGBoost, LightGBM and CatBoost) running on a relatively high-end workstation of 20 CPU cores. In comparison with the libraries on GPUs, ThunderGBM can handle higher dimensional problems which the libraries become extremely slow or simply fail. For the data sets the existing libraries on GPUs can handle, ThunderGBM achieves up to 10 times speedup on the same hardware, which demonstrates the significance of our GPU optimizations. Moreover, the models trained by ThunderGBM are identical to those trained by XGBoost, and have similar quality as those trained by LightGBM and CatBoost.

Fast Boosting Trees for Classification, Pose Detection, and Boundary Detection on a GPU.

InstantTrace: Fast Parallel Neuron Tracing on GPUs

GPU-acceleration for Large-scale Tree Boosting

Hybrid multi-GPU computing: accelerated kernels for segmentation and object detection with medical image processing applications

AUTOMATED EXTRACTION OF LIVER OUTLINES FROM COMPUTED TOMOGRAPHY SCAN IMAGES USING A CUDA-BASED SEGMENTATION METHOD

Efficient Gradient Boosted Decision Tree Training on GPUs

Exploiting GPUs for Efficient Gradient Boosting Decision Tree Training

FAST RCS PREDICTION USING MULTIRESOLUTION SHOOTING AND BOUNCING RAY METHOD ON THE GPU

Multi-class Image Classification Based on Fast Stochastic Gradient Boosting.

A GPU-based residual network for medical image classification in smart medicine

FastBDT: A speed-optimized and cache-friendly implementation of stochastic gradient-boosted decision trees for multivariate classification

A Boosting Regression Approach to Medical Anatomy Detection

To Boost or Not to Boost? On the Limits of Boosted Trees for Object Detection

A GPU-Based Training of BP Neural Network for Healthcare Data Analysis

Parallel Computation for Blood Cell Classification in Medical Hyperspectral Imagery

Faster Mean-shift: GPU-accelerated clustering for cosine embedding-based cell segmentation and tracking

One-stage Multi-task Detector for 3D Cardiac MR Imaging

Mri Image Segmentation Based on A Gpu Shortest Path Algorithm

GPU Implementation of an Automatic Target Detection and Classification Algorithm for Hyperspectral Image Analysis

GPU acceleration of a patient-specific airway image segmentation and its assessment

GPU-based Fast Scale Invariant Interest Point Detector.