Multi-Instance Causal Representation Learning for Instance Label Prediction and Out-of-Distribution Generalization (NeurIPS-2022)

If you use any code from this repository, please kindly cite the following paper:

Multi-Instance Causal Representation Learning for Instance Label Prediction and Out-of-Distribution Generalization

Weijia Zhang, Xuanhui Zhang, Han-Wen Deng, Min-Ling Zhang

Advances in Neural Information Processing Systems 35 (NeurIPS-2022).

Paper can be downloaded from here

For questions regarding the code, please contact [email protected]

Requirements: PyTorch 1.12

To reproduce the results in the paper:

For MNIST, FashionMNIST, KuzushijiMNIST multi-instance datasets results, please use MNIST_bags.ipynb for training, testing and visulization.

For Out-of-Distribution (OOD) generalization results, please use ColorMNIST_OOD.ipynb for training, testing and visulization.

For Colon Cancer results, please use colon_cancer.py.

This piece of code provides a dataloader for processing MIL bags organized as folders of images.

The dataset can be obtained (credit to Dr. Jiawen Yao) from https://drive.google.com/file/d/1RcNlwg0TwaZoaFO0uMXHFtAo_DCVPE6z/view?usp=sharing