Beyond Point Prediction: Artificial Representative Trees with Uncertainty
Lea L. Mairhöfer, Silke Szymczak, Björn-Hergen Laabs, Tuwe Löfström-Cavallin
Abstract
Random forests (RFs) predict well but are opaque, whereas single decision trees are interpretable but unstable. Artificial representative trees (ARTs) were developed as interpretable surrogate models for RFs, but their use as standalone prediction models with uncertainty quantification has not been systematically investigated. We combine ARTs with leaf-wise Mondrian conformal predictive systems (CPS), enabling a single tree to provide continuous predictions, prediction intervals, and probabilities of exceeding arbitrary thresholds. We compared ARTs with CPS against decision trees with CPS and separate regression and probability trees across five simulation scenarios, 21 benchmark datasets, and a cross-sectional NHANES example data set. Repeated cross-validation assessed predictive performance, interpretability, and stability. ARTs with CPS yield compact, structurally stable trees with substantially more reproducible split-variable selection than decision trees across benchmark datasets and NHANES. Decision trees showed slightly better predictive performance and narrower prediction intervals, while coverage was broadly comparable. CPS-based trees generally achieved lower and less variable Brier scores than multi-model approaches. Combining ARTs with CPS therefore provides a single, interpretable, and stable model for continuous predictions and calibrated probabilities, balancing predictive performance with reproducibility and transparency in settings where stability and interpretability are essential.
Create a lesson
Related papers
Empirical Auditing of Edge-Private Graph Generators
Anum Fatima, Stratis Limnios, James Adams et al.
Variational objectives for amortized Bayesian inference in inverse problems: The role of posterior conditioning
Abhishek Srivastava, Arijit Hazra, Rajesh Dubbaku
JAREX: An Acquisition Function for Multi-Objective Algorithmic Process Characterization
Xinyang Li, Kevin Stone, Ajit Vikram
Identifying Representational Biases in Datasets Using PCA: A Max-Disparity Partition Framework
Arjun KM, Shashi Jain
Adversarially Robust PAC Learning with Optimal VC Rates
Steve Hanneke, Amirreza Shaeiri
OSCAR: Order-aware Scoring and Calibration for AI Rankings
You Liu, Yue Liu, Quanchao Lu et al.