Bui TD, Wang L, Chen J, Lin W, Li G, Shen D. Multi-task Learning for
Neonatal Brain Segmentation Using 3D Dense-Unet with Dense Attention Guided by Geodesic Distance.
Domain Adapt Represent Transf Med Image Learn Less Labels Imperfect Data (2019) 2019;
11795:243-251. [PMID:
32090208 PMCID:
PMC7034948 DOI:
10.1007/978-3-030-33391-1_28]
[Citation(s) in RCA: 8] [Impact Index Per Article: 1.6] [Reference Citation Analysis] [What about the content of this article? (0)] [Affiliation(s)] [Abstract] [Key Words] [Grants] [Track Full Text] [Subscribe] [Scholar Register] [Indexed: 12/13/2022]
Abstract
The deep convolutional neural network has achieved outstanding performance on neonatal brain MRI tissue segmentation. However, it may fail to produce reasonable results on unseen datasets that have different imaging appearance distributions with the training data. The main reason is that deep learning models tend to have a good fitting to the training dataset, but do not lead to a good generalization on the unseen datasets. To address this problem, we propose a multi-task learning method, which simultaneously learns both tissue segmentation and geodesic distance regression to regularize a shared encoder network. Furthermore, a dense attention gate is explored to force the network to learn rich contextual information. By using three neonatal brain datasets with different imaging protocols from different scanners, our experimental results demonstrate superior performance of our proposed method over the existing deep learning-based methods on the unseen datasets.
Collapse