Distribution-Free Predictive Inference for Regression

Jing LeiMax G'SellAlessandro RinaldoRyan J. TibshiraniLarry Wasserman

article2016Journal of the American Statistical Association1,403 citations

Develops a conformal prediction framework for regression that guarantees finite-sample validity for any base estimator without distributional assumptions, introducing computationally efficient split-conformal algorithms and model-free variable importance metrics.

Listen

Modern decision-making increasingly relies on complex, high-dimensional machine learning and regression algorithms to forecast outcomes. However, standard statistical techniques for quantifying prediction uncertainty typically depend on rigid assumptions—such as correct model specification, normal error distributions, or constant variance—that rarely hold in real-world environments. When these assumptions fail, conventional prediction intervals either under-cover true outcomes or become excessively wide, introducing unquantified risk into critical operations. To resolve this problem, the article develops a general, distribution-free predictive inference framework based on conformal inference that constructs mathematically guaranteed prediction intervals around any regression algorithm without requiring restrictive distributional or model-correctness assumptions.

The article evaluates theoretical guarantees, computational trade-offs, and empirical performance across two primary conformal approaches—full conformal inference and split conformal inference—alongside a related leave-one-out jackknife technique. The analysis establishes finite-sample coverage bounds and proves that conformal prediction bands achieve near-optimal width when base models are stable and consistent. To demonstrate practical utility, the authors conduct extensive simulations spanning low-dimensional and high-dimensional settings (up to 2,000 features), including challenging scenarios with heavy-tailed errors, strong feature correlations, and heteroskedasticity. They benchmark linear regression, penalized models (lasso, elastic net, ridge), sparse additive models, and random forests, while providing an open-source R implementation package.

The findings provide strong support for conformal methodology. First, conformal prediction bands strictly achieve the nominal average coverage (for example, approximately 90% across all evaluated configurations) even when models are severely misspecified, nonlinear, or heavy-tailed, whereas standard parametric methods fail or produce overly wide intervals. Second, split conformal inference delivers virtually identical coverage to full conformal inference while drastically reducing computation time—executing in fractions of a second compared to hundreds of seconds in high-dimensional tests. Third, the width of the prediction intervals is directly governed by model accuracy: more predictive estimators yield narrower, more informative intervals. Fourth, combining multiple data splits via standard multi-split aggregation widens prediction bands due to conservative correction penalties, indicating that a single balanced split is preferable. Finally, the authors show that modifying residuals with local error-spread estimates allows the intervals to adapt to varying noise levels across the feature space, and introduce a model-free variable importance measure (leave-one-covariate-out) that reliably identifies predictive features without parametric assumptions.

These results demonstrate that organizations can reliably quantify predictive uncertainty across arbitrary black-box models without risking failure from misspecified assumptions. For operational deployment, decision-makers should adopt split conformal inference as the default standard for generating prediction intervals and evaluating model-free feature importance due to its extreme computational efficiency and robust finite-sample guarantees. Where heteroskedasticity is present, locally weighted conformal inference should be used to ensure uniform local coverage. Future work should focus on developing more efficient multi-split aggregation methods to eliminate data-splitting randomness without inflating interval length, as well as refining variable selection frameworks to address post-selection inference under data splitting.

arXiv: 1604.04173
  • Paper: A tutorial on conformal prediction, Glenn Shafer et al. (2007). This tutorial introduces the foundational framework and mathematical principles of conformal prediction under exchangeability that the source directly adapts and extends to regression.
  • Paper: Quantile Regression Forests, Nicolai Meinshausen (2006). This paper establishes non-parametric quantile estimation using random forests, providing key background for the source's distribution-free regression and heteroskedasticity-adapting prediction bands.
  • Paper: Stability and Generalization, Olivier Bousquet et al. (2002). This work establishes algorithmic stability tools and leave-one-out sensitivity analyses that underpin the theoretical justifications for sample-splitting and jackknife-based predictive inference.
  • Paper: A survey of cross-validation procedures for model selection, Sylvain Arlot et al. (2009). This survey provides essential theoretical foundations on cross-validation and sample-splitting mechanics that inform the source's development of split conformal and out-of-sample inference.
Cover for Distribution-Free Predictive Inference for Regression

Abstract

We develop a general framework for distribution-free predictive inference in regression, using conformal inference. The proposed methodology allows for the construction of a prediction band for the response variable using any estimator of the regression function. The resulting prediction band preserves the consistency properties of the original estimator under standard assumptions, while guaranteeing finite-sample marginal coverage even when these assumptions do not hold. We analyze and compare, both empirically and theoretically, the two major variants of our conformal framework: full conformal inference and split conformal inference, along with a related jackknife method. These methods offer different tradeoffs between statistical accuracy (length of resulting prediction intervals) and computational efficiency. As extensions, we develop a method for constructing valid in-sample prediction intervals called {\it rank-one-out} conformal inference, which has essentially the same computational efficiency as split conformal inference. We also describe an extension of our procedures for producing prediction bands with locally varying length, in order to adapt to heteroskedascity in the data. Finally, we propose a model-free notion of variable importance, called {\it leave-one-covariate-out} or LOCO inference. Accompanying this paper is an R package {\tt conformalInference} that implements all of the proposals we have introduced. In the spirit of reproducibility, all of our empirical results can also be easily (re)generated using this package.

