SPENet: Self-guided Prototype Enhancement Network for Few-shot Medical Image Segmentation
This work improves segmentation accuracy for medical imaging tasks with limited labeled data, representing an incremental advancement in few-shot learning methods.
The paper tackles the problem of few-shot medical image segmentation by addressing intra-class variations in prototype-based methods, proposing SPENet with multi-level prototype generation and query-guided enhancement, and reports superior performance on three public datasets.
Few-Shot Medical Image Segmentation (FSMIS) aims to segment novel classes of medical objects using only a few labeled images. Prototype-based methods have made significant progress in addressing FSMIS. However, they typically generate a single global prototype for the support image to match with the query image, overlooking intra-class variations. To address this issue, we propose a Self-guided Prototype Enhancement Network (SPENet). Specifically, we introduce a Multi-level Prototype Generation (MPG) module, which enables multi-granularity measurement between the support and query images by simultaneously generating a global prototype and an adaptive number of local prototypes. Additionally, we observe that not all local prototypes in the support image are beneficial for matching, especially when there are substantial discrepancies between the support and query images. To alleviate this issue, we propose a Query-guided Local Prototype Enhancement (QLPE) module, which adaptively refines support prototypes by incorporating guidance from the query image, thus mitigating the negative effects of such discrepancies. Extensive experiments on three public medical datasets demonstrate that SPENet outperforms existing state-of-the-art methods, achieving superior performance.