Eres-UNet++: Liver CT image segmentation based on high-efficiency channel attention and Res-UNet+

Comput Biol Med. 2023 May:158:106501. doi: 10.1016/j.compbiomed.2022.106501. Epub 2023 Jan 10.

Abstract

Computerized tomography (CT) is of great significance for the localization and diagnosis of liver cancer. Many scholars have recently applied deep learning methods to segment CT images of liver and liver tumors. Unlike natural images, medical image segmentation is usually more challenging due to its nature. Aiming at the problem of blurry boundaries and complex gradients of liver tumor images, a deep supervision network based on the combination of high-efficiency channel attention and Res-UNet++ (ECA residual UNet++) is proposed for liver CT image segmentation, enabling fully automated end-to-end segmentation of the network. In this paper, the UNet++ structure is selected as the baseline. The residual block feature encoder based on context awareness enhances the feature extraction ability and solves the problem of deep network degradation. The introduction of an efficient attention module combines the depth of the feature map with spatial information to alleviate the uneven sample distribution impact; Use DiceLoss to replace the cross-entropy loss function to optimize network parameters. The liver and liver tumor segmentation accuracy on the LITS dataset was 95.8% and 89.3%, respectively. The results show that compared with other algorithms, the method proposed in this paper achieves a good segmentation performance, which has specific reference significance for computer-assisted diagnosis and treatment to attain fine segmentation of liver and liver tumors.

Keywords: CT image of liver; Deep learning; High-efficiency channel attention; Image segmentation; Residual module; UNet++.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms
  • Diagnosis, Computer-Assisted
  • Humans
  • Image Processing, Computer-Assisted
  • Liver Neoplasms* / diagnostic imaging
  • Tomography, X-Ray Computed