Please wait a minute...
Journal of ZheJiang University (Engineering Science)  2022, Vol. 56 Issue (10): 1958-1966    DOI: 10.3785/j.issn.1008-973X.2022.10.007
    
Image segmentation algorithm based on multi-level feature adaptive fusion
Xiao-ping YUAN(),Xiang HE,Xiao-qian WANG,Yang-ming HU
School of Information and Control Engineering , China University of Mining and Technology, Xuzhou 221116, China
Download: HTML     PDF(1738KB) HTML
Export: BibTeX | EndNote (RIS)      

Abstract  

A novel FR-Unet image segmentation algorithm based on multi-layer feature adaptive fusion was proposed in order to solve the problem of low accuracy when segmenting medical images by traditional algorithms. The sampling weighting module was designed to replace the traditional convolutional layer in the encoder stage. The image spatial information was extracted and fused by layer, and the correlation between neighboring pixels and the semantic information at different levels was obtained. A multi-layer adaptive fusion module was designed in the decoder stage in order to extract image channel information layer by layer through nonlinear jump connections and adaptively fuse the contextual information of neighboring connected layers. Then each layer focuses on the extraction of different feature information. FR-Unet was substantially reduced in the number of model parameters, allowing the network to be better supported in scene deployment. The experimental results show that the network excels in numerous tasks such as animal cell segmentation, liver organ segmentation and skin lesion segmentation.



Key wordsimage segmentation      FR-Unet      layer-by-layer extraction      sampling weighting module      multi-level adaptive fusion module     
Received: 08 November 2021      Published: 25 October 2022
CLC:  TP 391  
Fund:  国家科技支撑计划资助项目(2013BAK06B08);江苏省研究生科研与实践创新计划资助项目(SJCX20_0812)
Cite this article:

Xiao-ping YUAN,Xiang HE,Xiao-qian WANG,Yang-ming HU. Image segmentation algorithm based on multi-level feature adaptive fusion. Journal of ZheJiang University (Engineering Science), 2022, 56(10): 1958-1966.

URL:

https://www.zjujournals.com/eng/10.3785/j.issn.1008-973X.2022.10.007     OR     https://www.zjujournals.com/eng/Y2022/V56/I10/1958


基于多层级特征自适应融合的图像分割算法

为了解决传统算法对医学图像分割时精度较低的问题,提出基于多层级特征自适应融合的新型FR-Unet图像分割算法. 在编码器阶段,设计采样加权模块替代传统卷积层,对图像空间信息进行逐层提取和特征融合,获得相邻像素之间的相关性和不同层次的语义信息. 在解码器阶段,设计多层级自适应融合模块,通过非线性跳跃连接逐层提取图像通道信息,自适应地融合邻近连接层的上下文信息,使各层专注不同特征信息的提取. FR-Unet在模型参数量上大幅度减少,让网络在场景部署上得到更好的支持. 实验结果表明,该网络在动物细胞分割、肝脏器官分割、皮肤病变分割等众多任务中均表现突出.


