Skip to content
Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning.
Python Jupyter Notebook
Branch: master
Clone or download
Latest commit 2304f16 Jul 5, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
laaos_results/paper Init BatchBALD repository with a git export. Jun 21, 2019
notebooks/paper Init BatchBALD repository with a git export. Jun 21, 2019
src Format the code. Jun 21, 2019
LICENSE
README.md Filename fix Jul 3, 2019
pyproject.toml Init BatchBALD repository with a git export. Jun 21, 2019
requirements.txt Init BatchBALD repository with a git export. Jun 21, 2019

README.md

BatchBALD

This is the code drop for our paper BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning.

The code comes as is.

Please cite us:

@misc{kirsch2019batchbald,
    title={BatchBALD: Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning},
    author={Andreas Kirsch and Joost van Amersfoort and Yarin Gal},
    year={2019},
    eprint={1906.08158},
    archivePrefix={arXiv},
    primaryClass={cs.LG}
}

How to run it

Make sure you install all requirements using

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
pip install -r requirements.txt

and you can start an experiment using:

python src/run_experiment.py --quickquick --num_inference_samples 10 --available_sample_k 40

which starts an experiment on a subset of MNIST with 10 MC dropout samples and acquisition size 40.

Have fun playing around with it!

You can’t perform that action at this time.