IRUS Total

Self-supervised learning for cardiac MR image segmentation by anatomical position prediction

File Description SizeFormat 
1907.02757v1.pdfWorking paper927.12 kBAdobe PDFView/Open
Title: Self-supervised learning for cardiac MR image segmentation by anatomical position prediction
Authors: Bai, W
Chen, C
Tarroni, G
Duan, J
Guitton, F
Petersen, SE
Guo, Y
Matthews, PM
Rueckert, D
Item Type: Working Paper
Abstract: In the recent years, convolutional neural networks have transformed the field of medical image analysis due to their capacity to learn discriminative image features for a variety of classification and regression tasks. However, successfully learning these features requires a large amount of manually annotated data, which is expensive to acquire and limited by the available resources of expert image analysts. Therefore, unsupervised, weakly-supervised and self-supervised feature learning techniques receive a lot of attention, which aim to utilise the vast amount of available data, while at the same time avoid or substantially reduce the effort of manual annotation. In this paper, we propose a novel way for training a cardiac MR image segmentation network, in which features are learnt in a self-supervised manner by predicting anatomical positions. The anatomical positions serve as a supervisory signal and do not require extra manual annotation. We demonstrate that this seemingly simple task provides a strong signal for feature learning and with self-supervised learning, we achieve a high segmentation accuracy that is better than or comparable to a U-net trained from scratch, especially at a small data setting. When only five annotated subjects are available, the proposed method improves the mean Dice metric from 0.811 to 0.852 for short-axis image segmentation, compared to the baseline U-net.
Issue Date: 5-Jul-2019
URI: http://hdl.handle.net/10044/1/73652
Publisher: arXiv
Copyright Statement: © 2019 The Author(s)
Keywords: cs.CV
Notes: Accepted by MICCAI 2019
Publication Status: Published
Appears in Collections:Computing
Department of Brain Sciences