Repository for the paper: "Agree to Disagree: Diversity through Disagreement for Better Transferability"
To reproduce the 2D D-BAT results (Figure 1), see this notebook.
- To reproduce the C-MNIST results, see this notebook (
notebooks/D_BAT_C_MNIST.ipynb
) - To reproduce the MF-Dominoes results, see this notebook (
notebooks/D_BAT_M_F_Dominoes.ipynb
) - To reproduce the MC-Dominoes results (Table 1), see this notebook (
notebooks/D_BAT_M_C_Dominoes.ipynb
)
Install WILDS: pip install wilds
- To reproduce the Waterbirds results, run
train-waterbird.sh
. - To reproduce the Camelyon17 dataset, run
train-camelyon.sh
. - To reproduce the results on the Office-Home dataset, first download the data (see
datasets/README.md
for link), then runtrain-office-home-ood_is_test.sh
andtrain-office-home-ood_is_not_test.sh
- To reproduce results on the MNIST OOD setup (Figure 3), see this notebook (
notebooks/D_BAT_MNIST_Uncertainty.ipynb
) - To reproduce results on the CIFAR10 OOD setup (Figure 4), see this notebook (
notebooks/D_BAT_CIFAR10_OOD_CIFAR100_MCDropout.ipynb
)