In recent works, more and more attention mechanisms have been used for medical image segmentation, however, attention mechanisms are not very good at distinguishing categories in multi-category medical image segmentation tasks. In this paper, we propose a category feature reconstruction module (CFRM) for multi-category pathological image segmentation of pancreatic adenosquamous carcinoma. Compared with the attention mechanism to enhance the features of the region of interest, the proposed CFRM pays more attention to the reconstruction of category features. The CFRM enables the network not only to highlight the features of the region of interest, but also to increase the discrimination of different categories of features. Based on UNet, the proposed CFRM is added at the top of the encoding path. Compared with other state-of-art methods, both the Dice coefficients and the Iou coefficients of the proposed method have reached the best level on our pancreatic adenosquamous carcinoma segmentation dataset.
|