Localized Conformal Multi-Quantile Regression
Yuan Lu
TL;DR
<3-5 sentence high-level summary> Localized Conformal Multi-Quantile Regression (LCMQR) addresses the dual challenges of inefficiency and lack of local adaptivity in uncertainty quantification by marrying multi-quantile regression with kernel-based localization within a split-conformal framework. The authors resolve a theoretical inconsistency in CCQR with a consistent Average-then-Max scoring, proving tighter, valid prediction intervals and establishing marginal and asymptotic conditional coverage. They further extend the approach to Group-Calibrated LCMQR (GC-LCMQR), adding stratified calibration to guarantee finite-sample validity within subgroups in heterogeneous populations, demonstrated on both benchmark regression tasks and an ITE setting. Empirical results show superior interval efficiency for LCMQR and robust group-wise validity for GC-LCMQR, including safe uncertainty quantification for high-variance focal groups in treatment-effect scenarios.
Abstract
Standard conformal prediction methods guarantee marginal coverage but often produce inefficient intervals that fail to adapt to local heteroscedasticity, while recent localized approaches often struggle to maintain validity across distinct subpopulations with varying noise profiles. To address these challenges, we introduce Localized Conformal Multi-Quantile Regression (LCMQR), a novel framework that synergizes multi-quantile information with kernel-based localization to construct efficient and adaptive prediction intervals. Theoretically, we resolve an inconsistency in Conformalized Composite Quantile Regression (CCQR) by proving that our consistent Average-then-Max scoring mechanism systematically yields tighter intervals than the Max-then-Average approach used in prior work. For heterogeneous environments, we extend this framework to Group-Calibrated LCMQR (GC-LCMQR) via a stratified calibration step that guarantees finite-sample validity within distinct subgroups. Experiments on benchmark datasets and an Individual Treatment Effect (ITE) task demonstrate that LCMQR achieves superior efficiency on standard benchmarks, while GC-LCMQR uniquely achieves group-level coverage for target subgroups in mixture populations where baselines fail.
