Advancing Medical Image Segmentation via Self-supervised Instance-adaptive Prototype Learning

Advancing Medical Image Segmentation via Self-supervised Instance-adaptive Prototype Learning

Guoyan Liang, Qin Zhou, Jingyuan Chen, Zhe Wang, Chang Yao

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence
Main Track. Pages 1056-1064. https://doi.org/10.24963/ijcai.2024/117

Medical Image Segmentation (MIS) plays a crucial role in medical therapy planning and robot navigation. Prototype learning methods in MIS focus on generating segmentation masks through pixel-to-prototype comparison. However, current approaches often overlook sample diversity by using a fixed prototype per semantic class and neglect intra-class variation within each input. In this paper, we propose to generate instance-adaptive prototypes for MIS, which integrates a common prototype proposal (CPP) capturing common visual patterns and an instance-specific prototype proposal (IPP) tailored to each input. To further account for the intra-class variation, we propose to guide the IPP generation by re-weighting the intermediate feature map according to their confidence scores. These confidence scores are hierarchically generated using a transformer decoder. Additionally we introduce a novel self-supervised filtering strategy to prioritize the foreground pixels during the training of the transformer decoder. Extensive experiments demonstrate favorable performance of our method.
Keywords:
Computer Vision: CV: Biomedical image analysis
Computer Vision: CV: Representation learning
Computer Vision: CV: Segmentation
Machine Learning: ML: Self-supervised Learning