PVT2DNet: Polyp segmentation with vision transformer and dual decoder refinement strategy

Yibiao Hu,Yan Jin,Zhiwei Jiang,Qiufu Zheng
DOI: https://doi.org/10.1016/j.jvcir.2024.104304
IF: 2.887
2024-09-30
Journal of Visual Communication and Image Representation
Abstract:Colorectal carcinoma is a prevalent malignancy worldwide. Accurate polyp segmentation, along with endoscopic resection, can significantly reduce its incidence and mortality. Most polyp segmentation neural networks are CNN-based and single decoder strategy architectures, which learn limited robust representations. In this paper, we propose a novel network with the vision transformer and dual decoder refinement strategy called PVT2DNet to overcome some limitations of current networks and achieve more precise automated polyp segmentation. The PVT2DNet adopts a pyramid vision transformer encoder and enhances the multi-level features with the context-enhanced module (CEM). Moreover, instead of directly feeding features into a single decoder, we introduce a dual partial cascaded decoder refinement strategy to excavate more informative polyp cues. Extensive experimentations on five widely adopted datasets demonstrate the proposed network outperforms other state-of-the-art on most metrics.
computer science, information systems, software engineering
What problem does this paper attempt to address?