This notebook shows how to estimate conformal prediction intervals with the marginaleffects package for R.

11.1 Confidence vs. prediction intervals

The predictions() function from the marginaleffects() package can compute confidence intervals for fitted values in over 80 model classes in R. These intervals quantify the uncertainty about the expected value of the response. A common misunderstanding is that these confidence intervals should be calibrated to cover a certain percentage of unseen data points. This is not the case. In fact, a 95% confidence interval reported by predictions() will typically cover a much smaller share of out-of-sample outcomes.

Consider this simulation where \(Y_{train}\) and \(Y_{test}\) are drawn from a normal distribution with mean \(\pi\) and standard deviation 1. We estimate a linear model with an intercept only, and compute a 90% confidence interval for the expected value of the response:

Test set coverage True mean coverage
0.2498 0.8770

We see that the confidence interval around predictions covers the true mean of \(\pi\) about 90% of the time, whereas coverage of individual observations in the test set is much lower.

If we care about out of sample predictions, that is, if we want our interval to cover a specific share of the actual outcome for unobserved individuals, we must compute “prediction intervals” instead of “confidence intervals.” How do we do this? Conformal prediction is very flexible and powerful approach.

11.2 Conformal prediction

In their excellent tutorial, Angelopoulos and Bates (2022) write that conformal prediction is

“a user-friendly paradigm for creating statistically rigorous uncertainty sets/intervals for the predictions of such models. Critically, the sets are valid in a distribution-free sense: they possess explicit, non-asymptotic guarantees even without distributional assumptions or model assumptions.

These are extraordinary claims which deserve to be underlined: In principle, conformal prediction should offer well-calibrated intervals, regardless of the prediction model we use, and even if that model is misspecified.

The main caveats are:

The conformal prediction algorithms implemented in marginaleffects are designed for exchangeable data.^{1} They do not offer coverage guarantees in contexts where exchangeability is violated, such as in time series data, when there is spatial dependence between observations, or when there is distribution drift between the training and test data.

The conformal prediction algorithms implemented in marginaleffects offer marginal coverage guarantees, that is, they guarantee that a random test point will fall within the interval with a given probability. Below, we show an example where the prediction interval covers the right number of test points overall, but is not well calibrated locally, in different strata of the predictors. Different algorithms have recently been proposed to offer class-conditional coverage guarantees (see Ding et al. (2023) for an example).

The width of the conformal prediction interval will typically depend on the quality of the prediction model and of the score function.

The score functions implemented in marginaleffects simply take the residual—or difference between the observed outcome and predicted value. This means that the type argument must ensure that observations and predictions are on commensurable scales (usually type="response" or type="prob").

11.3 Data and models: Linear, Logit, Multinomial Logit

Download data, split it into training and testing sets, and estimate a few different models:

# download datadat<-read.csv("https://vincentarelbundock.github.io/Rdatasets/csv/openintro/military.csv")# create a binary outcome variabledat<-transform(dat, officer =as.numeric(grepl("officer", grade)))# train/test splitidx<-sample(seq_len(nrow(dat)), 60000)test<-dat[idx[1:10000], ]train<-dat[idx[10001:length(idx)], ]# linear regressionm_lm<-lm(rank~gender*race, data =train)p_lm<-predictions(m_lm, newdata =train)# logit regressionm_glm<-glm(officer~gender*race, data =train, family =binomial)p_glm<-predictions(m_glm, newdata =train)# multinomial logit regressionm_mult<-multinom(branch~gender*race, data =train, trace =FALSE)p_mult<-predictions(m_mult, newdata =train)

For LM and GLM models, predictions() returns a data frame with one prediction for each row of the original data. This data frame includes confidence intervals:

For multinomial models, predictions() returns a data frame with one prediction for each row and for each outcome level. We can see the predicted probabilities of each outcome level for the first observation in the original data:

Group Estimate Std. Error z Pr(>|z|) S CI low CI high
air force 0.181 0.00479 37.8 <0.001 Inf 0.171 0.190
army 0.473 0.00621 76.2 <0.001 Inf 0.461 0.485
marine corps 0.101 0.00376 27.0 <0.001 530.9 0.094 0.109
navy 0.245 0.00535 45.7 <0.001 Inf 0.234 0.255
Columns: rowid, group, estimate, std.error, statistic, p.value, s.value, conf.low, conf.high, rownames, grade, branch, gender, race, hisp, rank, officer

11.4 Conformal predictions with inferences()

In the “Bootstrap and Simultation” vignette, we saw that the inferences() function can be used to compute confidence intervals for any marginaleffects package estimates. The workflow is simple:

