DCATNet: polyp segmentation with deformable convolution and contextual-aware attention network.

Journal: BMC medical imaging
PMID:

Abstract

Polyp segmentation is crucial in computer-aided diagnosis but remains challenging due to the complexity of medical images and anatomical variations. Current state-of-the-art methods struggle with accurate polyp segmentation due to the variability in size, shape, and texture. These factors make boundary detection challenging, often resulting in incomplete or inaccurate segmentation. To address these challenges, we propose DCATNet, a novel deep learning architecture specifically designed for polyp segmentation. DCATNet is a U-shaped network that combines ResNetV2-50 as an encoder for capturing local features and a Transformer for modeling long-range dependencies. It integrates three key components: the Geometry Attention Module (GAM), the Contextual Attention Gate (CAG), and the Multi-scale Feature Extraction (MSFE) block. We evaluated DCATNet on five public datasets. On Kvasir-SEG and CVC-ClinicDB, the model achieved mean dice scores of 0.9351 and 0.9444, respectively, outperforming previous state-of-the-art (SOTA) methods. Cross-validation further demonstrated its superior generalization capability. Ablation studies confirmed the effectiveness of each component in DCATNet. Integrating GAM, CAG, and MSFE effectively improves feature representation and fusion, leading to precise and reliable segmentation results. These findings underscore DCATNet's potential for clinical application and can be used for a wide range of medical image segmentation tasks.

Authors

  • Zenan Wang
  • Tianshu Li
    School of Integrated Circuits, Shandong University, 1500 Shunhua Road, Jinan, 250101, Shandong, China.
  • Ming Liu
    School of Land Engineering, Chang'an University, Xi'an 710064, China; Xi'an Key Laboratory of Territorial Spatial Information, School of Land Engineering, Chang'an University, Xi'an 710064, China. Electronic address: mingliu@chd.edu.cn.
  • Jue Jiang
    Department of Medical Physics, Memorial Sloan-Kettering Cancer Center, New York, NY, 10065, USA.
  • Xinjuan Liu
    Department of Gastroenterology, Beijing Chaoyang Hospital, The Third Clinical Medical College of Capital Medical University, Beijing, China. liuxinjuan@mail.ccmu.edu.cn.