A novel medical image segmentation network for colorectal polyp small targets and fuzzy boundaries.
Authors
Affiliations (1)
Affiliations (1)
- School of Mechanical Electronic and Vehicle Engineering, Beijing University of Civil Engineering and Architecture, Beijing, China.
Abstract
To address the challenges of complex feature variations and unclear boundary definitions between segmented targets and surrounding regions in medical images, a novel segmentation model based on Deformable Large Kernel Convolutional Attention (D-LKA) and Transformer is proposed. The model first uses Vision Transformer as the encoder to enhance the ability to capture global information, overcoming the limitations of convolutional neural networks' receptive field. In the decoder, a D-LKA decoder with deformable large kernel convolution attention is used, allowing the model to adapt to complex target features. Finally, the TRR module is introduced to coordinate information transfer between the convolutional neural network and Transformer, reducing semantic loss. The model is trained, validated, and tested on the Kvasir-SEG colon polyp dataset, with multiple ablation experiments. To validate generalization, experiments are also conducted on the CVC-ClinicDB dataset. Experimental results show that TDU-Net outperforms other methods in both segmentation accuracy and generalization. TDU-Net achieves excellent segmentation results, addressing small target and incomplete feature extraction issues, significantly improving clinical diagnosis efficiency and accuracy.