Condensed Gradient Boosting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Emami, Seyedsaman, Martínez-Muñoz, Gonzalo
Format: Preprint
Published: 2022
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911965240098816
author Emami, Seyedsaman
Martínez-Muñoz, Gonzalo
author_facet Emami, Seyedsaman
Martínez-Muñoz, Gonzalo
contents This paper presents a computationally efficient variant of gradient boosting for multi-class classification and multi-output regression tasks. Standard gradient boosting uses a 1-vs-all strategy for classifications tasks with more than two classes. This strategy translates in that one tree per class and iteration has to be trained. In this work, we propose the use of multi-output regressors as base models to handle the multi-class problem as a single task. In addition, the proposed modification allows the model to learn multi-output regression problems. An extensive comparison with other multi-ouptut based gradient boosting methods is carried out in terms of generalization and computational efficiency. The proposed method showed the best trade-off between generalization ability and training and predictions speeds.
format Preprint
id arxiv_https___arxiv_org_abs_2211_14599
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Condensed Gradient Boosting
Emami, Seyedsaman
Martínez-Muñoz, Gonzalo
Machine Learning
This paper presents a computationally efficient variant of gradient boosting for multi-class classification and multi-output regression tasks. Standard gradient boosting uses a 1-vs-all strategy for classifications tasks with more than two classes. This strategy translates in that one tree per class and iteration has to be trained. In this work, we propose the use of multi-output regressors as base models to handle the multi-class problem as a single task. In addition, the proposed modification allows the model to learn multi-output regression problems. An extensive comparison with other multi-ouptut based gradient boosting methods is carried out in terms of generalization and computational efficiency. The proposed method showed the best trade-off between generalization ability and training and predictions speeds.
title Condensed Gradient Boosting
topic Machine Learning
url https://arxiv.org/abs/2211.14599