LACINet: A Lesion-Aware Contextual Interaction Network for Polyp Segmentation
Wenxue Li, Wei Lü, Jinghui Chu, Fugui Fan
Abstract
Automatic polyp segmentation is critical for early prevention and diagnosis of colorectal cancer. However, diverse foreground appearance and complicated background interference severely degrade the performance of pixel-level prediction. The excessive computational overheads further hinder the practical clinical applications of existing methods. In this paper, we propose a novel Lesion-Aware Contextual Interaction Network (LACINet), which aims to explore the long-range dependencies and global contexts with friendly computing resource consumption for polyp segmentation. Specifically, we present a Lesion-aware Pyramid Mechanism (LPM) to weaken the influence of background noise and refine lesion-related features. We also develop a robust Representation Enhancement Decoder (RED) to learn global feature representations and aggregate the multi-level contexts. In RED, we first build a Non-local Contextual Lesion Interaction Module (NCLIM) to integrate the cross-level contextual information for obtaining the intrinsic feature representations, and then design a Tri-branching Multi-scale Perceptual Self-attention Module (TMPSM) to sufficiently excavate the global features. Notably, we introduce an asymmetric multi-branch strategy to alleviate the computational burden. The experimental results on several widely-used benchmark datasets demonstrate the superior performance of our proposed LACINet in comparison with state-of-the-art methods.