Multi-task Attention-Based Semi-supervised Learning for Medical Image Segmentation

Publikation: Bidrag til bog/antologi/rapportKonferencebidrag i proceedingsForskningfagfællebedømt

We propose a novel semi-supervised image segmentation method that simultaneously optimizes a supervised segmentation and an unsupervised reconstruction objectives. The reconstruction objective uses an attention mechanism that separates the reconstruction of image areas corresponding to different classes. The proposed approach was evaluated on two applications: brain tumor and white matter hyperintensities segmentation. Our method, trained on unlabeled and a small number of labeled images, outperformed supervised CNNs trained with the same number of images and CNNs pre-trained on unlabeled data. In ablation experiments, we observed that the proposed attention mechanism substantially improves segmentation performance. We explore two multi-task training strategies: joint training and alternating training. Alternating training requires fewer hyperparameters and achieves a better, more stable performance than joint training. Finally, we analyze the features learned by different methods and find that the attention mechanism helps to learn more discriminative features in the deeper layers of encoders.

OriginalsprogEngelsk
TitelMedical Image Computing and Computer Assisted Intervention – MICCAI 2019 - 22nd International Conference, Proceedings
RedaktørerDinggang Shen, Pew-Thian Yap, Tianming Liu, Terry M. Peters, Ali Khan, Lawrence H. Staib, Caroline Essert, Sean Zhou
Antal sider9
ForlagSpringer VS
Publikationsdato1 jan. 2019
Sider457-465
ISBN (Trykt)9783030322472
DOI
StatusUdgivet - 1 jan. 2019
Begivenhed22nd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2019 - Shenzhen, Kina
Varighed: 13 okt. 201917 okt. 2019

Konference

Konference22nd International Conference on Medical Image Computing and Computer-Assisted Intervention, MICCAI 2019
LandKina
ByShenzhen
Periode13/10/201917/10/2019
NavnLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Vol/bind11766 LNCS
ISSN0302-9743

Links

ID: 231952993