J Shanghai Jiaotong Univ Sci ›› 2022, Vol. 27 ›› Issue (4): 485-497.doi: 10.1007/s12204-022-2412-y

• Medicine-Engineering Interdisciplinary Research • Previous Articles     Next Articles

USSL Net: Focusing on Structural Similarity with Light U-Structure for Stroke Lesion Segmentation

JIANG Zhiguo (蒋志国), CHANG Qing∗ (常 青)   

  1. (School of Information Science and Engineering, East China University of Science and Technology, Shanghai 200237, China)
  • Received:2021-05-18 Online:2022-07-28 Published:2022-08-11

Abstract: Automatic segmentation of ischemic stroke lesions from computed tomography (CT) images is of great significance for identifying and curing this life-threatening condition. However, in addition to the problem of low image contrast, it is also challenged by the complex changes in the appearance of the stroke area and the difficulty in obtaining image data. Considering that it is difficult to obtain stroke data and labels, a data enhancement algorithm for one-shot medical image segmentation based on data augmentation using learned transformation was proposed to increase the number of data sets for more accurate segmentation. A deep convolutional neural network based algorithm for stroke lesion segmentation, called structural similarity with light U-structure (USSL) Net, was proposed. We embedded a convolution module that combines switchable normalization, multi-scale convolution and dilated convolution in the network for better segmentation performance. Besides, considering the strong structural similarity between multi-modal stroke CT images, the USSL Net uses the correlation maximized structural similarity loss (SSL) function as the loss function to learn the varying shapes of the lesions. The experimental results show that our framework has achieved results in the following aspects. First, the data obtained by adding our data enhancement algorithm is better than the data directly segmented from the multi- modal image. Second, the performance of our network model is better than that of other models for stroke segmentation tasks. Third, the way SSL functioned as a loss function is more helpful to the improvement of segmentation accuracy than the cross-entropy loss function.

Key words: structural similarity, medical image segmentation, deep convolution neural network, automatic data enhancement algorithm

CLC Number: