Skip to content

Implementation of our paper "Instance Explainable Multi-Instance Learning for ROI of Various Data"

Notifications You must be signed in to change notification settings

thu-west/MU-MIL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Instance Explainable Multi-Instance Learning for ROI of Various Data

by XuZhao (zhaoxu18@mails.tsinghua.edu.cn) and ZihaoWang (wzh17@mails.tsinghua.edu.cn)

Overview

PyTorch implementation of our paper "Instance Explainable Multi-Instance Learning for ROI of Various Data" accepted by DASFAA 2020.

Dependencies

Install Pytorch 1.3.0, using pip or conda, should resolve all dependencies. Install the package imgaug by pip. Tested with Python 3.6, but should work with 3.x as well. Tested on GPU.

Dataset

You can download the datasets we introduced in our paper from following links:

How to Use

cc_src/*: Evaluate our model specifically on the colon cancer dataset.

classic_src/*: Evaluate our model specifically on the MUSK1, MUSK2, Fox, Elephant, Tiger datasets.

mnist_src/*: Evaluate our model specifically on the MNIST-Bags dataset. NOTE: The codes will automatically download the original MNIST dataset and generate the MNIST-Bags dataset. It can handle any bag length without the dataset becoming unbalanced. It is most probably not the most efficient way to create the bags. Furthermore it is only test for the case that the target number is ‘9’.

src/*: We implement the uniform interface for various datasets introduced above. You can specify the dataset by setting hte parameter --datasetand simply run the main.py as follows:

python main.py --dataset cc

The program will print the AUC, Precision, Recall, Accuracy, Skrewness, dloss of the trained model. Part of settable parameters are listed as follows:

Parameter Options Usage
--dataset [cc, bc, mb, musk1, musk2, fox, tiger, elephant] Specify the dataset for evaluation
--attention [att, mu, datt, mhatt] Specify the attention type
--epochs Specify the epoch nums for training
--lr Specify the learning rate
--dim Sepcify the dimension for the attention

Some other settable parameters could be found in the src/main.py file.

Citation

If you want to refer to our work, please cite our paper as:

@article{
  title={Instance Explainable Multi-Instance Learning for ROI of Various Data},
  author={Xu Zhao, Zihao Wang, Yong Zhang and Chunxiao Xing},
  booktitle={DASFAA},
  year={2020},
}

About

Implementation of our paper "Instance Explainable Multi-Instance Learning for ROI of Various Data"

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages