Slim U-Net: Efficient Anatomical Feature Preserving U-net Architecture for Ultrasound Image Segmentation

Deepak Raina,Kashish Verma,SH Chandrashekhara,Subir Kumar Saha
DOI: https://doi.org/10.1145/3574198.3574205
2023-02-23
Abstract:We investigate the applicability of U-Net based models for segmenting Urinary Bladder (UB) in male pelvic view UltraSound (US) images. The segmentation of UB in the US image aids radiologists in diagnosing the UB. However, UB in US images has arbitrary shapes, indistinct boundaries and considerably large inter- and intra-subject variability, making segmentation a quite challenging task. Our study of the state-of-the-art (SOTA) segmentation network, U-Net, for the problem reveals that it often fails to capture the salient characteristics of UB due to the varying shape and scales of anatomy in the noisy US image. Also, U-net has an excessive number of trainable parameters, reporting poor computational efficiency during training. We propose a Slim U-Net to address the challenges of UB segmentation. Slim U-Net proposes to efficiently preserve the salient features of UB by reshaping the structure of U-Net using a less number of 2D convolution layers in the contracting path, in order to preserve and impose them on expanding path. To effectively distinguish the blurred boundaries, we propose a novel annotation methodology, which includes the background area of the image at the boundary of a marked region of interest (RoI), thereby steering the model's attention towards boundaries. In addition, we suggested a combination of loss functions for network training in the complex segmentation of UB. The experimental results demonstrate that Slim U-net is statistically superior to U-net for UB segmentation. The Slim U-net further decreases the number of trainable parameters and training time by 54% and 57.7%, respectively, compared to the standard U-Net, without compromising the segmentation accuracy.
Image and Video Processing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to segment the urinary bladder (UB) in the ultrasound images of male pelvic views. Specifically, the author points out the following challenges: 1. **Complexity of Anatomical Structures**: In ultrasound images, the shape and size of the urinary bladder vary greatly, the boundaries are blurred, and there are large inter - individual and intra - individual differences, which makes the segmentation task very challenging. 2. **Limitations of Existing Models**: The traditional U - Net model often fails to capture the key features of the urinary bladder when dealing with these complex ultrasound images, especially when facing anatomical structures of different shapes and scales. In addition, the U - Net model has too many parameters, resulting in low computational efficiency. To solve these problems, the author proposes the Slim U - Net model and makes the following improvements: 1. **Simplifying the Network Structure**: By reducing the number of 2D convolutional layers in the contraction path, Slim U - Net can more effectively retain the simple features extracted by the initial convolution and reuse these features in the expansion path. This helps to avoid feature complexity caused by speckle noise, thereby improving the segmentation effect. 2. **New Labeling Method**: Considering the importance of the division of anatomical structure boundaries for the segmentation model, the author proposes a new labeling method, that is, including the background area at the boundaries of the marked area to guide the model to focus on the urinary bladder boundaries. 3. **Combination of Loss Functions**: In order to better extract the fine - grained features of the urinary bladder, the author suggests using a combination of multiple loss functions for network training, including Binary Cross - Entropy (BCE), Dice Coefficient (DC) and Jaccard Index (JI). The specific forms of the loss function combinations are as follows: \[ L_{D}=L_{DC} \] \[ L_{DJ}=L_{DC}+L_{JI} \] \[ L_{DJB}=L_{DC}+L_{JI}+L_{BCE} \] The experimental results show that Slim U - Net performs excellently in the urinary bladder segmentation task. It not only outperforms the standard U - Net in segmentation accuracy, but also reduces 54% of the trainable parameters and 57.7% of the training time while maintaining high segmentation precision. In conclusion, this paper aims to solve the challenges faced by urinary bladder segmentation in ultrasound images and improve the segmentation accuracy and computational efficiency by improving the U - Net model and its related technologies.