Towards simultaneous segmentation of liver tumors and intrahepatic vessels via cross-attention mechanism
Haopeng Kuang (Fudan University); Dingkang Yang (Fudan University); Shunli Wang (Fudan University); Xiaoying Wang (Zhongshan Hospital, Fudan University); Lihua Zhang (Fudan University)
-
SPS
IEEE Members: $11.00
Non-members: $15.00
Accurate visualization of liver tumors and their surrounding blood vessels is essential for noninvasive diagnosis and prognosis prediction of tumors. In medical image segmentation, there is still a lack of in-depth research on the simultaneous segmentation of liver tumors and peritumoral blood vessels. To this end, we collect the first liver tumor, and vessel segmentation benchmark datasets containing 52 portal vein phase computed tomography images with liver, liver tumor, and vessel annotations. In this case, we propose a 3D U-shaped Cross-Attention Network (UCA-Net) that utilizes a tailored cross-attention mechanism instead of the traditional skip connection to effectively model the encoder and decoder feature. Specifically, the UCA-Net uses a channel-wise cross-attention module to reduce the semantic gap between encoder and decoder and a slice-wise cross-attention module to enhance the contextual semantic learning ability among distinct slices. The experimental results show that our proposed UCA-Net can accurately segment 3D medical images and achieve the best liver tumor and intrahepatic blood vessel segmentation performance.