Table of Contents

  • 1 Introduction
  • 1.1 Related Work
  • 1.2 Summary and Outline
  • 2 Conformal Inference
  • 2.1 Conformal Prediction Sets
  • 2.2 Split Conformal Prediction Sets
  • 2.3 Multiple Splits
  • 2.4 Jackknife Prediction Intervals
  • 3 Statistical Accuracy
  • 3.1 Comparing the Oracles
  • 3.2 Oracle Approximation Under Stability Assumptions
  • 3.3 Super Oracle Approximation Under Consistency Assumptions
  • 3.4 A High-dimensional Sparse Regression Example
  • 4 Empirical Study
  • 4.1 Comparisons to Parametric Intervals from Linear Regression
  • 4.2 Comparisons of Conformal Intervals Across Base Estimators
  • 5 Extensions of Conformal Inference
  • 5.1 In-Sample Split Conformal Inference
  • 5.2 Locally-Weighted Conformal Inference
  • 6 Model-Free Variable Importance: LOCO
  • 6.1 Local Measure of Variable Importance
  • 6.2 Global Measures of Variable Importance
  • 7 Conclusion
  • References
  • A Technical Proofs
  • A.1 Proofs for
  • A.2 Proofs for
  • A.3 Proofs for
  • B Additional Experiments

