Adaptive conformal prediction on the Grassmannian

arXiv:2610.23508 · stat.ML, cs.LG · Submitted 2026-10-16 · Read on arXiv

stat.ML, cs.LG

Submitted: 2026-10-16

Updated: 2026-10-16

Comments: 21 pages, 6 figures, 3 tables

License: http://creativecommons.org/licenses/by-nc-nd/4.0/

The gist: We introduce a conformal prediction framework for regression with Grassmann-valued responses.

Terminology

Abstract

We introduce a conformal prediction framework for regression with Grassmann-valued responses. The proposed method constructs prediction regions in the form of metric balls directly on the Grassmann manifold, using basis-invariant geometric distances as non-conformity measures and scaling the corresponding non-conformity scores through a learned prediction difficulty estimator. This yields locally adaptive, input-dependent prediction regions that respect the underlying manifold geometry, while retaining the finite-sample marginal coverage guarantee of split conformal prediction under exchangeability. We evaluate several Grassmannian distance metrics and compare their prediction-region efficiency on synthetic benchmarks and an airfoil inverse shape-design problem. The results highlight the advantages of geometry-aware non-conformity scores, while local adaptation improves empirical coverage uniformity across prediction difficulty levels and, in strongly heteroscedastic settings, can substantially reduce prediction-region size while maintaining the target coverage.

Related papers