Self-supervised learning for few-shot medical image segmentation
File(s)few_shot_segmentation_no_template.pdf (9.73 MB)
Accepted version
Author(s)
Type
Journal Article
Abstract
Fully-supervised deep learning segmentation models are inflexible when encountering new unseen semantic classes and their fine-tuning often requires significant amounts of annotated data. Few-shot semantic segmentation (FSS) aims to solve this inflexibility by learning to segment an arbitrary unseen semantically meaningful class by referring to only a few labeled examples, without involving fine-tuning. State-of-the-art FSS methods are typically designed for segmenting natural images and rely on abundant annotated data of training classes to learn image representations that generalize well to unseen testing classes. However, such a training mechanism is impractical in annotation-scarce medical imaging scenarios. To address this challenge, in this work, we propose a novel self-supervised FSS framework for medical images, named SSL-ALPNet, in order to bypass the requirement for annotations during training. The proposed method exploits superpixel-based pseudo-labels to provide supervision signals. In addition, we propose a simple yet effective adaptive local prototype pooling module which is plugged into the prototype networks to further boost segmentation accuracy. We demonstrate the general applicability of the proposed approach using three different tasks: organ segmentation of abdominal CT and MRI images respectively, and cardiac segmentation of MRI images. The proposed method yields higher Dice scores than conventional FSS methods which require manual annotations for training in our experiments.
Date Issued
2022-02-09
Date Acceptance
2022-02-02
Citation
IEEE Transactions on Medical Imaging, 2022, 41 (7), pp.1837-1848
ISSN
0278-0062
Publisher
Institute of Electrical and Electronics Engineers
Start Page
1837
End Page
1848
Journal / Book Title
IEEE Transactions on Medical Imaging
Volume
41
Issue
7
Copyright Statement
© 2021 IEEE. Personal use is permitted, but republication/redistribution requires IEEE permission. See http://www.ieee.org/publications_standards/publications/rights/index.html for more information.
Sponsor
Engineering & Physical Science Research Council (EPSRC)
Identifier
https://www.ncbi.nlm.nih.gov/pubmed/35139014
Grant Number
EP/P001009/1
Subjects
Nuclear Medicine & Medical Imaging
08 Information and Computing Sciences
09 Engineering
Publication Status
Published
Coverage Spatial
United States
Date Publish Online
2022-02-09