DSML-UNet: Depthwise separable convolution network with multiscale large kernel for medical image segmentation

被引:9
作者
Wang, Biao [1 ]
Qin, Juan [1 ]
Lv, Lianrong [1 ]
Cheng, Mengdan [1 ]
Li, Lei [1 ]
He, Junjie [1 ]
Li, Dingyao [1 ]
Xia, Dan [1 ]
Wang, Meng [1 ]
Ren, Haiping [1 ]
Wang, Shike [1 ]
机构
[1] Tianjin Univ Technol, Sch Integrated Circuit Sci & Engn, Tianjin 300384, Peoples R China
关键词
Deep learning; Medical image segmentation; Depthwise separable convolution; Large convolution kernel; Receptive fields;
D O I
10.1016/j.bspc.2024.106731
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
Computer-aided diagnosis is becoming increasingly important in modern medicine, and computer-aided diagnosis systems require accurate and automatic segmentation of medical images. Due to the unique jump connection structure, U-based convolutional neural networks (CNN) have been widely used in medical image segmentation tasks. However, the CNN-based method has the small receptive field and is not effective in learning relationships between distant features. To address this problem, a depthwise separable convolution network with multiscale large kernel for medical image segmentation (DSML-UNet) is proposed. The depthwise separable property of large kernel depthwise separable convolution increases the receptive field while adding little computational complexity. In the DSML-UNet, different sizes of large kernel depth-separable convolution modules are proposed. The different sizes of large kernel convolution provide multi-scale information and large receptive fields, which can effectively learn the rich semantic information and relationship between distant features. The proposed DSML-UNet is used in the spine dataset (SpineSagT2Wdataset3), the skin dataset (ISIC) and the lung dataset (FML) with Dice Scores of 0.8273, 0.8159, and 0.9581 respectively. The experimental results show that DSML-UNet improves the segmentation performance compared with other related advanced work.
引用
收藏
页数:9
相关论文
共 36 条
[1]   Weakly Supervised Learning using Attention gates for colon cancer histopathological image segmentation [J].
Ben Hamida, A. ;
Devanne, M. ;
Weber, J. ;
Truntzer, C. ;
Derangere, V ;
Ghiringhelli, F. ;
Forestier, G. ;
Wemmert, C. .
ARTIFICIAL INTELLIGENCE IN MEDICINE, 2022, 133
[2]  
Cao Hu, 2021, arXiv
[3]  
Chen J., 2021, arXiv, DOI DOI 10.48550/ARXIV.2102.04306
[4]  
Chollet F, 2017, Arxiv, DOI [arXiv:1610.02357, DOI 10.48550/ARXIV.1610.02357]
[5]  
Deng Y., 2021, CTSpine1K: a large-scale dataset for spinal vertebrae segmentation in computed tomography.
[6]   Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs [J].
Ding, Xiaohan ;
Zhang, Xiangyu ;
Han, Jungong ;
Ding, Guiguang .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :11953-11965
[7]  
Dosovitskiy A, 2021, Arxiv, DOI [arXiv:2010.11929, DOI 10.48550/ARXIV.2010.11929]
[8]   Multi-input Unet model based on the integrated block and the aggregation connection for MRI brain tumor segmentation [J].
Fang, Lingling ;
Wang, Xin .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2023, 79
[9]   Multi-scale contrast based skin lesion segmentation in digital images [J].
Filali, Idir ;
Belkadi, Malika .
OPTIK, 2019, 185 :794-811
[10]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778