IEEE - Institute of Electrical and Electronics Engineers, Inc. - Exploiting GPUs for Efficient Gradient Boosting Decision Tree Training

Author(s): Zeyi Wen ; Jiashuai Shi ; Bingsheng He ; Jian Chen ; Kotagiri Ramamohanarao ; Qinbin Li
Sponsor(s): IEEE Computer Society
Publisher: IEEE - Institute of Electrical and Electronics Engineers, Inc.
Publication Date: 1 December 2019
Volume: 30
Page(s): 2,706 - 2,717
ISSN (CD): 2161-9883
ISSN (Electronic): 1558-2183
ISSN (Paper): 1045-9219
DOI: 10.1109/TPDS.2019.2920131
Regular:

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... View More

Advertisement