/ISL4MIS

Image-tag-supervised learning for medical image segmentation.

Primary LanguagePython

Image-tag-supervised learning for medical image segmentation (ISL4MIS).

1. Code for image-tag-supervised learning

  1. train the classification network

    python scripts/train_cls.py --img_dir your_img_dir
  2. generate cam using trained classification network

    python scripts/cam.py --checkpoint best_checkpoint.pth

2. Literature reviews of image-tag-supervised learning approach for medical image segmentation.

2.1 State-of-the-art methods

Date Paper source Title Code
2022.09 MICCAI Online Easy Example Mining for Weakly-supervised Gland Segmentation from Histology Images PyTorch
2022.08 MedIA Weakly Supervised Segmentation on Neural Compressed Histopathology with Self-Equivariant Regularization PyTorch
2022.08 MedIA Multi-Layer Pseudo-Supervision for Histopathology Tissue Semantic Segmentation using Patch-level Classification Labels PyTorch
2022.06 CVPR (oral) Adaptive Early-Learning Correction for Segmentation from Noisy Annotations PyTorch
2022.06 CVPR C-CAM: Causal CAM for Weakly Supervised Semantic Segmentation on Medical Image
2022.03 TMI Deep Interpretable Classification and Weakly-Supervised Segmentation of Histology Images via Max-Min Uncertainty PyTorch
2021.06 JBHI Lung Lesion Localization of COVID-19 From Chest CT Image: A Novel Weakly Supervised Learning Method PyTorch
2021.02 MedIA An Interpretable Classifier for High-Resolution Breast Cancer Screening Images Utilizing Weakly Supervised Localization PyTorch
2020.12 JBHI MS-CAM: Multi-Scale Class Activation Maps for Weakly-Supervised Segmentation of Geographic Atrophy Lesions in SD-OCT Images Tensorflow
2020.09 MICCAI Weakly Supervised Organ Localization with Attention Maps Regularized by Local Area Reconstruction
2020.09 Artificial Intelligence in Medicine Weakly-Supervised Segmentation for Disease Localization in Chest X-Ray Images PyTorch

2.2 Literature reviews of CAM Variants.

Date Paper source Title Code
2020.06 CVPR workshop Score-CAM: Score-Weighted Visual Explanations for Convolutional Neural Networks PyTorch
2018.05 WACV Grad-CAM++: Generalized Gradient-based Visual Explanations for Deep Convolutional Networks
2017.10 ICCV Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization PyTorch
2016.06 CVPR Learning Deep Features for Discriminative Localization PyTorch

Acknowledgement

  • Part of the code is adapted from open-source codebase and original implementations of algorithms, we thank these authors for their fantastic and efficient codebase, such as DRS and pytorch-grad-cam.