Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2210010(2023)

A 3D Renal Tumor Image Segmentation Method Based on U2-Net

Siyuan Li, Qiang Li, and Xin Guan*
Author Affiliations
  • School of Microelectronics, Tianjin University, 300072, China
  • show less

    Renal tumors pose great harm to and seriously affect human health. Early detection and diagnosis of renal tumors can help patients' treatment and recovery. To efficiently segment kidneys and tumors from abdominal computed tomography (CT) images, this paper proposes a method based on 3D U2-Net. First, we upgrade the 2D U2-Net, adjust loss function, depth of network, and supervision strategy. To improve the feature expression ability of the decoder, we propose a residual feature enhancement module, which enhances the channel and spatial domain of the feature map at the decoder. To further improve the model's ability to extract global information, we propose a multi-head self-attention module based on global features, which calculates the long-term dependencies between all voxel points in the feature map and obtains more contextual information of 3D medical images. The method is tested on the official KiTS19 dataset and the results show that the average Dice value is 0.9008 and the parameter quantity is 4.60 MB. Compared with existing methods, our method can achieve better segmentation accuracy with small parameter quantity, and has great application value for small memory embedded system used for kidneys and tumors image segmentation.

    Tools

    Get Citation

    Copy Citation Text

    Siyuan Li, Qiang Li, Xin Guan. A 3D Renal Tumor Image Segmentation Method Based on U2-Net[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2210010

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 28, 2023

    Accepted: May. 30, 2023

    Published Online: Nov. 6, 2023

    The Author Email: Guan Xin (guanxin@tju.edu.cn)

    DOI:10.3788/LOP231203

    Topics