Knowls

  1. Knowl 1 — Full Conformal Prediction Algorithm and Finite-Sample Coverage

    algorithm

    Full conformal prediction is a distribution-free method that constructs a prediction interval Cconf(Xn+1)C_{\text{conf}}(X_{n+1}) for a new response Yn+1∈RY_{n+1} \in \mathbb{R} given features Xn+1∈RdX_{n+1} \in \mathbb{R}^d, using any symmetric regression algorithm A\mathcal{A} on i.i.d. training data (X1,Y1),…,(Xn,Yn)∼P(X_1, Y_1), \dots, (X_n, Y_n) \sim P.

    Input: Training data (Xi,Yi)(X_i, Y_i) for i=1,…,ni = 1, \dots, n, miscoverage level α∈(0,1)\alpha \in (0, 1), symmetric regression algorithm A\mathcal{A}, query points Xnew\mathcal{X}_{\text{new}}, candidate response values Ytrial={y1,y2,… }\mathcal{Y}_{\text{trial}} = \{y_1, y_2, \dots\}
    Output: Prediction intervals Cconf(x)C_{\text{conf}}(x) for each x∈Xnewx \in \mathcal{X}_{\text{new}}
    for x∈Xnewx \in \mathcal{X}_{\text{new}} do
        for y∈Ytrialy \in \mathcal{Y}_{\text{trial}} do
            Fit model on augmented data: μ^y=A({(X1,Y1),…,(Xn,Yn),(x,y)})\hat{\mu}_y = \mathcal{A}(\{(X_1, Y_1), \dots, (X_n, Y_n), (x, y)\})
            Compute fitted residuals: Ry,i=∣Yi−μ^y(Xi)∣R_{y,i} = |Y_i - \hat{\mu}_y(X_i)| for i=1,…,ni = 1, \dots, n, and Ry,n+1=∣y−μ^y(x)∣R_{y,n+1} = |y - \hat{\mu}_y(x)|
            Compute rank statistic: π(y)=1n+1(1+∑i=1n1{Ry,i≤Ry,n+1})\pi(y) = \frac{1}{n+1} \left(1 + \sum_{i=1}^n \mathbf{1}\{R_{y,i} \le R_{y,n+1}\}\right)
        end for
        Set Cconf(x)={y∈Ytrial:(n+1)π(y)≤⌈(1−α)(n+1)⌉}C_{\text{conf}}(x) = \{y \in \mathcal{Y}_{\text{trial}} : (n + 1)\pi(y) \le \lceil(1 - \alpha)(n + 1)\rceil\}
    end for
    return Cconf(x)C_{\text{conf}}(x) for each x∈Xnewx \in \mathcal{X}_{\text{new}}

    If (X1,Y1),…,(Xn,Yn),(Xn+1,Yn+1)(X_1, Y_1), \dots, (X_n, Y_n), (X_{n+1}, Y_{n+1}) are i.i.d. draws from PP, the full conformal prediction set satisfies the nonasymptotic marginal validity guarantee:

    P(Yn+1∈Cconf(Xn+1))≥1−α\mathbb{P}(Y_{n+1} \in C_{\text{conf}}(X_{n+1})) \ge 1 - \alpha

    Furthermore, if the absolute residuals Ry,i=∣Yi−μ^y(Xi)∣R_{y,i} = |Y_i - \hat{\mu}_y(X_i)| have a continuous joint distribution for all y∈Ry \in \mathbb{R}, the procedure is non-conservative with upper bound:

    P(Yn+1∈Cconf(Xn+1))≤1−α+1n+1\mathbb{P}(Y_{n+1} \in C_{\text{conf}}(X_{n+1})) \le 1 - \alpha + \frac{1}{n + 1}

  2. Knowl 2 — Split Conformal Prediction Algorithm and Finite-Sample Bounds

    algorithm

    Split conformal prediction separates model training from residual ranking via sample splitting, computing distribution-free prediction intervals with the computational cost of a single model fit.

    Input: Data (Xi,Yi)(X_i, Y_i) for i=1,…,ni = 1, \dots, n (with nn even), nominal miscoverage level α∈(0,1)\alpha \in (0, 1), regression algorithm A\mathcal{A}
    Output: Prediction band Csplit(x)C_{\text{split}}(x) for all x∈Rdx \in \mathbb{R}^d
    Randomly split {1,…,n}\{1, \dots, n\} into two equal-sized subsets I1I_1 and I2I_2 of size n/2n/2
    Fit model on training subset: μ^=A({(Xi,Yi):i∈I1})\hat{\mu} = \mathcal{A}(\{(X_i, Y_i) : i \in I_1\})
    Compute calibration residuals: Ri=∣Yi−μ^(Xi)∣R_i = |Y_i - \hat{\mu}(X_i)| for each i∈I2i \in I_2
    Compute quantile cutoff: d=k-th smallest value in {Ri:i∈I2} with k=⌈(n/2+1)(1−α)⌉d = k\text{-th smallest value in } \{R_i : i \in I_2\} \text{ with } k = \lceil(n/2 + 1)(1 - \alpha)\rceil
    return Csplit(x)=[μ^(x)−d,μ^(x)+d]C_{\text{split}}(x) = [\hat{\mu}(x) - d, \hat{\mu}(x) + d] for all x∈Rdx \in \mathbb{R}^d

    For i.i.d. sample pairs (X1,Y1),…,(Xn,Yn),(Xn+1,Yn+1)∼P(X_1, Y_1), \dots, (X_n, Y_n), (X_{n+1}, Y_{n+1}) \sim P, the split conformal band satisfies:

    P(Yn+1∈Csplit(Xn+1))≥1−α\mathbb{P}(Y_{n+1} \in C_{\text{split}}(X_{n+1})) \ge 1 - \alpha

    If the calibration residuals RiR_i for i∈I2i \in I_2 have a continuous joint distribution, the coverage is bounded above by:

    P(Yn+1∈Csplit(Xn+1))≤1−α+2n+2\mathbb{P}(Y_{n+1} \in C_{\text{split}}(X_{n+1})) \le 1 - \alpha + \frac{2}{n + 2}

    Additionally, split conformal prediction provides approximate in-sample coverage on the calibration split I2I_2: there exists an absolute constant c>0c > 0 such that for any ϵ>0\epsilon > 0,

    P(∣2n∑i∈I21{Yi∈Csplit(Xi)}−(1−α)∣≥ϵ)≤2exp⁡(−cn2(ϵ−4/n)+2)\mathbb{P}\left(\left| \frac{2}{n} \sum_{i \in I_2} \mathbf{1}\{Y_i \in C_{\text{split}}(X_i)\} - (1 - \alpha) \right| \ge \epsilon\right) \le 2 \exp\left(-c n^2 (\epsilon - 4/n)_+^2\right)

  3. Knowl 3 — Rank-One-Out Split Conformal Inference for In-Sample Prediction Intervals

    algorithm

    Rank-one-out (ROO) split conformal inference constructs distribution-free, finite-sample valid prediction intervals Croo(Xi)C_{\text{roo}}(X_i) at all observed data points X1,…,XnX_1, \dots, X_n using two model fits and leave-one-out residual ranking across splits.

    Input: Data (Xi,Yi)(X_i, Y_i) for i=1,…,ni = 1, \dots, n (with nn even), nominal miscoverage level α∈(0,1)\alpha \in (0, 1), regression algorithm A\mathcal{A}
    Output: Prediction intervals Croo(Xi)C_{\text{roo}}(X_i) for each i=1,…,ni = 1, \dots, n
    Randomly split {1,…,n}\{1, \dots, n\} into two equal-sized subsets I1,I2I_1, I_2
    for k∈{1,2}k \in \{1, 2\} do
        Fit regression estimator on split IkI_k: μ^k=A({(Xi,Yi):i∈Ik})\hat{\mu}_k = \mathcal{A}(\{(X_i, Y_i) : i \in I_k\})
        for i∉Iki \notin I_k do
            Compute residual Ri=∣Yi−μ^k(Xi)∣R_i = |Y_i - \hat{\mu}_k(X_i)|
        end for
        for i∉Iki \notin I_k do
            Compute di=m-th smallest value in {Rj:j∉Ik,j≠i} where m=⌈(n/2)(1−α)⌉d_i = m\text{-th smallest value in } \{R_j : j \notin I_k, j \ne i\} \text{ where } m = \lceil(n/2)(1 - \alpha)\rceil
            Set Croo(Xi)=[μ^k(Xi)−di,μ^k(Xi)+di]C_{\text{roo}}(X_i) = [\hat{\mu}_k(X_i) - d_i, \hat{\mu}_k(X_i) + d_i]
        end for
    end for
    return Croo(Xi)C_{\text{roo}}(X_i) for each i=1,…,ni = 1, \dots, n

    By symmetry, for each individual point i∈{1,…,n}i \in \{1, \dots, n\}, the in-sample coverage satisfies P(Yi∈Croo(Xi))≥1−α\mathbb{P}(Y_i \in C_{\text{roo}}(X_i)) \ge 1 - \alpha. For the empirical average in-sample coverage over the entire sample, McDiarmid's inequality yields absolute constant c>0c > 0 such that for any ϵ>0\epsilon > 0:

    P(1n∑i=1n1{Yi∈Croo(Xi)}≥1−α−ϵ)≥1−2exp⁡(−cnϵ2)\mathbb{P}\left(\frac{1}{n} \sum_{i=1}^n \mathbf{1}\{Y_i \in C_{\text{roo}}(X_i)\} \ge 1 - \alpha - \epsilon\right) \ge 1 - 2\exp(-c n \epsilon^2)

    If the residuals have a continuous joint distribution, the empirical coverage is two-sided bounded:

    P(1−α−ϵ≤1n∑i=1n1{Yi∈Croo(Xi)}≤1−α+2n+ϵ)≥1−2exp⁡(−cnϵ2)\mathbb{P}\left(1 - \alpha - \epsilon \le \frac{1}{n} \sum_{i=1}^n \mathbf{1}\{Y_i \in C_{\text{roo}}(X_i)\} \le 1 - \alpha + \frac{2}{n} + \epsilon\right) \ge 1 - 2\exp(-c n \epsilon^2)

  4. Knowl 4 — Locally-Weighted Conformal Inference for Heteroskedastic Errors

    model/method

    Standard conformal prediction intervals have roughly constant length over the feature space Rd\mathbb{R}^d. When error variability depends on features, locally-weighted conformal inference scales the fitted residuals inversely by an estimate of the conditional error spread, producing prediction intervals with locally varying width while preserving exact distribution-free finite-sample marginal coverage.

    Let μ^(x)\hat{\mu}(x) estimate the conditional mean E[Y∣X=x]\mathbb{E}[Y \mid X = x] and let ρ^(x)\hat{\rho}(x) estimate the conditional mean absolute deviation (MAD) E[∣Y−μ(X)∣∣X=x]\mathbb{E}[|Y - \mu(X)| \mid X = x].

    In split conformal prediction, the estimators μ^\hat{\mu} and ρ^\hat{\rho} are fitted on training split I1I_1. The locally-weighted residuals on calibration split I2I_2 are defined as:

    Ri=∣Yi−μ^(Xi)∣ρ^(Xi),i∈I2R_i = \frac{|Y_i - \hat{\mu}(X_i)|}{\hat{\rho}(X_i)}, \quad i \in I_2

    Let dd be the ⌈(n/2+1)(1−α)⌉\lceil(n/2 + 1)(1 - \alpha)\rceil-th smallest value in {Ri:i∈I2}\{R_i : i \in I_2\}. The locally-weighted split conformal prediction interval at any x∈Rdx \in \mathbb{R}^d is:

    Csplit, lw(x)=[μ^(x)−ρ^(x)d, μ^(x)+ρ^(x)d]C_{\text{split, lw}}(x) = [\hat{\mu}(x) - \hat{\rho}(x)d, \, \hat{\mu}(x) + \hat{\rho}(x)d]

    In full conformal prediction with trial point (x,y)(x, y), augmented estimators μ^y\hat{\mu}_y and ρ^y\hat{\rho}_y are fitted on {(X1,Y1),…,(Xn,Yn),(x,y)}\{(X_1, Y_1), \dots, (X_n, Y_n), (x, y)\}, yielding locally-weighted conformity scores:

    Ry,i=∣Yi−μ^y(Xi)∣ρ^y(Xi)(i=1,…,n),Ry,n+1=∣y−μ^y(x)∣ρ^y(x)R_{y,i} = \frac{|Y_i - \hat{\mu}_y(X_i)|}{\hat{\rho}_y(X_i)} \quad (i=1,\dots,n), \qquad R_{y,n+1} = \frac{|y - \hat{\mu}_y(x)|}{\hat{\rho}_y(x)}

  5. Knowl 5 — Leave-One-Covariate-Out Local Variable Importance Prediction Sets

    model/method

    Leave-one-covariate-out (LOCO) local variable importance provides a model-free, predictive measure of the importance of individual covariates without requiring model correctness or distributional assumptions.

    Let μ^\hat{\mu} be a regression estimator fitted on data using all dd covariates, and let μ^(−j)\hat{\mu}_{(-j)} be the estimator fitted on the same data with the jj-th covariate excluded, for j∈{1,…,d}j \in \{1, \dots, d\}. The excess prediction error of covariate jj at a new draw (Xn+1,Yn+1)(X_{n+1}, Y_{n+1}) is defined as:

    Δj(Xn+1,Yn+1)=∣Yn+1−μ^(−j)(Xn+1)∣−∣Yn+1−μ^(Xn+1)∣\Delta_j(X_{n+1}, Y_{n+1}) = |Y_{n+1} - \hat{\mu}_{(-j)}(X_{n+1})| - |Y_{n+1} - \hat{\mu}(X_{n+1})|

    Given a valid 1−α1 - \alpha conformal prediction band C(x)C(x) for Yn+1Y_{n+1} given Xn+1=xX_{n+1} = x, the LOCO prediction interval for Δj(Xn+1,Yn+1)\Delta_j(X_{n+1}, Y_{n+1}) is constructed by propagating C(x)C(x):

    Wj(x)={∣y−μ^(−j)(x)∣−∣y−μ^(x)∣:y∈C(x)}W_j(x) = \left\{ |y - \hat{\mu}_{(-j)}(x)| - |y - \hat{\mu}(x)| : y \in C(x) \right\}

    Because inclusion of Yn+1∈C(Xn+1)Y_{n+1} \in C(X_{n+1}) implies Δj(Xn+1,Yn+1)∈Wj(Xn+1)\Delta_j(X_{n+1}, Y_{n+1}) \in W_j(X_{n+1}) for all jj, the sets W1,…,WdW_1, \dots, W_d enjoy simultaneous finite-sample marginal validity across all features without multiplicity adjustments:

    P(Δj(Xn+1,Yn+1)∈Wj(Xn+1) for all j=1,…,d)≥1−α\mathbb{P}\left(\Delta_j(X_{n+1}, Y_{n+1}) \in W_j(X_{n+1}) \text{ for all } j = 1, \dots, d\right) \ge 1 - \alpha

  6. Knowl 6 — Leave-One-Covariate-Out Global Variable Importance Inference

    theoretical result

    Global LOCO inference quantifies the overall predictive importance of feature jj across the population distribution conditional on training data split D1={(Xi,Yi):i∈I1}D_1 = \{(X_i, Y_i) : i \in I_1\}. Let Δj(X,Y)=∣Y−μ^(−j)(X)∣−∣Y−μ^(X)∣\Delta_j(X, Y) = |Y - \hat{\mu}_{(-j)}(X)| - |Y - \hat{\mu}(X)|, and let Gj(t)=P(Δj(Xn+1,Yn+1)≤t∣D1)G_j(t) = \mathbb{P}(\Delta_j(X_{n+1}, Y_{n+1}) \le t \mid D_1) denote its conditional cumulative distribution function.

    1. Mean Excess Test Error θj=E[Δj(Xn+1,Yn+1)∣D1]\theta_j = \mathbb{E}[\Delta_j(X_{n+1}, Y_{n+1}) \mid D_1]: Using evaluation split D2={(Xi,Yi):i∈I2}D_2 = \{(X_i, Y_i) : i \in I_2\} of size n/2n/2, let θ^j=2n∑i∈I2Δj(Xi,Yi)\hat{\theta}_j = \frac{2}{n} \sum_{i \in I_2} \Delta_j(X_i, Y_i) and sj2s_j^2 be the sample variance of {Δj(Xi,Yi):i∈I2}\{\Delta_j(X_i, Y_i) : i \in I_2\}. An asymptotic 1−α1 - \alpha confidence interval is:

    [θ^j−zα/2sjn/2, θ^j+zα/2sjn/2]\left[ \hat{\theta}_j - \frac{z_{\alpha/2} s_j}{\sqrt{n/2}}, \, \hat{\theta}_j + \frac{z_{\alpha/2} s_j}{\sqrt{n/2}} \right]

    Testing H0:θj≤0H_0: \theta_j \le 0 vs H1:θj>0H_1: \theta_j > 0 rejects when n/2 (θ^j/sj)>zα\sqrt{n/2} \, (\hat{\theta}_j / s_j) > z_\alpha. The asymptotic convergence rate is uniform and free of feature dimension dd.

    1. Median Excess Test Error mj=median(Δj(Xn+1,Yn+1)∣D1)m_j = \text{median}(\Delta_j(X_{n+1}, Y_{n+1}) \mid D_1): Nonasymptotic hypothesis tests (H0:mj≤0H_0: m_j \le 0 vs H1:mj>0H_1: m_j > 0) and confidence intervals for mjm_j are constructed with exact finite-sample validity by applying standard nonparametric tests (the sign test requiring only continuity of GjG_j, or the Wilcoxon signed-rank test requiring continuity and symmetry) directly to the evaluation sample {Δj(Xi,Yi):i∈I2}\{\Delta_j(X_i, Y_i) : i \in I_2\}. Multiplicity across a tested feature set SS is controlled by replacing α\alpha with α/∣S∣\alpha / |S|.
  7. Knowl 7 — Oracle vs Super Oracle Prediction Bands and Accuracy Bound

    theoretical result

    Under i.i.d. regression data (X,Y)∼P(X, Y) \sim P where the noise ϵ=Y−μ(X)\epsilon = Y - \mu(X) is independent of XX and has symmetric density f0f_0 about 0 (Assumptions A0 and A1):

    • The Super Oracle band is Cs∗(x)=[μ(x)−qα,μ(x)+qα]C_s^*(x) = [\mu(x) - q_\alpha, \mu(x) + q_\alpha], where qαq_\alpha is the (1−α)(1 - \alpha)-quantile of ∣ϵ∣|\epsilon|. It achieves exact conditional coverage P(Y∈Cs∗(x)∣X=x)≥1−α\mathbb{P}(Y \in C_s^*(x) \mid X = x) \ge 1 - \alpha and has minimal expected length among all valid bands.
    • The Regular Oracle band for an estimator μ^n\hat{\mu}_n trained on nn samples is Co∗(x)=[μ^n(x)−qn,α,μ^n(x)+qn,α]C_o^*(x) = [\hat{\mu}_n(x) - q_{n,\alpha}, \hat{\mu}_n(x) + q_{n,\alpha}], where qn,αq_{n,\alpha} is the unconditional (1−α)(1 - \alpha)-quantile of ∣Y−μ^n(X)∣|Y - \hat{\mu}_n(X)|.

    Let Δn(x)=μ^n(x)−μ(x)\Delta_n(x) = \hat{\mu}_n(x) - \mu(x), and let FF and FnF_n denote the CDFs of ∣ϵ∣|\epsilon| and ∣Y−μ^n(X)∣|Y - \hat{\mu}_n(X)| respectively. If f0f_0 has a continuous derivative uniformly bounded by M>0M > 0, then:

    sup⁡t>0∣Fn(t)−F(t)∣≤M2E[Δn2(X)]\sup_{t > 0} |F_n(t) - F(t)| \le \frac{M}{2} \mathbb{E}[\Delta_n^2(X)]

    Furthermore, if the density ff of ∣ϵ∣|\epsilon| is lower-bounded by r>0r > 0 on (qα−η,qα+η)(q_\alpha - \eta, q_\alpha + \eta) for some η>M2rE[Δn2(X)]\eta > \frac{M}{2r} \mathbb{E}[\Delta_n^2(X)], the difference in half-widths is bounded quadratically in the estimation error:

    ∣qn,α−qα∣≤M2rE[Δn2(X)]|q_{n,\alpha} - q_\alpha| \le \frac{M}{2r} \mathbb{E}[\Delta_n^2(X)]

  8. Knowl 8 — Conformal Approximation of the Regular Oracle Under Stability

    theoretical result

    Conformal prediction bands closely approximate the Regular Oracle band Co∗(X)=[μ^n(X)−qn,α,μ^n(X)+qn,α]C_o^*(X) = [\hat{\mu}_n(X) - q_{n,\alpha}, \hat{\mu}_n(X) + q_{n,\alpha}] under sampling stability and perturbation sensitivity conditions.

    Assumption A2 (Sampling Stability): For large nn, P(∥μ^n−μ~∥∞≥ηn)≤ρn\mathbb{P}(\|\hat{\mu}_n - \tilde{\mu}\|_\infty \ge \eta_n) \le \rho_n for some target function μ~\tilde{\mu} and sequences ηn=o(1),ρn=o(1)\eta_n = o(1), \rho_n = o(1).

    Assumption A3 (Perturb-One Sensitivity): For candidate values y∈Yy \in \mathcal{Y}, P(sup⁡y∈Y∥μ^n−μ^n,(X,y)∥∞≥ηn)≤ρn\mathbb{P}\left(\sup_{y \in \mathcal{Y}} \|\hat{\mu}_n - \hat{\mu}_{n,(X,y)}\|_\infty \ge \eta_n\right) \le \rho_n.

    Under Assumptions A0, A1, A2, if the density f~\tilde{f} of ∣Y−μ~(X)∣|Y - \tilde{\mu}(X)| is lower bounded away from zero near its (1−α)(1 - \alpha)-quantile, the width νn,split\nu_{n,\text{split}} of the split conformal band satisfies:

    νn,split−2qn,α=OP(ρn+ηn+n−1/2)\nu_{n,\text{split}} - 2q_{n,\alpha} = O_P(\rho_n + \eta_n + n^{-1/2})

    Under the additional assumption that the response space Y\mathcal{Y} satisfies Assumption A3, the width νn,conf(X)\nu_{n,\text{conf}}(X) of the full conformal band at XX satisfies:

    νn,conf(X)−2qn,α=OP(ηn+ρn+n−1/2)\nu_{n,\text{conf}}(X) - 2q_{n,\alpha} = O_P(\eta_n + \rho_n + n^{-1/2})

  9. Knowl 9 — Super Oracle Approximation and Asymptotic Conditional Coverage

    theoretical result

    A sequence of prediction bands CnC_n has asymptotic conditional coverage at level 1−α1 - \alpha if there exist sets Λn⊆Rd\Lambda_n \subseteq \mathbb{R}^d such that P(X∈Λn∣Λn)=1−oP(1)\mathbb{P}(X \in \Lambda_n \mid \Lambda_n) = 1 - o_P(1) and

    inf⁡x∈Λn∣P(Y∈Cn(x)∣X=x)−(1−α)∣=oP(1)\inf_{x \in \Lambda_n} |\mathbb{P}(Y \in C_n(x) \mid X = x) - (1 - \alpha)| = o_P(1)

    Assumption A4 (Consistency): P(EX[(μ^n(X)−μ(X))2∣μ^n]≥ηn)≤ρn\mathbb{P}\left(\mathbb{E}_X[(\hat{\mu}_n(X) - \mu(X))^2 \mid \hat{\mu}_n] \ge \eta_n\right) \le \rho_n with ηn=o(1),ρn=o(1)\eta_n = o(1), \rho_n = o(1).

    Under Assumptions A0, A1, and A4, if ∣Y−μ(X)∣|Y - \mu(X)| has density bounded away from zero near its (1−α)(1 - \alpha)-quantile:

    1. The split conformal band satisfies L(Cn,split(X) Δ Cs∗(X))=oP(1)\mathcal{L}(C_{n,\text{split}}(X) \, \Delta \, C_s^*(X)) = o_P(1), where L\mathcal{L} is the Lebesgue measure, A Δ BA \, \Delta \, B denotes symmetric set difference, and Cs∗(x)=[μ(x)−qα,μ(x)+qα]C_s^*(x) = [\mu(x) - q_\alpha, \mu(x) + q_\alpha] is the super oracle band. Hence, Cn,splitC_{n,\text{split}} attains asymptotic conditional coverage at level 1−α1 - \alpha.
    2. If in addition perturb-one sensitivity (Assumption A3) holds, the full conformal band satisfies L(Cn,conf(X) Δ Cs∗(X))=oP(1)\mathcal{L}(C_{n,\text{conf}}(X) \, \Delta \, C_s^*(X)) = o_P(1), guaranteeing asymptotic conditional coverage at level 1−α1 - \alpha.
  10. Knowl 10 — Suboptimality of Multi-Split Conformal Prediction

    theoretical result

    To reduce the split-dependent variance of split conformal inference, one can combine NN independent random splits by intersecting individual split conformal intervals constructed at Bonferroni-corrected miscoverage levels α/N\alpha / N:

    Csplit(N)(x)=⋂j=1NCsplit,j(x)C_{\text{split}}^{(N)}(x) = \bigcap_{j=1}^N C_{\text{split}, j}(x)

    By Bonferroni's inequality, Csplit(N)C_{\text{split}}^{(N)} maintains marginal coverage of at least 1−α1 - \alpha.

    However, under sampling stability (Assumptions A0, A1, and A2 with ρn=o(n−1)\rho_n = o(n^{-1})) and continuous residual distribution ∣Y−μ~(X)∣|Y - \tilde{\mu}(X)|, the Bonferroni inflation penalty strictly dominates the intersection shrinking effect. Consequently, with probability tending to 1 as n→∞n \to \infty:

    Csplit(N)(X) is strictly wider than a single split interval Csplit(X)C_{\text{split}}^{(N)}(X) \text{ is strictly wider than a single split interval } C_{\text{split}}(X)

    Therefore, aggregating multiple conformal splits via Bonferroni intersections increases prediction interval length, and a single split is theoretically preferred.

  11. Knowl 11 — Jackknife Prediction Band Algorithm and Coverage Limitations

    algorithm

    The jackknife prediction method constructs prediction intervals using leave-one-out residuals.

    Input: Data (Xi,Yi)(X_i, Y_i) for i=1,…,ni = 1, \dots, n, miscoverage level α∈(0,1)\alpha \in (0, 1), regression algorithm A\mathcal{A}
    Output: Prediction band Cjack(x)C_{\text{jack}}(x) for all x∈Rdx \in \mathbb{R}^d
    Fit full estimator μ^=A({(Xi,Yi):i=1,…,n})\hat{\mu} = \mathcal{A}(\{(X_i, Y_i) : i = 1, \dots, n\})
    for i=1,…,ni = 1, \dots, n do
        Fit leave-one-out estimator: μ^(−i)=A({(Xℓ,Yℓ):ℓ≠i})\hat{\mu}_{(-i)} = \mathcal{A}(\{(X_\ell, Y_\ell) : \ell \ne i\})
        Compute leave-one-out residual: Ri=∣Yi−μ^(−i)(Xi)∣R_i = |Y_i - \hat{\mu}_{(-i)}(X_i)|
    end for
    Compute cutoff d=k-th smallest value in {Ri:i=1,…,n} where k=⌈n(1−α)⌉d = k\text{-th smallest value in } \{R_i : i = 1, \dots, n\} \text{ where } k = \lceil n(1 - \alpha)\rceil
    return Cjack(x)=[μ^(x)−d,μ^(x)+d]C_{\text{jack}}(x) = [\hat{\mu}(x) - d, \hat{\mu}(x) + d] for all x∈Rdx \in \mathbb{R}^d

    By symmetry, the jackknife band provides exact in-sample finite-sample validity:

    P(Yi∈Cjack(Xi))≥1−α,for all i=1,…,n\mathbb{P}(Y_i \in C_{\text{jack}}(X_i)) \ge 1 - \alpha, \quad \text{for all } i = 1, \dots, n

    However, unlike conformal methods, the jackknife fails to guarantee finite-sample out-of-sample validity P(Yn+1∈Cjack(Xn+1))≥1−α\mathbb{P}(Y_{n+1} \in C_{\text{jack}}(X_{n+1})) \ge 1 - \alpha. Asymptotic validity requires stringent conditions including uniform asymptotic mean squared error bounds, stability of μ^\hat{\mu}, homoskedasticity, and independence between features and noise.

  12. Knowl 12 — Empirical Robustness of Conformal Prediction Across Model Violations

    empirical result

    In simulation experiments comparing standard parametric linear model prediction intervals against full conformal, split conformal, and jackknife prediction intervals across three data regimes (Setting A: linear mean, Gaussian errors; Setting B: nonlinear additive mean, heavy-tailed t(2)t(2) errors; Setting C: linear mean, heteroskedastic t(2)t(2) errors, correlated features) in low (n=100,d=10n=100, d=10) and high (n=500,d=490n=500, d=490) dimensions:

    1. Marginal Coverage Robustness: Across all settings and all base estimators (Lasso, Elastic Net, forward stepwise, SPAM, Random Forests), conformal prediction intervals consistently achieve the nominal 90% marginal coverage (approx90.0%−90.7% approx 90.0\% - 90.7\%), even when the regression model is completely misspecified or errors are non-Gaussian and heteroskedastic.
    2. Failure of Parametric Intervals: Parametric prediction intervals over-cover drastically (e.g., 99.8%−100%99.8\% - 100\% coverage) and produce excessively wide prediction intervals (e.g., length 249.9249.9 vs 15.515.5 for conformal with ridge regression in Setting C) when homoskedasticity or normality assumptions break down.
    3. Computational vs Statistical Tradeoff: Split conformal prediction achieves coverage and interval widths comparable to full conformal prediction (e.g., in Setting A high-dimensional ridge: full conformal length 3.3483.348 vs split conformal length 3.3803.380), while reducing computation time by orders of magnitude (0.1550.155 seconds vs 167.19167.19 seconds).

