Semi-supervised Batch Active Learning via Bilevel Optimization

Code

• github.com/zalanborsos/bilevel_coresets
  Officialpytorch★ 68

Abstract

Active learning is an effective technique for reducing the labeling cost by improving data efficiency. In this work, we propose a novel batch acquisition strategy for active learning in the setting where the model training is performed in a semi-supervised manner. We formulate our approach as a data summarization problem via bilevel optimization, where the queried batch consists of the points that best summarize the unlabeled data pool. We show that our method is highly effective in keyword detection tasks in the regime when only few labeled samples are available.

Tasks

Reproductions