Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Stochastic Batch Acquisition: A Simple Baseline for Deep Active Learning

About

We examine a simple stochastic strategy for adapting well-known single-point acquisition functions to allow batch active learning. Unlike acquiring the top-K points from the pool set, score- or rank-based sampling takes into account that acquisition scores change as new data are acquired. This simple strategy for adapting standard single-sample acquisition strategies can even perform just as well as compute-intensive state-of-the-art batch acquisition functions, like BatchBALD or BADGE, while using orders of magnitude less compute. In addition to providing a practical option for machine learning practitioners, the surprising success of the proposed method in a wide range of experimental settings raises a difficult question for the field: when are these expensive batch acquisition methods pulling their weight?

Andreas Kirsch, Sebastian Farquhar, Parmida Atighehchian, Andrew Jesson, Frederic Branchaud-Charron, Yarin Gal• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 (test)
Accuracy83.63
882
Image ClassificationFood101
Accuracy90.1
457
Text ClassificationAG News (test)
Accuracy83.31
293
Text ClassificationYelp (test)
Accuracy78.54
100
Image ClassificationfMoW (test)
Top-1 Accuracy91.2
60
Image ClassificationCIFAR100
Mean Accuracy91.2
55
Image ClassificationDomainNet Real
Mean Accuracy84.7
55
ClassificationCredit Card Fraud (test)
Accuracy93.51
45
ClassificationAirline Passenger Satisfaction (test)
Accuracy88.46
45
Image ClassificationiWildCam (test)
Accuracy81.72
45
Showing 10 of 22 rows

Other info

Follow for update