An Implicit-Explicit Prototypical Alignment Framework for Semi-Supervised Medical Image Segmentation

Area of Science:

• Medical Image Analysis
• Computer Vision
• Machine Learning

Background:

• Semi-supervised learning (SSL) is crucial for medical image segmentation due to limited pixel-level annotations.
• Existing SSL methods, like consistency learning, face challenges with efficiency and stability from inaccurate supervision and poor feature representation.
• Prototypical learning offers potential for feature aggregation but requires further exploration in SSL for enhanced supervision and representation.

Purpose of the Study:

• To propose an Implicit-Explicit Prototype Alignment (IEPAlign) framework to enhance semi-supervised consistency training for medical image segmentation.
• To improve supervision quality and feature representation by leveraging prototypical learning within an SSL context.
• To address the limitations of current methods in efficient and stable semi-supervised medical image segmentation.

Main Methods:

• Developed an implicit prototype alignment using dynamic, on-the-fly multiple prototypes.
• Implemented a multiple prediction voting strategy for reliable unlabeled mask generation and prototype calculation.
• Introduced region-aware hierarchical prototype alignment to boost intra-class consistency and inter-class separability of pixel-wise features.

Main Results:

• The proposed IEPAlign framework significantly improves semi-supervised medical image segmentation.
• IEPAlign demonstrates superior performance compared to other popular semi-supervised segmentation methods.
• The method achieves performance comparable to fully-supervised training approaches on multiple medical image segmentation tasks.

Conclusions:

• IEPAlign effectively enhances supervision quality and feature representation in semi-supervised medical image segmentation.
• The framework offers a robust and efficient solution for medical image segmentation with limited labeled data.
• IEPAlign represents a significant advancement in semi-supervised learning for medical imaging applications.