English

LoBoost: Fast Model-Native Local Conformal Prediction for Gradient-Boosted Trees

Machine Learning 2026-02-27 v1 Machine Learning

Abstract

Gradient-boosted decision trees are among the strongest off-the-shelf predictors for tabular regression, but point predictions alone do not quantify uncertainty. Conformal prediction provides distribution-free marginal coverage, yet split conformal uses a single global residual quantile and can be poorly adaptive under heteroscedasticity. Methods that improve adaptivity typically fit auxiliary nuisance models or introduce additional data splits/partitions to learn the conformal score, increasing cost and reducing data efficiency. We propose LoBoost, a model-native local conformal method that reuses the fitted ensemble's leaf structure to define multiscale calibration groups. Each input is encoded by its sequence of visited leaves; at resolution level k, we group points by matching prefixes of leaf indices across the first k trees and calibrate residual quantiles within each group. LoBoost requires no retraining, auxiliary models, or extra splitting beyond the standard train/calibration split. Experiments show competitive interval quality, improved test MSE on most datasets, and large calibration speedups.

Keywords

Cite

@article{arxiv.2602.22432,
  title  = {LoBoost: Fast Model-Native Local Conformal Prediction for Gradient-Boosted Trees},
  author = {Vagner Santos and Victor Coscrato and Luben Cabezas and Rafael Izbicki and Thiago Ramos},
  journal= {arXiv preprint arXiv:2602.22432},
  year   = {2026}
}
R2 v1 2026-07-01T10:53:00.991Z