CSUnet: a dual attention and hybrid convolutional network for polyp segmentation

被引:0
作者
Liu, Shangwang [1 ,2 ]
Si, Feiyan [1 ]
Lin, Yinghai [1 ]
机构
[1] Henan Normal Univ, Sch Comp & Informat Engn, 46 Jianshe East Rd, Xinxiang 453000, Henan, Peoples R China
[2] Henan Normal Univ, Engn Lab Intelligence Business Internet Things, 46 Jianshe East Rd, Xinxiang 453000, Henan, Peoples R China
关键词
Deep learning; Convolutional network; Attention mechanism; Polyp segmentation;
D O I
10.1007/s11760-024-03485-7
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Medical polyp segmentation is a key task in the field of computer vision and medical image analysis, which aims to automatically detect and segment polyps in medical images for early cancer screening and treatment planning. Neural networks, in particular convolutional neural networks (CNNS), play a key role in this task. We present CSUNet, a traditional U-shaped encoder-decoder network architecture specifically designed for polyp segmentation. Firstly, the hybrid convolution mechanism was used to effectively capture the features of different scales and improve the segmentation performance. In addition, an attention module is added to the skip connection to facilitate efficient feature map extraction and fusion. We trained and verified the network with two public datasets: Kvasir-SEG and CVC-ClinicDB. In the polyp segmentation task, the Mean Intersection over Union value of CSUNet on these datasets reached 87.78% and 92.07%, respectively. The effectiveness of the proposed method enables the model to show strong segmentation performance and excellent visual results. Experiments on the collected public large-scale datasets are better than other state-of-the-art methods.
引用
收藏
页码:8445 / 8456
页数:12
相关论文
共 41 条
[11]  
He K., 2019, P IEEE INT C COMP VI, P1026
[12]  
Hou J., 2020, IEEE ACCESS, V8, P139284
[13]   Coordinate Attention for Efficient Mobile Network Design [J].
Hou, Qibin ;
Zhou, Daquan ;
Feng, Jiashi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13708-13717
[14]  
Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/CVPR.2018.00745, 10.1109/TPAMI.2019.2913372]
[15]  
Huang HM, 2020, INT CONF ACOUST SPEE, P1055, DOI [10.1109/icassp40776.2020.9053405, 10.1109/ICASSP40776.2020.9053405]
[16]  
Huang J., 2022, IEEE T MED IMAGING, V41, P2609
[17]   MISSFormer: An Effective Transformer for 2D Medical Image Segmentation [J].
Huang, Xiaohong ;
Deng, Zhifang ;
Li, Dandan ;
Yuan, Xueguang ;
Fu, Ying .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2023, 42 (05) :1484-1494
[18]  
Jha D., 2019, C MULT MOD
[19]   A ConvNet for the 2020s [J].
Liu, Zhuang ;
Mao, Hanzi ;
Wu, Chao-Yuan ;
Feichtenhofer, Christoph ;
Darrell, Trevor ;
Xie, Saining .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :11966-11976
[20]  
Liu ZY, 2021, AAAI CONF ARTIF INTE, V35, P2233