Brief Review — RCU-Net: Tumor Segmentation in Breast Ultrasound Image
RCU-Net, Improve Skip Connections Using Residual Blocks and Dense Block
Tumor Segmentation in Breast Ultrasound Image by Means of Res Path Combined with Dense Connection Neural Network,
RCU-Net, by University of Shanghai for Science and Technology,
2021 MDPI J. Diagnostics, Over 5 Citations (Sik-Ho Tsang @ Medium)
Medical Imaging, Medical Image Analysis, Image Segmentation, U-Net
- An improved U-Net, namely RCU-Net, is proposed in this paper:
- First, Res Path was introduced into the U-Net to reduce the difference between the feature maps of the encoder and decoder.
- Then, a new connection, dense block, originated from DenseNet, is introduced at the top level of U-Net as well, was added to reduce the feature information loss and alleviate the vanishing gradient problem.
- The Contrast Limited Adaptive Histogram Equalization (CLAHE) is used to accentuates the contrast between the tumor and surrounding tissue.
After CLAHE, the range of gray values is much larger, which is helpful for deep learning model training.
1.2. Overall Architecture
Res Path is proposed at the skip connections of U-Net.
Also, Dense block, originated from DenseNet, is proposed to add the input to the near end of network.
1.3. Res Path
- And there is 2 Res Blocks at the above example.
1.4. Dense Block
A new connection is introduced that incorporates a dense block from the input of the feature maps in the encoding-to-decoding component.
2.1. Ablation Study
- 10-fold cross validation is used for evaluation.
Compared with that of the U-Net, the TP, JS, and DC of the proposed model improved by 1.0625%, 2.135%, and 2.0105%, respectively. The HD index decreased by 1.41023.
U-Net had lesser number of pixels false predicted due to Res Path.
2.2. SOTA Comparisons
Additionally, the DIU-Net had close performance to the proposed model due to the deeper layer. However, excessive layers may lead to too many parameters, making the model more difficult and slower to train.
Though the model performed better than others, there were still some failed segmentation cases owing to the difficulty of extracting features in some images.