Abstract
In recent years, pulmonary vessel segmentation has aroused widespread interest in medical image analysis. However, the structure of pulmonary vessels is complex and CT images have a lot of noise. Therefore, it is a difficult task to extract pulmonary vessels accurately. In terms of pulmonary vessel segmentation, medical image segmentation methods based on deep learning only utilize single volume data in CT image, but cannot fully fuse multiple volume data, resulting in the accuracy of pulmonary vessel segmentation is low. In order to fully utilize the complementary advantages of multiple volume data, we propose a DS-ResUNet to segment pulmonary vessels in multi-view. The DS-ResUNet uses feature fusion module to fuse 2 mm and 5 mm volume data and makes full use of the detailed vessel textures in 2 mm volume data and coarse vessel structures in 5 mm volume data. By this multi-view fusion module, the accuracy of pulmonary vessel segmentation can be effectively improved. In addition, in order to strengthen the spatial weight of vessels and reduce the model parameters, we design a lightweight 3D axial attention module by separable convolution. To confirm the improved performance, we design some comparison experiments with the state-of-the-art segmentation methods. As a result, our DS-ResUNet has a better performance than other state-of-the-art methods on pulmonary vessel segmentation, but also has fewer parameters.
Similar content being viewed by others
References
Sookpotharom and Supot. The segmentation of dermoscopy images by k-means thresholding
Osuna-Enciso, V., Cuevas, E., Sossa, H.: A comparison of nature inspired algorithms for multi-threshold image segmentation. Expert Syst. Appl. 40(4), 1213–1219 (2013)
Rouhi, R., Jafari, M., Kasaei, S., Keshavarzian, P.: Benign and malignant breast tumors classification based on region growing and CNN segmentation. Expert Syst. Appl. Int. J. 42(3), 990–1002 (2015)
Bertolotto, M., Vo, A.-V., Truong-Hong, L., Laefer, D.F.: Octree-based region growing for point cloud segmentation. ISPRS J. Photogramm. Remote Sens. 104(Jun), 88–100 (2015)
Zhao, Y.Q., Wang, X.H., Wang, X.F., Shih, F.Y.: Retinal vessels segmentation based on level set and region growing. Pattern Recognit. 47(7), 2437–2446 (2014)
Shen T., Wang, Y.: Medical image segmentation based on improved watershed algorithm. In: 2018 IEEE 3rd Advanced Information Technology, Electronic and Automation Control Conference (IAEAC), (2018)
Long, J., Shelhamer, E., Darrell, T.: Fully convolutional networks for semantic segmentation. IEEE Trans. Pattern Anal. Mach. Intell. 39(4), 640–651 (2015)
Ronneberger, O., Fischer, P., Brox, T.: U-net: Convolutional Networks for Biomedical Image Segmentation. Springer, Cham (2015)
Lizarazo, I.: SVM-based segmentation and classification of remotely sensed data. Int. J. Remote Sens. 29(23–24), 7277–7283 (2008)
Park, S., Lee, H.S., Kim, J.: Seed growing for interactive image segmentation using SVM classification with geodesic distance. Electron. Lett. 53(1), 22–24 (2016)
Alush, Amir: Goldberger, Jacob: Hierarchical image segmentation using correlation clustering. IEEE Trans. Neural Netw. Learn. Syst. 27(6), 1358 (2016)
Burney, S.A., Tariq, H.: K-means cluster analysis for image segmentation. Int. J. Comput. Appl. 96(4), 1–8 (2014)
He, B., Huang, C., Sharp, G., Zhou, S., Hu, Q., Fang, C., Fan, Y., Jia, F.: Fast automatic 3D liver segmentation based on a three-level AdaBoost-guided active shape model. Med. Phys. 43(5), 2421–2434 (2016)
Lupascu, C.A., Tegolo, D., Trucco, E.: FABC: retinal vessel segmentation using AdaBoost. IEEE Trans. Inf. Technol. Biomed. A Publ. IEEE Eng. Med. Biol. Soc, 14(5), 1267–74 (2010)
Abdelhamid, O., Deng, L., Yu D., Jiang, H., Penn, G., Bouvrie, J., Lawrence, S., Giles, C.L., Tsoi, A.C., Back, A.D.: Gradient-based learning applied to document recognition (2013)
Everingham, M., Eslami, S., Van Gool, L., Williams, C., Winn, J., Zisserman, A.: The pascal visual object classes challenge: a retrospective. Int. J. Comput. Vis. 111(1), 98–136 (2015)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. arXiv, (2014)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: IEEE, (2016)
Trivizakis, E., Manikis, G.C., Nikiforaki, K., Drevelegas, K., Constantinides, M., Drevelegas, A., Marias, K.: Extending 2D convolutional neural networks to 3D for advancing deep learning cancer classification with application to MRI liver tumor differentiation. IEEE J. Biomed. Health Inf. 23(3), 923–930 (2018)
Badrinarayanan, V., Kendall, A., Cipolla, R.: Segnet: a deep convolutional encoder-decoder architecture for image segmentation. IEEE Trans. Pattern Anal. Mach. Intell. 39(12), 2481–2495 (2017)
Chen, L.C., Papandreou, G., Kokkinos, I., Murphy, K., Yuille, A.L.: Deeplab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected crfs. IEEE Trans. Pattern Anal. Mach. Intell. 40(4), 834–848 (2018)
Zhao, H., Shi, J., Qi, X., Wang, X., Jia, J.: Pyramid scene parsing network. In: IEEE Computer Society, (2016)
Yu, L., Cheng, J.Z., Qi, D., Xin, Y., Heng, P.A.: Automatic 3d cardiovascular MR segmentation with densely-connected volumetric convnets (2017)
Alom, M.Z., Hasan, M., Yakopcic, C., Taha, TM, Asari, V.K.: Recurrent residual convolutional neural network based on u-net (r2u-net) for medical image segmentation (2018)
Wang, B., Qiu, S., He, H.: Dual Encoding U-Net for Retinal Vessel Segmentation. In: Medical Image Computing and Computer Assisted Intervention—MICCAI 2019, 22nd International Conference, Shenzhen, China, October 13–17, 2019, Proceedings, Part I (2019)
Wu, H., Zhang, J., Huang, K., Liang, K., Yu, Y.: Fastfcn: rethinking dilated convolution in the backbone for semantic segmentation (2019)
Yisong, H.E., Jiang, J., Hang, Y.U., Yuchuan, F.U., Radiotherapy, D.O., Hospital, W.C., University, S.: Comparison of dice similarity coefficient and Hausdorff distance in image segmentation. Chin. J. Med. Phys. 36(11), 1307–1311 (2019)
Acknowledgements
We acknowledge the support of the Natural Science Foundation of Zhejiang Province (Grant No. LY22F020001) and the 3315 Plan Foundation of Ningbo (Grant No. 2019B-18-G).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Wu, R., Xin, Y., Dong, Y. et al. A dual-path U-Net for pulmonary vessel segmentation method based on lightweight 3D attention. Machine Vision and Applications 34, 87 (2023). https://doi.org/10.1007/s00138-023-01442-x
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s00138-023-01442-x