ASPEST: Bridging the Gap Between Active Learning and Selective Prediction

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Chen, Jiefeng, Yoon, Jinsung, Ebrahimi, Sayna, Arik, Sercan, Jha, Somesh, Pfister, Tomas
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909123694559232
author Chen, Jiefeng
Yoon, Jinsung
Ebrahimi, Sayna
Arik, Sercan
Jha, Somesh
Pfister, Tomas
author_facet Chen, Jiefeng
Yoon, Jinsung
Ebrahimi, Sayna
Arik, Sercan
Jha, Somesh
Pfister, Tomas
contents Selective prediction aims to learn a reliable model that abstains from making predictions when uncertain. These predictions can then be deferred to humans for further evaluation. As an everlasting challenge for machine learning, in many real-world scenarios, the distribution of test data is different from the training data. This results in more inaccurate predictions, and often increased dependence on humans, which can be difficult and expensive. Active learning aims to lower the overall labeling effort, and hence human dependence, by querying the most informative examples. Selective prediction and active learning have been approached from different angles, with the connection between them missing. In this work, we introduce a new learning paradigm, active selective prediction, which aims to query more informative samples from the shifted target domain while increasing accuracy and coverage. For this new paradigm, we propose a simple yet effective approach, ASPEST, that utilizes ensembles of model snapshots with self-training with their aggregated outputs as pseudo labels. Extensive experiments on numerous image, text and structured datasets, which suffer from domain shifts, demonstrate that ASPEST can significantly outperform prior work on selective prediction and active learning (e.g. on the MNIST$\to$SVHN benchmark with the labeling budget of 100, ASPEST improves the AUACC metric from 79.36% to 88.84%) and achieves more optimal utilization of humans in the loop.
format Preprint
id arxiv_https___arxiv_org_abs_2304_03870
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle ASPEST: Bridging the Gap Between Active Learning and Selective Prediction
Chen, Jiefeng
Yoon, Jinsung
Ebrahimi, Sayna
Arik, Sercan
Jha, Somesh
Pfister, Tomas
Machine Learning
Selective prediction aims to learn a reliable model that abstains from making predictions when uncertain. These predictions can then be deferred to humans for further evaluation. As an everlasting challenge for machine learning, in many real-world scenarios, the distribution of test data is different from the training data. This results in more inaccurate predictions, and often increased dependence on humans, which can be difficult and expensive. Active learning aims to lower the overall labeling effort, and hence human dependence, by querying the most informative examples. Selective prediction and active learning have been approached from different angles, with the connection between them missing. In this work, we introduce a new learning paradigm, active selective prediction, which aims to query more informative samples from the shifted target domain while increasing accuracy and coverage. For this new paradigm, we propose a simple yet effective approach, ASPEST, that utilizes ensembles of model snapshots with self-training with their aggregated outputs as pseudo labels. Extensive experiments on numerous image, text and structured datasets, which suffer from domain shifts, demonstrate that ASPEST can significantly outperform prior work on selective prediction and active learning (e.g. on the MNIST$\to$SVHN benchmark with the labeling budget of 100, ASPEST improves the AUACC metric from 79.36% to 88.84%) and achieves more optimal utilization of humans in the loop.
title ASPEST: Bridging the Gap Between Active Learning and Selective Prediction
topic Machine Learning
url https://arxiv.org/abs/2304.03870