Coverage note — The specific derivation verifying assumptions A2-A4 for high-dimensional lasso under restricted eigenvalue and irrepresentable conditions was omitted as it instantiates the general stability and consistency theorems.

References

  1. 1.Belloni, A., Chen, D., Chernozhukov, V., & Hansen, C. (2012). Sparse models and methods for optimal instruments with an application to eminent domain. Econometrica, 80 (6), 2369–2429.
  2. 2.Berk, R., Brown, L., Buja, A., Zhang, K., & Zhao, L. (2013). Valid post-selection inference. Annals of Statistics, 41 (2), 802–837.
  3. 3.Bickel, P. J., Ritov, Y., & Tsybakov, A. B. (2009). Simultaneous analysis of lasso and dantzig selector. The Annals of Statistics, (pp. 1705–1732).
  4. 4.Breiman, L. (2001). Random forests. Machine Learning, 45 (1), 5–32.
  5. 5.Buhlmann, P. (2013). Statistical significance in high-dimensional linear models. Bernoulli, 19 (4), 1212–1242.
  6. 6.Buja, A., Berk, R., Brown, L., George, E., Pitkin, E., Traskin, M., Zhang, K., & Zhao, L. (2014). Models as approximations: How random predictors and model violations invalidate classical inference in regression. ArXiv: 1404.1578.
  7. 7.Bunea, F., Tsybakov, A., & Wegkamp, M. (2007). Sparsity oracle inequalities for the lasso. Electronic Journal of Statistics, 1 , 169–194.
  8. 8.Burnaev, E., & Vovk, V. (2014). Efficiency of conformalized ridge regression. Proceedings of the Annual Conference on Learning Theory, 25 , 605–622.
  9. 9.Butler, R., & Rothman, E. (1980). Predictive intervals based on reuse of the sample. Journal of the American Statistical Association, 75 (372), 881–889.
  10. 10.Efroymson, M. A. (1960). Multiple regression analysis. In Mathematical Methods for Digital Computers, vol. 1, (pp. 191–203). Wiley.
  11. 11.Fithian, W., Sun, D., & Taylor, J. (2014). Optimal inference after model selection. ArXv: 1410.2597.
  12. 12.Hebiri, M. (2010). Sparse conformal predictors. Statistics and Computing, 20 (2), 253–266.
  13. 13.Javanmard, A., & Montanari, A. (2014). Confidence intervals and hypothesis testing for high-dimensional regression. Journal of Machine Learning Research, 15 , 2869–2909.
  14. 14.Lee, J., Sun, D., Sun, Y., & Taylor, J. (2016). Exact post-selection inference, with application to the lasso. Annals of Statistics, 44 (3), 907–927.
  15. 15.Lei, J. (2014). Classification with confidence. Biometrika, 101 (4), 755–769.
  16. 16.Lei, J., Rinaldo, A., & Wasserman, L. (2015). A conformal prediction approach to explore functional data. Annals of Mathematics and Artificial Intelligence, 74 (1), 29–43.
  17. 17.Lei, J., Robins, J., & Wasserman, L. (2013). Distribution free prediction sets. Journal of the American Statistical Association, 108 , 278–287.
  18. 18.Lei, J., & Wasserman, L. (2014). Distribution-free prediction bands for non-parametric regression. Journal of the Royal Statistical Society: Series B, 76 (1), 71–96.
  19. 19.Meinshausen, N., & Buhlmann, P. (2010). Stability selection. Journal of the Royal Statistical Society: Series B, 72 (4), 417–473.
  20. 20.Papadopoulos, H., Proedrou, K., Vovk, V., & Gammerman, A. (2002). Inductive confidence machines for regression. In Machine Learning: ECML 2002 , (pp. 345–356). Springer.
  21. 21.Ravikumar, P., Lafferty, J., Liu, H., & Wasserman, L. (2009). Sparse additive models. Journal of the Royal Statistical Society: Series B, 71 (5), 1009–1030.
  22. 22.Steinberger, L., & Leeb, H. (2016). Leave-one-out prediction intervals in linear regression models with many variables. ArXiv: 1602.05801.
  23. 23.Thakurta, A. G., & Smith, A. (2013). Differentially private feature selection via stability arguments, and the robustness of the lasso. In Conference on Learning Theory, (pp. 819–850).
  24. 24.Tian, X., & Taylor, J. (2015a). Asymptotics of selective inference. ArXiv: 1501.03588.
  25. 25.Tian, X., & Taylor, J. (2015b). Selective inference with a randomized response. ArXiv: 1507.06739.
  26. 26.Tibshirani, R. (1996). Regression shrinkage and selection via the lasso. Journal of the Royal Statistical Society: Series B, 58 (1), 267–288.
  27. 27.Tibshirani, R. J., Taylor, J., Lockhart, R., , & Tibshirani, R. (2016). Exact post-selection inference for sequential regression procedures. Journal of the American Statistical Association, 111 (514), 600–620.
  28. 28.van de Geer, S., Buhlmann, P., Ritov, Y., & Dezeure, R. (2014). On asymptotically optimal confidence regions and tests for high-dimensional models. Annals of Statistics, 42 (3), 1166–1201.
  29. 29.Vovk, V. (2013). Conditional validity of inductive conformal predictors. Machine Learning, 92 , 349–376.
  30. 30.Vovk, V., Gammerman, A., & Shafer, G. (2005). Algorithmic Learning in a Random World. Springer.
  31. 31.Vovk, V., Nouretdinov, I., & Gammerman, A. (2009). On-line predictive linear regression. The Annals of Statistics, 37 (3), 1566–1590.
  32. 32.Wasserman, L. (2014). Discussion: A significance test for the lasso. Annals of Statistics, 42 (2), 501–508.
  33. 33.Zhang, C.-H., & Zhang, S. (2014). Confidence intervals for low dimensional parameters in high dimensional linear models. Journal of the Royal Statistical Society: Series B, 76 (1), 217–242.
  34. 34.Zou, H., & Hastie, T. (2005). Regularization and variable selection via the elastic net. Journal of the Royal Statistical Society: Series B, 67 (2), 301–320.