Pass the resulting object to inferences(), along with arguments to specify how to perform inference to obtain uncertainty estimates.

inferences() supports two strategies for conformal prediction: split or CV+ Ding et al. (2023). The former is faster but less efficient. In the rest of this vignette, we illustrate how to use this same workflow to compute conformal prediction intervals.

11.4.1 Cross-validation +

The p_lm, p_glm, and p_mult objects are predictions() objects. They contain the point predictions and confidence intervals for each observation in the training set. Now, we use the inferences() function to compute predictions and prediction intervals for every observation in the test set:

p<-predictions(m_lm, conf_level =.9)|>inferences( R =5, method ="conformal_cv+", conformal_test =test)p

The coverage also seems adequate (about 80%) for the logit model:

p<-predictions(m_glm, conf_level =.8)|>inferences( R =5, method ="conformal_cv+", conformal_test =test)mean(p$officer<=p$pred.high&p$officer>=p$pred.low)

[1] 0.7998

When the outcome is categorical, we use conformal_score="softmax". With this argument, inferences() generates “conformal prediction sets,” that is, sets of possible outcome classes with coverage guarantees. inferences() returns a list column of sets for each observation. On average, those sets should cover the true value about 70% of the time:

branch Pred Set
army air force, army
navy air force, army , navy
navy air force, army
army army, navy
air force air force, army
army army, navy
Columns: rowid, branch, pred.set

For example, for the first observation in the dataset, the conformal prediction is {air force, army} and the true value is army. The conformal prediction set thus covers the true value. The coverage rate is:

For split conformal prediction, we must first split the training set into a training and a calibration set (see Angelopoulos and Bates (2022)). Then, we pass the calibration set to the inferences() function:

As noted above, the conformal prediction interval should be valid even if the model is misspecified. To illustrate this, we generate data from a linear model with polynomials, but estimate a linear model without polynomials. Then, we plot the results and compute the coverage of the prediction interval:

N<-1000X<-rnorm(N*2)dat<-data.frame( X =X, Y =X+X^2+X^3+rnorm(N*2))train<-dat[1:N,]test<-dat[(N+1):nrow(dat),]m<-lm(Y~X, data =train)p<-predictions(m)|>inferences( R =5, method ="conformal_cv+", conformal_test =test)mean(p$Y<=p$pred.high&p$Y>=p$pred.low)

[1] 0.953

ggplot(p, aes(X, Y))+geom_point(alpha =.1)+geom_ribbon(aes(X, ymin =pred.low, ymax =pred.high), alpha =.2, fill ="#F0E442")+geom_ribbon(aes(X, ymin =conf.low, ymax =conf.high), alpha =.4, fill ="#D55E00")+theme_bw()+labs( title ="Confidence and prediction intervals for a misspecified linear model", subtitle =sprintf("Confidence coverage (orange): %.2f%%; Prediction coverage (yellow): %.2f%%.",mean(p$Y<=p$conf.high&p$Y>=p$conf.low),mean(p$Y<=p$pred.high&p$Y>=p$pred.low)))

This example is interesting, because it shows that the prediction interval has adquate marginal coverage. However, the intervals are not necessarily well calibrated “locally”, in different strata of \(X\). In the figure above, our model is misspecified, so we make more mistakes in the tails, where predictions are bad. In contrast, the interval catches more observations in the middle of the distribution, which ensures that the overall error rate is adequate.

11.5.2 Poisson vs Negative Binomial

Here is a second example of model misspecification. We generate data from a negative binomial model, but estimate a Poisson model. Nevertheless, the conformal prediction interval has good coverage:

n<-10000X<-rnorm(n)eta<--1+2*Xmu<-exp(eta)Y<-rnegbin(n, mu =mu, theta =1)dat<-data.frame(X =X, Y =Y)train<-dat[1:5000,]test<-dat[5001:nrow(dat),]mod<-glm(Y~X, data =train, family =poisson)p<-predictions(mod, conf_level =.9)|>inferences( method ="conformal_cv+", R =10, conformal_test =test)mean(p$Y>=p$pred.low&p$Y<=p$pred.high)

[1] 0.8968

Angelopoulos, Anastasios N., and Stephen Bates. 2022. “A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantification,” no. arXiv:2107.07511 (September). https://doi.org/10.48550/arXiv.2107.07511.

Ding, Tiffany, Anastasios N. Angelopoulos, Stephen Bates, Michael I. Jordan, and Ryan J. Tibshirani. 2023. “Class-Conditional Conformal Prediction with Many Classes,” no. arXiv:2306.09335 (June). https://doi.org/10.48550/arXiv.2306.09335.