Random Jump Forests

Martin Bladt; Rasmus Frigaard Lemvig

Home/Archive/arXiv

arXiv · 2610.32498

Random Jump Forests

Martin Bladt·Rasmus Frigaard Lemvig

View original source ↗Read source document ↗Export metadata ↓

Abstract

Random jump forests are introduced as a nonparametric tool for estimating conditional cumulative transition rates and occupation probabilities in multi-state models. The main idea is that a single tree partitions the covariate space for all transitions simultaneously, using partially observed trajectories, and with Nelson--Aalen estimators at the terminal nodes. Occupation probabilities are recovered through the product integral without requiring the Markov assumption. Uniform consistency and a functional central limit theorem for the estimators of cumulative rates can be established under the assumption that the covariate vector takes finitely many values, and under suitable splitting conditions. In particular, pointwise confidence intervals and simultaneous bands are available in the Markov case. Simulation studies show a predictive accuracy close to the correctly specified benchmark models and marked improvements when benchmarks are misspecified, with further illustration on four biomedical data sets. In the coverage study, confidence bands attain near-nominal coverage in several settings, with pooling of heterogeneous rates being the main source of possible undercoverage. The method, including prediction assessment and variable importance, is implemented in the R package JumpForests.

Explore related subjects

stat.ME cs.LG stat.ML

Keep this discovery

Explore connections, maps & timelines

BibTeX RIS

Martin Bladt, Rasmus Frigaard Lemvig. 2026-10-23. Random Jump Forests. https://arxiv.org/abs/2610.32498

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

Selection from Hierarchical Data with Conformal e-values

Distribution-free predictive inference beyond the construction of prediction sets has gained a lot of interest in recent applications. One such application is the selection task, where the objective is to design a reliable selection rule to pick out individuals with desired unobserved outcomes while controlling the error rate. In this work, we address the selection problem in the context of hierarchical data, where groups of observations may exhibit distinct within-group distributions. This generalizes existing techniques beyond the standard i.i.d./exchangeable data settings. As a correction, For hierarchical data, we introduce methods to construct valid conformal e-values, enabling control of the false discovery rate (FDR) through the e-BH procedure. In particular, we introduce and compare two approaches -- subsampling conformal e-values and hierarchical conformal e-values. Empirical results demonstrate that both approaches achieve valid FDR control while highlighting a tradeoff between stability and power. The subsampling-based method, though random, typically offers higher power, whereas the hierarchical approach, being deterministic, tends to be slightly less powerful. The effectiveness of the proposed methods is illustrated in two real-world applications.

stat.ME↗

arXiv2026

Conformalized Polynomial Chaos Expansion for Uncertainty-aware Surrogate Modeling

This work introduces a method to equip data-driven polynomial chaos expansion surrogate models with intervals that quantify the predictive uncertainty of the surrogate. To that end, jackknife-based conformal prediction is integrated into regression-based polynomial chaos expansions. The jackknife algorithm uses leave-one-out residuals to generate predictive intervals around the predictions of the polynomial chaos surrogate. The jackknife+ extension additionally requires leave-one-out model predictions. Both methods allow to use the entire dataset for model training and do not require a hold-out dataset for prediction interval calibration. The key to efficient implementation is to leverage the linearity of the polynomial chaos regression model, so that leave-one-out residuals and, if necessary, leave-one-out model predictions can be computed with analytical, closed-form expressions. This eliminates the need for repeated model re-training. The conformalized polynomial chaos expansion method is first validated on four benchmark models and then applied to two electrical engineering design use-cases. The method produces predictive intervals that provide the target coverages, even for low-accuracy models trained with small datasets. At the same time, training data availability plays a crucial role in improving the empirical coverage and narrowing the predictive interval, as well as in reducing their variability over different training datasets.

stat.ME↗

arXiv2026

High-dimensional reliability-oriented Shapley effects estimation

This article presents a new estimation scheme for the reliability-oriented Shapley effects when there is a large number of correlated input variables, using a unique sample of failure points. First, we propose a new reformulation of the reliability-oriented closed Sobol indices. This reformulation circumvents the estimation issues associated with existing methods in the literature, but requires estimating densities in potentially high dimensions. Once the reliability-oriented closed Sobol indices have been estimated through these densities, the reliability-oriented Shapley effects are estimated through permutation aggregation to circumvent the issue related to the dimension. In addition, we provide an error estimation procedure relying on the same sample of failing points resulting in a probabilistic error bound, and avoiding any additional call to the model. We illustrate our methodology on numerical use-cases where we compare several density estimators and we propose an application to the monitoring of adaptive importance sampling.

stat.ME↗