This is the demo for the paper ``Loss-based Attention for Deep Multiple Instance Learning'' AAAI 2020.

The method is implemented by PyTorch 1.1.0.

Image classification contains an example for image classification on CIFAR10

MIL contains an example for MIL classification using MNIST-bags