Annular Prior Prompt Learning for Medical Images Segmentation.

Abstract

Excellent performance has been achieved on medical image segmentation. Still, existing algorithms perform relatively poorly for annular objects with high intra-class variability and inter-class similarity, which easily leads to regional confusion, especially in medical images with annular regions. In this paper, we first propose an annular prior prompt learning method based on MedSAM, coined APPL, which not only can strengthen the learning for annular region features, but also effectively handles the regional confusion caused by high intra-class variability and inter-class similarity. Specifically, a novel annular prior prompt encoder (APPE) is proposed based on our designed annular constraint to alleviate the negative impact of inter-class similarity, which maps multiple prompt points into a linear regression feature space to provide the standardized annular prompt feature for the mask decoder. In addition, a new region connectivity enhanced image encoder (RCEIE) is developed to introduce morphology attention into supervised training, which reduces feature noise caused by intra-class variability, significantly improving the connectivity of region features. Powered by the collaboration across different annular features, the mask decoder can effectively avoid inter-class confusion to reduce the impact of weak boundaries, further boosting the performance for tackling highly diverse pixel distributions in medical images. Compared with most state-of-the-art methods, the proposed method achieves 90.07% of the mIoU and 94.71% of the DSC on an in-house dataset, and shows strong generalization on two public heart segmentation datasets with end-diastolic DSC scores of 94.2% and 83.9%, and end-systolic DSC scores of 95.5% and 87.3%. The experimental results demonstrate the consistently superior performance of our method quantitatively and qualitatively.