A PyTorch implementation of BatchBALD on the MNIST dataset.
Bayesian Active Learning by Disagreement (BALD) is a mutual-information-based method for active learning -- selecting the most informative samples from a pool of data.
- Followed description of BALD in Deep Bayesian Active Learning with Image Data
- Follow description of BatchBALD in BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning
Requires Python3.
pip3 install torch torchvision matplotlib numpy
python3 main.py