Out-of-bag prediction balls for random forests in metric spaces

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Serrano, Diego, García-Portugués, Eduardo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918426924023808
author Serrano, Diego
García-Portugués, Eduardo
author_facet Serrano, Diego
García-Portugués, Eduardo
contents Statistical methods for metric spaces provide a general and versatile framework for analyzing complex data types. We introduce a novel approach for constructing confidence regions around new predictions from any bagged regression algorithm with metric-space-valued responses. This includes the recent extensions of random forests for metric responses: Fréchet random forests (Capitaine et al., 2024), random forest weighted local constant Fréchet regression (Qiu et al., 2024), and metric random forests (Bulté and Sørensen, 2024). Our prediction regions leverage out-of-bag observations generated during a single forest training, employing the entire data set for both prediction and uncertainty quantification. We establish asymptotic guarantees of out-of-bag prediction balls for four coverage types under certain regularity conditions. Moreover, we demonstrate the superior stability and smaller radius of out-of-bag balls compared to split-conformal methods through extensive numerical experiments where the response lies on the Euclidean space, sphere, hyperboloid, and space of positive definite matrices. A real data application illustrates the potential of the confidence regions for quantifying the uncertainty in the study of solar dynamics and the use of data-driven non-isotropic distances on the sphere.
format Preprint
id arxiv_https___arxiv_org_abs_2510_04299
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Out-of-bag prediction balls for random forests in metric spaces
Serrano, Diego
García-Portugués, Eduardo
Methodology
62R20, 62G08, 62G15
Statistical methods for metric spaces provide a general and versatile framework for analyzing complex data types. We introduce a novel approach for constructing confidence regions around new predictions from any bagged regression algorithm with metric-space-valued responses. This includes the recent extensions of random forests for metric responses: Fréchet random forests (Capitaine et al., 2024), random forest weighted local constant Fréchet regression (Qiu et al., 2024), and metric random forests (Bulté and Sørensen, 2024). Our prediction regions leverage out-of-bag observations generated during a single forest training, employing the entire data set for both prediction and uncertainty quantification. We establish asymptotic guarantees of out-of-bag prediction balls for four coverage types under certain regularity conditions. Moreover, we demonstrate the superior stability and smaller radius of out-of-bag balls compared to split-conformal methods through extensive numerical experiments where the response lies on the Euclidean space, sphere, hyperboloid, and space of positive definite matrices. A real data application illustrates the potential of the confidence regions for quantifying the uncertainty in the study of solar dynamics and the use of data-driven non-isotropic distances on the sphere.
title Out-of-bag prediction balls for random forests in metric spaces
topic Methodology
62R20, 62G08, 62G15
url https://arxiv.org/abs/2510.04299