Skip to content

HiLab-git/MetaSSL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

3 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

MetaSSL: A General Heterogeneous Loss for Semi-Supervised Medical Image Segmentation

πŸ“Œ Overall Framework

Framework

The overall MetaSSL framework is summarized as follows:

  • MetaSSL is a versatile semi-supervised medical image segmentation method driven by a spatially heterogeneous loss function. This loss function categorizes the reference prediction into four groups, each assigned different weights. These weights are determined by the confidence level and the consistency between the reference and supervised predictions, allowing the model to focus on more valuable regions for effective learning from unannotated images.
  • The spatially heterogeneous loss function is extended to labeled images in the semi-supervised learning process. This extension enables the model to identify challenging regions for learning while suppressing the impact of potential noisy annotations, thereby enhancing segmentation performance.
  • A self-adaptive thresholding strategy is introduced to differentiate confident predictions from uncertain ones. This strategy adapts dynamically across different target classes and training stages, enabling more effective uncertainty-aware semi-supervised learning.

πŸ‘€ Usage

We provide code, data_split and models (Include pre-trained models and fully trained models) for LA and ACDC dataset.

To train a 2D model

python ./code/ACDC_train_CPS.py  #for ACDC training
python ./code/ACDC_train_MT.py  #for ACDC training
python ./code/ACDC_train_CCT.py  #for ACDC training
python ./code/ACDC_train_RDrop.py  #for ACDC training
python ./code/ACDC_train_Fixmatch.py  #for ACDC training

To train a 3D model

python ./code/LA_train_CPS.py  #for LA training
python ./code/LA_train_MT.py  #for LA training
python ./code/LA_train_CCT.py  #for LA training
python ./code/LA_train_RDrop.py  #for LA training
python ./code/LA_train_Fixmatch.py  #for LA training

To test a model

python ./code/test_ACDC.py  #for ACDC testing
python ./code/test_LA.py  #for LA testing

πŸ“ Dataset Download

Data could be got at:

Environment Setup

We recommend using Python 3.8+ with PyTorch 2.3.1 and MONAI 1.3+.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages