arXiv · 2301.05490
Scalable Batch Acquisition for Deep Bayesian Active Learning
Abstract
In deep active learning, it is especially important to choose multiple examples to markup at each step to work efficiently, especially on large datasets. At the same time, existing solutions to this problem in the Bayesian setup, such as BatchBALD, have significant limitations in selecting a large number of examples, associated with the exponential complexity of computing mutual information for joint random variables. We, therefore, present the Large BatchBALD algorithm, which gives a well-grounded approximation to the BatchBALD method that aims to achieve comparable quality while being more computationally efficient. We provide a complexity analysis of the algorithm, showing a reduction in computation time, especially for large batches. Furthermore, we present an extensive set of experimental results on image and text data, both on toy datasets and larger ones such as CIFAR-100.
Explore related subjects
Keep this discovery
Aleksandr Rubashevskii, Daria Kotova, Maxim Panov. 2023-01-13. Scalable Batch Acquisition for Deep Bayesian Active Learning. https://arxiv.org/abs/2301.05490
Cite the original work for its findings. Save a collection to share your selection of sources.