Uncertainty Herding: One Active Learning Method for All Label Budgets

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bae, Wonho, Oliveira, Gabriel L., Sutherland, Danica J.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916632516886528
author Bae, Wonho
Oliveira, Gabriel L.
Sutherland, Danica J.
author_facet Bae, Wonho
Oliveira, Gabriel L.
Sutherland, Danica J.
contents Most active learning research has focused on methods which perform well when many labels are available, but can be dramatically worse than random selection when label budgets are small. Other methods have focused on the low-budget regime, but do poorly as label budgets increase. As the line between "low" and "high" budgets varies by problem, this is a serious issue in practice. We propose uncertainty coverage, an objective which generalizes a variety of low- and high-budget objectives, as well as natural, hyperparameter-light methods to smoothly interpolate between low- and high-budget regimes. We call greedy optimization of the estimate Uncertainty Herding; this simple method is computationally fast, and we prove that it nearly optimizes the distribution-level coverage. In experimental validation across a variety of active learning tasks, our proposal matches or beats state-of-the-art performance in essentially all cases; it is the only method of which we are aware that reliably works well in both low- and high-budget settings.
format Preprint
id arxiv_https___arxiv_org_abs_2412_20644
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty Herding: One Active Learning Method for All Label Budgets
Bae, Wonho
Oliveira, Gabriel L.
Sutherland, Danica J.
Machine Learning
Most active learning research has focused on methods which perform well when many labels are available, but can be dramatically worse than random selection when label budgets are small. Other methods have focused on the low-budget regime, but do poorly as label budgets increase. As the line between "low" and "high" budgets varies by problem, this is a serious issue in practice. We propose uncertainty coverage, an objective which generalizes a variety of low- and high-budget objectives, as well as natural, hyperparameter-light methods to smoothly interpolate between low- and high-budget regimes. We call greedy optimization of the estimate Uncertainty Herding; this simple method is computationally fast, and we prove that it nearly optimizes the distribution-level coverage. In experimental validation across a variety of active learning tasks, our proposal matches or beats state-of-the-art performance in essentially all cases; it is the only method of which we are aware that reliably works well in both low- and high-budget settings.
title Uncertainty Herding: One Active Learning Method for All Label Budgets
topic Machine Learning
url https://arxiv.org/abs/2412.20644