SVF-Net: spatial and visual feature enhancement network for brain structure segmentation

Qian Hu,Ying Wei,Xiang Li,Chuyuan Wang,Huan Wang,Shanze Wang
DOI: https://doi.org/10.1007/s10489-022-03706-x
IF: 5.3
2022-06-07
Applied Intelligence
Abstract:Brain structure segmentation in Magnetic Resonance Images (MRI) is essential to the assessment and treatment of medical disorders, especially neuropsychiatric diseases. The key to semantic segmentation is to understand the low-level visual semantics and the high-level spatial semantics of the image. Due to the complex anatomical structures, the current approaches lack the ability to effectively extract rich semantic information, resulting in the inevitable loss of details in prediction results. To address this problem, we propose a novel spatial and visual feature enhancement network (SVF-Net) to accurately segment the brain structures. The SVF-Net is designed as a multi-task learning framework, in which an auxiliary coarse segmentation task is used for spatial information acquisition, an auxiliary image reconstruction task is used for visual information preservation, and a major refined segmentation task is used for brain structure segmentation. In this algorithm, multitasking optimization is mainly based on two strategies: firstly, a spatial feature enhancement (SFE) module is introduced to extract the location and spatial relationships of objects from the coarse prediction, which are then sent to the refined segmentation model for spatial information enhancement. Secondly, a visual feature preservation (VFP) model is introduced for image reconstruction, which shares the feature extractor with the refined segmentation model, so as to retain more useful low-level visual features for the model. Extensive experiments are performed on three public brain MRI T1 scan datasets (the IBSR dataset, the MALC dataset and the LPBA dataset) to evaluate the effectiveness of the proposed algorithm. The experimental results show that the SVF-Net achieves the best performance compared with the state-of-the-art methods. In addition, the ablation experiments and the noise interference experiments demonstrate that proposed SFE and VFP module have obvious advantages in improving segmentation accuracy and resisting noise interference.
computer science, artificial intelligence
What problem does this paper attempt to address?