Improving Conditional Coverage via Orthogonal Quantile Regression
arXiv:2106.00394
Abstract
We develop a method to generate prediction intervals that have a user-specified coverage level across all regions of feature-space, a property called conditional coverage. A typical approach to this task is to estimate the conditional quantiles with quantile regression -- it is well-known that this leads to correct coverage in the large-sample limit, although it may not be accurate in finite samples. We find in experiments that traditional quantile regression can have poor conditional coverage. To remedy this, we modify the loss function to promote independence between the size of the intervals and the indicator of a miscoverage event. For the true conditional quantiles, these two quantities are independent (orthogonal), so the modified loss function continues to be valid. Moreover, we empirically show that the modified loss function leads to improved conditional coverage, as evaluated by several metrics. We also introduce two new metrics that check conditional coverage by looking at the strength of the dependence between the interval size and the indicator of miscoverage.
arXiv comment20 pages, 5 figures
References in corpus (9)
- Generative Moment Matching Networks
- Estimating conditional quantiles with the help of the pinball loss
- The Randomized Dependence Coefficient
- Nested conformal prediction and quantile out-of-bag ensemble methods
- Adaptive, Distribution-Free Prediction Intervals for Deep Networks
- Uncertainty Sets for Image Classifiers using Conformal Prediction
- Beyond Pinball Loss: Quantile Methods for Calibrated Uncertainty Quantification
- CD-split and HPD-split: efficient conformal regions in high dimensions
- Deep Learning for Quantile Regression under Right Censoring: DeepQuantreg