/TransMIL

Implementation of the TransMIL model

Primary LanguagePython

TransMIL Model

This is the Pytorch implementation of the MIL model described in TransMIL: Transformer based Correlated Multiple Instance Learning for Whole Slide Image Classication.

Please cite the original authors when using this code.

@article{DBLP:journals/corr/abs-2106-00908,
  author    = {Zhuchen Shao and
               Hao Bian and
               Yang Chen and
               Yifeng Wang and
               Jian Zhang and
               Xiangyang Ji and
               Yongbing Zhang},
  title     = {TransMIL: Transformer based Correlated Multiple Instance Learning
               for Whole Slide Image Classication},
  journal   = {CoRR},
  volume    = {abs/2106.00908},
  year      = {2021},
  url       = {https://arxiv.org/abs/2106.00908},
  eprinttype = {arXiv},
  eprint    = {2106.00908},
  timestamp = {Wed, 09 Jun 2021 18:45:08 +0200},
  biburl    = {https://dblp.org/rec/journals/corr/abs-2106-00908.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}