关键词: 图像分割,  FR-Unet,  逐层提取,  采样加权模块,  多层级自适应融合模块 
Fig.1 General structure of proposed algorithm model
Fig.2 Structure of sampling weighting module for asymmetric feature extraction
Fig.3 Structure of FR-conv(1,1) convolution layer
Fig.4 Structure diagram of multi-level adaptive fusion module
Fig.5 Data enhancement diagram
Fig.6 Confusion matrix diagram
Fig.7 Comparison diagram of network parameters
方法 IOU Dice L P tr/ms
FCN 0.7421 0.8320 0.0661 0.8751 67
Unet 0.8705 0.9207 0.0637 0.9187 74
UnetPlus 0.9858 0.9879 0.0020 0.9891 141
DenseASPP 0.9862 0.9898 0.0003 0.9999 588
DeepLabV3 0.9874 0.9898 0.0002 0.9999 626
Base(Unet) 0.8705 0.9207 0.0637 0.9187 74
Base+采样加权模块 0.9584 0.9786 0.0334 0.9467 64
Base+多层级自适应融合模块 0.9650 0.9647 0.0398 0.9374 509
本文方法 0.9898 0.9882 0.0001 0.9992 288
Tab.1 Results of comparative tests and ablation studies of animal cell segmentation tasks
Fig.8 Comparison chart of segmentation of dataset 1 on different algorithms
方法 IOU Dice L P tr/ms
FCN 0.7834 0.8473 0.0611 0.8679 65
Unet 0.8388 0.8667 0.0634 0.8786 76
UnetPlus 0.9041 0.9163 0.0502 0.8994 149
DenseASPP 0.9072 0.9474 0.0405 0.9267 592
DeepLabV3 0.9541 0.9513 0.0378 0.9451 631
Base(Unet) 0.8388 0.8667 0.0634 0.8786 76
Base+采样加权模块 0.9127 0.9479 0.0527 0.9114 60
Base+多层级自适应融合模块 0.9433 0.9487 0.0401 0.9421 510
本文方法 0.9437 0.9623 0.0279 0.9536 289
Tab.2 Results of comparative trials and ablation studies of liver segmentation tasks
Fig.9 Comparison chart of segmentation of dataset 2 on different algorithms
Fig.10 Graph of objective evaluation indexes of different segmentation algorithms
Fig.11 Comparison chart of segmentation of dataset 3 on different algorithms
[1]   扈晓君, 杨宝栋. 基于深度学习的医学图像分析综述[J]电子技术与软件工程, 2021(18): 137-138.
HU Xiao-jun, YANG Bao-dong. A review of medical image analysis based on deep learning [J]. Electronic Technology and Software Engineering, 2021 (18): 137-138.
[2]   LONG J, SHELHAMER E, DARRELL T. Fully convolutional networks for semantic segmentation [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Boston: IEEE, 2015: 3431-3440.
[3]   RONNEBERGER O, FISCHER P, BROX T. U-net: convolutional networks for biomedical image segmentation [C]// International Conference on Medical Image Computing and Computer-Assisted Intervention. Cham: Springer, 2015: 234-241.
[4]   张娴, 刘玉林, 高智勇 深度学习在医学影像中的应用研究进展[J]. 临床放射学杂志, 2021, 40 (10): 2041- 2044
ZHANG Xian, LIU Yu-lin, GAO Zhi-yong Research advances in the application of deep learning in medical imaging[J]. Journal of Clinical Radiology, 2021, 40 (10): 2041- 2044
[5]   CHEN C, QIN C, QIU H, et al Deep learning for cardiac image segmentation: a review[J]. Frontiers in Cardiovascular Medicine, 2020, 7: 25
doi: 10.3389/fcvm.2020.00025
[6]   李继凡, 陈硕, 章强 基于U-Net神经网络的多模态MR颈动脉血管成像的分割方法研究[J]. 中华放射学杂志, 2019, (12): 1091- 1095
LI Ji-fan, CHEN Shuo, ZHANG Qiang Study on segmentation method of multimodal MR carotid angiography based on U-Net neural network[J]. Chinese Journal of Radiology, 2019, (12): 1091- 1095
[7]   DUAN L, XIONG X, LIU Q, et al Field rice panicle segmentation based on deep full convolutional neural network[J]. Transactions of the Chinese Society of Agricultural Engineering, 2018, 34 (12): 202- 209
[8]   ZHOU Z, SIDDIQUEE M, TAJBAKHSH N, et al. Unet++: a nested Unet architecture for medical image segmentation [M]// Deep learning in medical image analysis and multimodal learning for clinical decision support. Cham: Springer, 2018: 3-11.
[9]   TAN L, MA W, XIA J, et al Multimodal magnetic resonance image brain tumor segmentation based on ACU-net network[J]. IEEE Access, 2021, 9: 14608- 14618
doi: 10.1109/ACCESS.2021.3052514
[10]   TREBING K, STANCZYK T, MEHRKANOON S SmaAt-U-net: precipitation nowcasting using a small attention-U-net architecture[J]. Pattern Recognition Letters, 2021, (145): 275- 289
[11]   CAO K, ZHANG X An improved Res-UNet model for tree species classification using airborne high-resolution images[J]. Remote Sensing, 2020, 12 (7): 1128- 1135
doi: 10.3390/rs12071128
[12]   HE N, FANG L, PLAZA A Hybrid first and second order attention U-net for building segmentation in remote sensing images[J]. Science China Information Sciences, 2020, 63 (4): 346- 362
[13]   赵小虎, 李晓, 叶圣, 等 基于改进U-Net网络的多尺度番茄病害分割算法[J]. 计算机工程与应用, 2022, 58 (10): 216- 223
ZHAO Xiao-hu, LI Xiao, YE Sheng, et al Multiscale tomato disease segmentation algorithm based on improved U-Net network[J]. Computer Engineering and Application, 2022, 58 (10): 216- 223
doi: 10.3778/j.issn.1002-8331.2105-0201
[14]   周涛, 董雅丽, 霍兵强, 等 U-Net网络医学图像分割应用综述[J]. 中国图象图形学报, 2021, 26 (9): 2058- 2077
ZHOU Tao, DONG Ya-li, HUO Bing-qiang, et al A review of U-Net network medical image segmentation applications[J]. Journal of Image and Graphics, 2021, 26 (9): 2058- 2077
[15]   MINAEE S, BOYKOV Y Y, PORIKLI F, et al Image segmentation using deep learning: a survey[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021, 40 (7): 3523- 3542
[16]   冯婷婷, 葛华勇, 孙家慧 一种基于SSD与FRN相结合的密集连接行人检测算法[J]. 信息技术与网络安全, 2020, 39 (12): 56- 60
FENG Ting-ting, GE Hua-yong, SUN Jia-hui A densely connected pedestrian detection algorithm based on the combination of SSD and FRN[J]. Information Technology and Network Security, 2020, 39 (12): 56- 60
[17]   ARORA R, SAINI I, SOOD N. Modified UNet++ model: a deep model for automatic segmentation of lungs from chest X-ray images [C]// 2021 2nd International Conference on Secure Cyber Computing and Communications. Shanghai: IEEE, 2021: 166-169.
[1] Wan-liang WANG,Tie-jun WANG,Jia-cheng CHEN,Wen-bo YOU. Medical image segmentation method combining multi-scale and multi-head attention[J]. Journal of ZheJiang University (Engineering Science), 2022, 56(9): 1796-1805.
[2] ZHENG Zhou, ZHANG Xue-chang, ZHENG Si-ming, SHI Yue-ding. Liver segmentation in CT images based on region-growing and unified level set method[J]. Journal of ZheJiang University (Engineering Science), 2018, 52(12): 2382-2396.
[3] LIAO Miao, ZHAO Yu-qian, ZENG Ye-zhan, HUANG Zhong-chao, ZHANG Bing-kui, ZOU Bei-ji. Automatic segmentation for cell images based on support vector machine and ellipse fitting[J]. Journal of ZheJiang University (Engineering Science), 2017, 51(4): 722-728.
[4] ZHANG Jian ting,ZHANG Li min. Image segmentation using novel adaptive robust bilateral filter[J]. Journal of ZheJiang University (Engineering Science), 2016, 50(9): 1703-1710.
[5] HU Zhu-hua, ZHAO Yao-chi, CHENG Jie-ren, PENG Jin-lian. Moving object segmentation method based on improved DRLSE[J]. Journal of ZheJiang University (Engineering Science), 2014, 48(8): 1488-1495.
[6] LIU Zhong, CHEN Wei-hai, WU Xing-ming, ZOU Yu-hua, WANG Jian-hua. Salient region detection based on stereo vision[J]. Journal of ZheJiang University (Engineering Science), 2014, 48(2): 354-359.
[7] PENG Hai, ZHAO Ju-feng, FENG Hua-jun, XU Zhi-hai, LI Qi, CHEN Yue-ting. Dual band image fusion method based on region saliency[J]. Journal of ZheJiang University (Engineering Science), 2012, 46(11): 2109-2115.
[8] LI Guang-ting, YU Wei-dong. Fast implementation of SAR image segmentation using
Markov random fields
[J]. Journal of ZheJiang University (Engineering Science), 2012, 46(10): 1810-1815.
[9] WU Yi-quan, ZHANG Xiao-jie, WU Shi-hua, ZHANG Sheng-wei. Two-dimensional minimum error thresholding based on chaotic
particle swarm optimization or decomposition
[J]. Journal of ZheJiang University (Engineering Science), 2011, 45(7): 1198-1205.
[10] XIE Jiang-Jun, HOU Di-Bei, HUANG Beng-Cha, ZHANG Guang-Xin, ZHOU Ze-Kuai. Fast single parameter level set segmentation based on
semi-implicit schemes
[J]. Journal of ZheJiang University (Engineering Science), 2010, 44(8): 1496-1501.
[11] CA Jin-Hui, ZHANG Guang-Xin, CAI Hui. Theory and application of connectivity maskbased reconstruction
opening operator
[J]. Journal of ZheJiang University (Engineering Science), 2010, 44(4): 675-680.
[12] KONG Ding-ke, WANG Guo-zhao. Edge-preserving local fitting model for image segmentation[J]. Journal of ZheJiang University (Engineering Science), 2010, 44(12): 2236-2240.
[13] ZHANG Zan-Chao, JIA Shun-Ren. Automatic urinary image processing[J]. Journal of ZheJiang University (Engineering Science), 2009, 43(5): 832-838.
[14] HU Xiao-Yan, JIA Shun-Ren, LIU Tian-ming, et al. Hough transformation based analysis technique for zebrafish embryo images[J]. Journal of ZheJiang University (Engineering Science), 2009, 43(11): 2034-2037.