AtlasSeg: Atlas Prior Guided Dual-U-Net for Cortical Segmentation in Fetal Brain MRI

Haoan Xu,Tianshu Zheng,Xinyi Xu,Yao Shen,Jiwei Sun,Cong Sun,Guangbin Wang,Dan Wu
2024-11-05
Abstract:Accurate tissue segmentation in fetal brain MRI remains challenging due to the dynamically changing anatomical anatomy and contrast during fetal development. To enhance segmentation accuracy throughout gestation, we introduced AtlasSeg, a dual-U-shape convolution network incorporating gestational age (GA) specific information as guidance. By providing a publicly available fetal brain atlas with segmentation label at the corresponding GA, AtlasSeg effectively extracted the contextual features of age-specific patterns in atlas branch and generated tissue segmentation in segmentation branch. Multi-scale attentive atlas feature fusions were constructed in all stages during encoding and decoding, giving rise to a dual-U-shape network to assist feature flow and information interactions between two branches. AtlasSeg outperformed six well-known segmentation networks in both our internal fetal brain MRI dataset and the external FeTA dataset. Ablation experiments demonstrate the efficiency of atlas guidance and the attention mechanism. The proposed AtlasSeg demonstrated superior segmentation performance against other convolution networks with higher segmentation accuracy, and may facilitate fetal brain MRI analysis in large-scale fetal brain studies.
Image and Video Processing,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
This paper attempts to address the challenges in fetal brain MRI tissue segmentation, especially the problem of low segmentation accuracy caused by the dynamic changes of anatomical structures and contrast during fetal development. The authors propose a dual - U - shaped convolutional network named AtlasSeg, which improves segmentation accuracy by introducing information of specific gestational age (GA) as guidance. Specifically, AtlasSeg utilizes publicly available fetal brain atlases corresponding to specific gestational ages and their segmentation labels, effectively extracts the context features of age - specific patterns, and generates tissue segmentation results. By constructing multi - scale attention atlas feature fusion at all stages of encoding and decoding, a dual - U - shaped network is formed, which facilitates feature flow and information interaction between the two branches. Experimental results show that AtlasSeg outperforms six well - known segmentation networks on both the internal fetal brain MRI dataset and the external FeTA dataset, especially in terms of segmentation accuracy.