Citation

MLA
Lei, J., et al. “Distribution-Free Predictive Inference For Regression”. arXiv, 2016, http://arxiv.org/abs/1604.04173v2.
APA
Lei, J., G'Sell, M., Rinaldo, A., Tibshirani, R. J., & Wasserman, L. (2016). Distribution-Free Predictive Inference For Regression. arXiv. http://arxiv.org/abs/1604.04173v2
Chicago
Lei, J., M. G'Sell, A. Rinaldo, R. J. Tibshirani, and L. Wasserman. 2016. “Distribution-Free Predictive Inference For Regression”. arXiv. http://arxiv.org/abs/1604.04173v2.
Harvard
Lei, J. et al. (2016) “Distribution-Free Predictive Inference For Regression”, arXiv [Preprint]. Available at: http://arxiv.org/abs/1604.04173v2.
Vancouver
1. Lei J, G'Sell M, Rinaldo A, Tibshirani RJ, Wasserman L (2016) Distribution-Free Predictive Inference For Regression. arXiv

BibTeX

@article{lei2016distribution,
  title = {Distribution-Free Predictive Inference For Regression},
  author = {Lei, Jing and G'Sell, Max and Rinaldo, Alessandro and Tibshirani, Ryan J. and Wasserman, Larry},
  year = {2016},
  journal = {arXiv},
  url = {http://arxiv.org/abs/1604.04173v2},
  eprint = {1604.04173}
}
Metadata:arXiv

Access the Paper

This paper is available from its original source. Click below to access the PDF.

Open PDF