Geometric Conformal Prediction with Spatial Ranks and Multivariate Quantiles
Anton Conrad, Eric Moulines, julien perez
Abstract
In multi-target regression and multi-class classification, uncertainty is inherently multivariate: prediction regions must capture joint dependencies across correlated outputs. Conformal prediction provides distribution-free guarantees, yet extending it to vector-valued outputs remains challenging—scalar aggregation discards geometric structure, while optimal transport (OT) approaches are computationally demanding and sensitive to outliers. We introduce two conformal methods based on geometric quantiles and spatial ranks: Geometric Conformalized Quantile Regression (GCQR) constructs prediction regions from learned conditional geometric quantiles, while Geometric Rank Conformal Prediction (GRCP) uses the radial rank of vector-valued conformity scores as the nonconformity measure. We propose multiple estimators offering different tradeoffs between computational cost and adaptivity to feature-dependent heterogeneity, with scalable learning via partially input-convex neural networks. On multi-target regression and multi-class classification benchmarks, GCQR and GRCP attain near-nominal coverage with consistently tighter prediction regions than scalarized and multivariate baselines.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e25e89ce-2a5e-42a0-821a-4cf672ee1046Builds on16
- Classification with Valid and Adaptive CoverageYaniv Romano, Matteo Sesia, Emmanuel J. CandèsNeurIPS 2020 · 586 citations
- Optimal transport mapping via input convex neural networksAshok Vardhan Makkuva, Amirhossein Taghvaei, Sewoong Oh, Jason D. LeeICML 2020 · 254 citations
- Neural Optimal TransportAlexander Korotin, Daniil Selikhanovych, Evgeny BurnaevICLR 2023 · 151 citations
- Convex Potential Flows: Universal Probability Distributions with Optimal Transport and Convex OptimizationChin-Wei Huang, Ricky T. Q. Chen, Christos Tsirigotis, Aaron C. CourvilleICLR 2021 · 107 citations
- Improving Conditional Coverage via Orthogonal Quantile RegressionShai Feldman, Stephen Bates, Yaniv RomanoNeurIPS 2021 · 68 citations
Related papers
- Optimal transport-based conformal predictionGauthier Thurin, Kimia Nadjahi, Claire BoyerICML 2025
- Neural Optimal Transport Meets Multivariate Conformal PredictionVladimir Kondratyev, Alexander Fishkov, Mahmoud Hegazy, Nikita Kotelevskii et al.ICLR 2026 · 2 citations
- A Unified Comparative Study with Generalized Conformity Scores for Multi-Output Conformal RegressionVictor Dheur, Matteo Fontana, Yorick Estievenart, Naomi Desobry et al.ICML 2025
- Conformal Prediction for Ensembles: Improving Efficiency via Score-Based AggregationYash Patel, Eduardo Ochoa Rivera, Ambuj TewariNeurIPS 2025 · 9 citations
- Relational Conformal Prediction for Correlated Time SeriesAndrea Cini, Alexander Jenkins, Danilo P. Mandic, Cesare Alippi et al.ICML 2025
