Adaptive conformal prediction on the Grassmannian

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.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Machine Learning (Statistics)

• arXiv preprint · 16 October 2026

  Unit-level aggregation breaks coverage in causal foundation models

  Gabe Schulman, Lucius E. J. Bynum and Yoav Wald
• arXiv preprint · 17 October 2026

  Uniform convergence at optimal rate and scale

  Noga Alon, Steve Hanneke, Shay Moran and 1 other
• arXiv preprint · 16 October 2026

  Conformal Policy Selection: When to Deploy AI Decisions for Targeted Intervention

  Junyu Cao, Ruijiang Gao, Maytal Saar-Tsechansky and 1 other
• arXiv preprint · 17 October 2026

  The Certification Exponent: Instance-Optimal Sequential Certification of Policy Improvement from History Dependent Logs

  Rui Wang
• arXiv preprint · 16 October 2026

  Towards Hyperparameter Transfer for Differentially Private Optimization

  Tianze Wang, Zhiqi Bu and Linjun Zhang

All new papers in Machine Learning (Statistics)