Low-confidence Heterogeneous Pixel-Prototype Contrastive Learning for Semi-supervised Medical Image Segmentation
Abstract: Medical image segmentation is a challenging task especially when dealing with unlabeled data. It has been proven that the use of complementary information for co-training is effective for medical image segmentation. We find that high-confidence regions are easy to segment, but low-confidence regions are difficult to correctly segment, which results in poor segmentation performance. Therefore, accurate segmentation in low-confidence regions is the effective way to enhance the performance of medical image segmentation. To achieve this goal, we propose a novel co-training strategy named low-confidence heterogeneous pixel-prototype contrastive learning with uncertainty-guide cross supervision for semi-supervised medical image segmentation. Specifically, an uncertainty-guided cross supervision module is firstly designed to estimate the confidence score of predictions from multiple outputs and then perform cross supervision between high-confidence regions. Moreover, we design a low-confidence heterogeneous prototype contrastive learning, which builds the relationship between low-confidence pixel and heterogeneous prototype to mine discriminative information in low-confidence regions so as to improve the segmentation performance. Extensive experiments on medical image datasets demonstrate that our method outperforms state-of-the-art methods in segmentation results.
Loading