Adapting Prediction Sets to Distribution Shifts Without Labels

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Kasa, Kevin, Zhang, Zhiyu, Yang, Heng, Taylor, Graham W.
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916894407131136
author Kasa, Kevin
Zhang, Zhiyu
Yang, Heng
Taylor, Graham W.
author_facet Kasa, Kevin
Zhang, Zhiyu
Yang, Heng
Taylor, Graham W.
contents Recently there has been a surge of interest to deploy confidence set predictions rather than point predictions in machine learning. Unfortunately, the effectiveness of such prediction sets is frequently impaired by distribution shifts in practice, and the challenge is often compounded by the lack of ground truth labels at test time. Focusing on a standard set-valued prediction framework called conformal prediction (CP), this paper studies how to improve its practical performance using only unlabeled data from the shifted test domain. This is achieved by two new methods called ECP and EACP, whose main idea is to adjust the score function in CP according to its base model's own uncertainty evaluation. Through extensive experiments on a number of large-scale datasets and neural network architectures, we show that our methods provide consistent improvement over existing baselines and nearly match the performance of fully supervised methods.
format Preprint
id arxiv_https___arxiv_org_abs_2406_01416
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Adapting Prediction Sets to Distribution Shifts Without Labels
Kasa, Kevin
Zhang, Zhiyu
Yang, Heng
Taylor, Graham W.
Machine Learning
Recently there has been a surge of interest to deploy confidence set predictions rather than point predictions in machine learning. Unfortunately, the effectiveness of such prediction sets is frequently impaired by distribution shifts in practice, and the challenge is often compounded by the lack of ground truth labels at test time. Focusing on a standard set-valued prediction framework called conformal prediction (CP), this paper studies how to improve its practical performance using only unlabeled data from the shifted test domain. This is achieved by two new methods called ECP and EACP, whose main idea is to adjust the score function in CP according to its base model's own uncertainty evaluation. Through extensive experiments on a number of large-scale datasets and neural network architectures, we show that our methods provide consistent improvement over existing baselines and nearly match the performance of fully supervised methods.
title Adapting Prediction Sets to Distribution Shifts Without Labels
topic Machine Learning
url https://arxiv.org/abs/2406.01416