Target-aware U-Net with fuzzy skip connections for refined pancreas segmentation

Yufei Chen,Chang Xu,Weiping Ding,Shichen Sun,Xiaodong Yue,Hamido Fujita
DOI: https://doi.org/10.1016/j.asoc.2022.109818
2022-01-01
Abstract:Medical image segmentation is one of the important steps in the computer-aided diagnosis of pancreas diseases. Although some models have been proposed to deal with the task of automatic pancreas segmentation, it is still challenging due to the small size, variable shape and unclear boundary of pancreas. In this paper, we propose a target-aware U-Net (tU-Net) using fuzzy skip connection for pancreas segmentation. Through adding a fuzzy skip connection module into the U-Net architecture, the low-level features can be transformed into the high-level semantic features, which facilitates the segmentation of small and changeable targets of pancreas. Based on the fuzzy feature mapping, we also design a target attention mechanism consists of global average pooling and depthwise convolution. It makes the decoder of the network more sensitive to target features by increasing weights of important channels. The proposed method is evaluated on the NIH dataset of 82 CT volumes, and the pancreas Medical Segmentation Decathlon (MSD) challenge dataset of 281 CT volumes. The proposed model achieves better results comparing with other state-of-the-art models.(c) 2022 Elsevier B.V. All rights reserved.
What problem does this paper attempt to address?