CP$^2$: Leveraging Geometry for Conformal Prediction via Canonicalization

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: van der Linden, Putri A., Timans, Alexander, Bekkers, Erik J.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918064352657408
author van der Linden, Putri A.
Timans, Alexander
Bekkers, Erik J.
author_facet van der Linden, Putri A.
Timans, Alexander
Bekkers, Erik J.
contents We study the problem of conformal prediction (CP) under geometric data shifts, where data samples are susceptible to transformations such as rotations or flips. While CP endows prediction models with post-hoc uncertainty quantification and formal coverage guarantees, their practicality breaks under distribution shifts that deteriorate model performance. To address this issue, we propose integrating geometric information--such as geometric pose--into the conformal procedure to reinstate its guarantees and ensure robustness under geometric shifts. In particular, we explore recent advancements on pose canonicalization as a suitable information extractor for this purpose. Evaluating the combined approach across discrete and continuous shifts and against equivariant and augmentation-based baselines, we find that integrating geometric information with CP yields a principled way to address geometric shifts while maintaining broad applicability to black-box predictors.
format Preprint
id arxiv_https___arxiv_org_abs_2506_16189
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CP$^2$: Leveraging Geometry for Conformal Prediction via Canonicalization
van der Linden, Putri A.
Timans, Alexander
Bekkers, Erik J.
Machine Learning
Artificial Intelligence
We study the problem of conformal prediction (CP) under geometric data shifts, where data samples are susceptible to transformations such as rotations or flips. While CP endows prediction models with post-hoc uncertainty quantification and formal coverage guarantees, their practicality breaks under distribution shifts that deteriorate model performance. To address this issue, we propose integrating geometric information--such as geometric pose--into the conformal procedure to reinstate its guarantees and ensure robustness under geometric shifts. In particular, we explore recent advancements on pose canonicalization as a suitable information extractor for this purpose. Evaluating the combined approach across discrete and continuous shifts and against equivariant and augmentation-based baselines, we find that integrating geometric information with CP yields a principled way to address geometric shifts while maintaining broad applicability to black-box predictors.
title CP$^2$: Leveraging Geometry for Conformal Prediction via Canonicalization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2506.16189