2.5D Lightweight RIU-Net for Automatic Liver and Tumor Segmentation from CT

Peiqing Lv,Jinke Wang,Haiying Wang
DOI: https://doi.org/10.1016/j.bspc.2022.103567
IF: 5.1
2022-01-01
Biomedical Signal Processing and Control
Abstract:Purpose: One critical factor that restricts the clinical application of computer-aided liver and tumor segmentation is the method's high complexity and low accuracy. Overcoming this limitation is what we are concerned about in this study. Method: This paper presented a new 2.5D lightweight network for fast and accurate liver and tumor segmentation from CT images. The method is grounded in the U-Net framework, which leverages the techniques from the residual and Inception theories. We first adopted the 2.5D training mode for CNN networks to improve the utilization of spatial information. Then, we designed an improved U-type architecture to substantially reduce the parameters by introducing residual block and InceptionV3, named RIU-Net. Finally, a hybrid loss function combined BCE and Dice is employed to speed up the convergence and improve accuracy. Result: We evaluated the proposed method on two publicly available databases, LiTS17 and 3DIRCADb. The performance of our approach is compared with five closely related techniques. Our result outperforms the others on both accuracy and time cost. Specifically, the total number of parameters is reduced by 70% compared to U Net. Conclusion: Both quantitative and qualitative results demonstrated the superior applicability of our method and thus proved to be a promising lightweight tool for computer-aided liver and tumor segmentation..
What problem does this paper attempt to address?