Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

CrossMatch: Enhance Semi-Supervised Medical Image Segmentation with Perturbation Strategies and Knowledge Distillation

About

Semi-supervised learning for medical image segmentation presents a unique challenge of efficiently using limited labeled data while leveraging abundant unlabeled data. Despite advancements, existing methods often do not fully exploit the potential of the unlabeled data for enhancing model robustness and accuracy. In this paper, we introduce CrossMatch, a novel framework that integrates knowledge distillation with dual perturbation strategies-image-level and feature-level-to improve the model's learning from both labeled and unlabeled data. CrossMatch employs multiple encoders and decoders to generate diverse data streams, which undergo self-knowledge distillation to enhance consistency and reliability of predictions across varied perturbations. Our method significantly surpasses other state-of-the-art techniques in standard benchmarks by effectively minimizing the gap between training on labeled and unlabeled data and improving edge accuracy and generalization in medical image segmentation. The efficacy of CrossMatch is demonstrated through extensive experimental validations, showing remarkable performance improvements without increasing computational costs. Code for this implementation is made available at https://github.com/AiEson/CrossMatch.git.

Bin Zhao, Chunshi Wang, Shuxue Ding• 2024

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationLA
Dice91.61
97
Medical Image SegmentationACDC 10% labeled (test)
Dice85.26
40
Medical Image SegmentationACDC 5% labeled (test)
Dice0.8827
30
Medical Image SegmentationACDC (5% labeled)
DICE9.8
29
Pancreas SegmentationPancreas-CT (20% labeled, 80% unlabeled)
Dice0.8313
20
Cardiac SegmentationACDC 10% labeled scans
Dice89.08
19
3D Medical Image SegmentationPancreas 1.0 (10% labeled data)
Dice Coefficient0.7969
17
Medical Image SegmentationFIVES 14 (5%)
DICE62.24
9
Medical Image SegmentationFIVES 14 (10%)
DICE60.68
9
Skin Lesion SegmentationISIC 2018 (5% labeled)
Dice84.1
6
Showing 10 of 15 rows

Other info

Code

Follow for update