Contextual-wise Discriminative Feature Extraction and Robust Network Learning for Subcortical Structure Segmentation

Li Xiang,Wei Ying,Wang Chuyuan,Hu Qian,Liu Chengan
DOI: https://doi.org/10.1007/s10489-022-03848-y
IF: 5.3
2022-01-01
Applied Intelligence
Abstract:Robust and accurate segmentation of subcortical structures in MR images is difficult due to: (1) low image contrast and spatial resolution, (2) ambiguous boundaries and large appearance variances, and (3) corrupted training sample annotations. In this paper, we propose a novel CNN architecture to address the above problem from two aspects: increasing the discriminating ability of image feature representations and alleviating the influence of incorrect annotations. Specifically, we first propose the contextual-wise multi-scale feature aggregation (C-MSFA) module to extract multi-scale context. In contrast to the existing methods, the C-MSFA module aggregates the contextual information of each subcortical structure by fusing different scales of encoder features on the corresponding soft regions. Moreover, the shifted-window strategy is used to keep detailed information. Then we propose the Transformer-like decoder feature recalibration (TDFR) module to obtain discriminative decoder feature representations by learning the feature context descriptors through the cross-attention between the decoder features and the contextual-wise multi-scale features, which are used to refine the decoder features in a channel recalibration manner. Finally, we propose a novel online meta-mask learning method using a meta-mask branch to evaluate the influence of training pixels and generate a binary meta-mask to exclude unfavorable pixels and labels. The proposed method is evaluated on two benchmark datasets (the IBSR dataset and the MALC dataset). The experimental results show that our method has better performance than several state-of-the-art medical image segmentation networks and subcortical structure segmentation methods.
What problem does this paper attempt to address?