Skip to content
master
Switch branches/tags
Code

BatchBALD

Note: A more modular re-implementation can be found at https://github.com/BlackHC/batchbald_redux.


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

The code comes as is.

See https://github.com/BlackHC/batchbald_redux and https://blackhc.github.io/batchbald_redux/ for a reimplementation.

ElementAI's Baal framework also supports BatchBALD: https://github.com/ElementAI/baal/.

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!

About

Efficient and Diverse Batch Acquisition for Deep Bayesian Active Learning.

Topics

Resources

License

Releases

No releases published

Packages

No packages published