Unpaired multi-modal segmentation via knowledge distillation
File(s)08979396.pdf (21.02 MB)
Published version
OA Location
Author(s)
Dou, Q
Liu, Quande
Heng, Pheng Ann
Glocker, Benjamin
Type
Journal Article
Abstract
Multi-modal learning is typically performed with network architectures containing modality-specific layers and shared layers, utilizing co-registered images of different modalities. We propose a novel learning scheme for unpaired cross-modality image segmentation, with a highly compact architecture achieving superior segmentation accuracy. In our method, we heavily reuse network parameters, by sharing all convolutional kernels across CT and MRI, and only employ modality-specific internal normalization layers which compute respective statistics. To effectively train such a highly compact model, we introduce a novel loss term inspired by knowledge distillation, by explicitly constraining the KL-divergence of our derived prediction distributions between modalities. We have extensively validated our approach on two multi-class segmentation problems: i) cardiac structure segmentation, and ii) abdominal organ segmentation. Different network settings, i.e., 2D dilated network and 3D U-net, are utilized to investigate our method's general efficacy. Experimental results on both tasks demonstrate that our novel multi-modal learning scheme consistently outperforms single-modal training and previous multi-modal approaches.
Date Issued
2020-07-01
Date Acceptance
2019-12-24
Citation
IEEE Transactions on Medical Imaging, 2020, 39 (7), pp.2415-2425
ISSN
0278-0062
Publisher
Institute of Electrical and Electronics Engineers
Start Page
2415
End Page
2425
Journal / Book Title
IEEE Transactions on Medical Imaging
Volume
39
Issue
7
Copyright Statement
© 2019 IEEE. This work is licensed under a Creative Commons Attribution 4.0 License. For more information, see https://creativecommons.org/licenses/by/4.0/.
Sponsor
Commission of the European Communities
Identifier
https://ieeexplore.ieee.org/document/8979396
Grant Number
H2020 - 757173
Subjects
cs.CV
cs.CV
eess.IV
08 Information and Computing Sciences
09 Engineering
Nuclear Medicine & Medical Imaging
Publication Status
Published
Date Publish Online
2020-02-03