Saved in:
Bibliographic Details
Main Authors: Sluijterman, Laurens, Kreuwel, Frank, Cator, Eric, Heskes, Tom
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.02293
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909216680181760
author Sluijterman, Laurens
Kreuwel, Frank
Cator, Eric
Heskes, Tom
author_facet Sluijterman, Laurens
Kreuwel, Frank
Cator, Eric
Heskes, Tom
contents This paper explores the use of XGBoost for composite quantile regression. XGBoost is a highly popular model renowned for its flexibility, efficiency, and capability to deal with missing data. The optimization uses a second order approximation of the loss function, complicating the use of loss functions with a zero or vanishing second derivative. Quantile regression -- a popular approach to obtain conditional quantiles when point estimates alone are insufficient -- unfortunately uses such a loss function, the pinball loss. Existing workarounds are typically inefficient and can result in severe quantile crossings. In this paper, we present a smooth approximation of the pinball loss, the arctan pinball loss, that is tailored to the needs of XGBoost. Specifically, contrary to other smooth approximations, the arctan pinball loss has a relatively large second derivative, which makes it more suitable to use in the second order approximation. Using this loss function enables the simultaneous prediction of multiple quantiles, which is more efficient and results in far fewer quantile crossings.
format Preprint
id arxiv_https___arxiv_org_abs_2406_02293
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Composite Quantile Regression With XGBoost Using the Novel Arctan Pinball Loss
Sluijterman, Laurens
Kreuwel, Frank
Cator, Eric
Heskes, Tom
Machine Learning
This paper explores the use of XGBoost for composite quantile regression. XGBoost is a highly popular model renowned for its flexibility, efficiency, and capability to deal with missing data. The optimization uses a second order approximation of the loss function, complicating the use of loss functions with a zero or vanishing second derivative. Quantile regression -- a popular approach to obtain conditional quantiles when point estimates alone are insufficient -- unfortunately uses such a loss function, the pinball loss. Existing workarounds are typically inefficient and can result in severe quantile crossings. In this paper, we present a smooth approximation of the pinball loss, the arctan pinball loss, that is tailored to the needs of XGBoost. Specifically, contrary to other smooth approximations, the arctan pinball loss has a relatively large second derivative, which makes it more suitable to use in the second order approximation. Using this loss function enables the simultaneous prediction of multiple quantiles, which is more efficient and results in far fewer quantile crossings.
title Composite Quantile Regression With XGBoost Using the Novel Arctan Pinball Loss
topic Machine Learning
url https://arxiv.org/abs/2406.02293