Calibrating probabilistic predictions of quantile regression forests with conformal predictive systems

Di Wang,Ping Wang,Cong Wang,Pingping Wang
DOI: https://doi.org/10.1016/j.patrec.2022.02.003
IF: 4.757
2022-04-01
Pattern Recognition Letters
Abstract:Quantile regression forests (QRF) is a generalization of random forests for quantile regression, which can also output probabilistic prediction for regression problems. QRF delivers a nonlinear and nonparametric way of probabilistic prediction and is widely used in many applications. To improve the prediction quality of QRF, this paper builds conformal predictive systems (CPSs) on top of QRF to calibrate the probabilistic prediction, which is the first attempt of combining CPSs with QRF to our best knowledge. Conformal predictive systems aim to produce valid probabilistic predictions in machine learning community, whose predictive distributions are compatible with realizations. Three algorithms are proposed in this paper. One is a fast approximation algorithm to split conformal predictive system. Building on that, the other two algorithms are proposed based on the out-of-bag samples and the weighted approach of QRF. Experiments with 20 public data sets were conducted and the results demonstrated that our algorithms are empirically valid and compared favourably with the original QRF.
computer science, artificial intelligence
What problem does this paper attempt to address?