Hybrid-attention densely connected U-Net with GAP for extracting livers from CT volumes

Liver segmentation is an important step in the clinical treatment of liver cancer, and accurate and automatic liver segmentation methods are extremely important. U-Net has been used as the benchmark for many medical segmentation networks, but it cannot fully utilize low-resolution information and global contextual information. To solve these problems, we propose a new network architecture named the hybrid-attention densely connected U-Net (HDU-Net).

The proposed HDU-Net has three main changes relative to U-Net, as follows: (1) It uses a densely connected structure and dilated convolution to achieve feature reuse and avoid information loss. (2) A global average pooling block is proposed to further augment the receptive field and improve the segmentation accuracy of the network for small or disconnected liver regions. (3) By combining the spatial attention and channel attention mechanisms, a hybrid attention structure is proposed to replace the skip connection component to filter and integrate low-resolution information.

Experiments conducted on the LITS2017, 3Dircadb and Sliver07 datasets show that the proposed model can segment the liver accurately and effectively. Dice scores reach 96.5%, 96.18%, and 97.57% on these datasets, respectively, constituting results that are superior to many previously proposed methods.

The experimental liver segmentation results have demonstrated that our proposed network provides improved segmentation performance in comparison with other networks. The experimental results without postprocessing confirmed that our network solves the oversegmentation and undersegmentation problems to some extent. The proposed model is effective, robust, and efficient in terms of liver segmentation without requiring extensive training time or a very large dataset.