DISENTANGLEMENT ENABLES CROSS-DOMAIN HIPPOCAMPUS SEGMENTATION

被引:2
作者
Kalkhof, John [1 ]
Gonzalez, Camila [1 ]
Mukhopadhyay, Anirban [1 ]
机构
[1] Tech Univ Darmstadt, Karolinenpl 5, D-64289 Darmstadt, Germany
来源
2022 IEEE INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING (IEEE ISBI 2022) | 2022年
关键词
feature disentanglement; domain generalisation; distribution shift;
D O I
10.1109/ISBI52829.2022.9761560
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Limited amount of labelled training data are a common problem in medical imaging. This makes it difficult to train a well-generalised model and therefore often leads to failure in unknown domains. Hippocampus segmentation from magnetic resonance imaging (MRI) scans is critical for the diagnosis and treatment of neuropsychatric disorders. Domain differences in contrast or shape can significantly affect segmentation. We address this issue by disentangling a T1weighted MRI image into its content and domain. This separation enables us to perform a domain transfer and thus convert data from new sources into the training domain. This step thus simplifies the segmentation problem, resulting in higher quality segmentations. We achieve the disentanglement with the proposed novel methodology 'Content Domain Disentanglement GAN', and we propose to retrain the UNet on the transformed outputs to deal with GAN-specific artefacts. With these changes, we are able to improve performance on unseen domains by 6-13% and outperform stateof-the-art domain transfer methods.
引用
收藏
页数:5
相关论文
共 16 条
  • [1] [Anonymous], 2017, INT C LEARN REPR
  • [2] Fast and Precise Hippocampus Segmentation Through Deep Convolutional Neural Network Ensembles and Transfer Learning
    Ataloglou, Dimitrios
    Dimou, Anastasios
    Zarpalas, Dimitrios
    Daras, Petros
    [J]. NEUROINFORMATICS, 2019, 17 (04) : 563 - 582
  • [3] Multi-task neural networks for joint hippocampus segmentation and clinical score regression
    Cao, Liang
    Li, Long
    Zheng, Jifeng
    Fan, Xin
    Yin, Feng
    Shen, Hui
    Zhang, Jun
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2018, 77 (22) : 29669 - 29686
  • [4] Chartsias A, 2019, Arxiv, DOI arXiv:1903.09467
  • [5] Semantic-Aware Generative Adversarial Nets for Unsupervised Domain Adaptation in Chest X-Ray Segmentation
    Chen, Cheng
    Dou, Qi
    Chen, Hao
    Heng, Pheng-Ann
    [J]. MACHINE LEARNING IN MEDICAL IMAGING: 9TH INTERNATIONAL WORKSHOP, MLMI 2018, 2018, 11046 : 143 - 151
  • [6] Eastwood Cian, 2018, P 6 INT C LEARN REPR
  • [7] Goodfellow I., 2014, ADV NEUR IN, V27, P2672, DOI DOI 10.1145/3422622
  • [8] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [9] Kingma D. P., 2013, ICLR 2014
  • [10] Simpson AL, 2019, Arxiv, DOI [arXiv:1902.09063, 10.48550/arxiv.1902.09063]