Abstract
Coronavirus Disease 2019 (COVID-19) ravaged the world in early 2020, causing great harm to human health. However, there are several challenges to segment the infected areas from computed tomography (CT) image, including blurry boundaries between the lesion and normal lung tissues, and uncertain characteristics about lesion’s scale, location, and texture. To solve these problems, a COVID-19 lesion segmentation network (LS-Net) based on probabilistic diffusion model is proposed to segment lesion areas from CT images. The feature fusion decoder module is introduced to aggregate high-level features and generate a guidance as the next steps so that the small lesion could not be omitted. In addition, the attention mechanism is set to pay attention to the information about position of lesion’s edge. So, the LS-Net framework can improve the precision of lesion segmentation result from CT image slice. Experiments on datasets such as the COVID-19 CT Segmentation dataset shows that LS-Net is advanced than most current segmentation models.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Wang, C., Horby, P.W., Hayden, F.G., Gao, G.F.: A novel coronavirus outbreak of global health concern. Lancet 395(10223), 470–473 (2020)
Statement on the second meeting of the international health regulations (2005) emergency committee regarding the outbreak of novel coronavirus (2019-nCoV). WHO, Geneva, Switzerland (2020)
Clinical management of severe acute respiratory infection when novel coronavirus (nCoV) infection is suspected: Interim Guidance 12 January 2020. WTO, nCoV, Clinical (2020)
Scheme for Diagnosis and Treatment of 2019 Novel Coronavirus Pneumonia (The 9th Trial Edition). China (2019)
Fang, Y., et al.: Sensitivity of chest CT for COVID-19: comparison to RT-PCR. Radiology 296, 200432 (2020)
Ye, Z., Zhang, Y., Wang, Y., Huang, Z., Song, B.: Chest CT manifestations of new coronavirus disease 2019 (COVID-19): a pictorial review. Eur. Radiol. 2019(37), 1–9 (2020)
Pan, F., et al.: Time course of lung changes at chest CT during recovery from coronavirus disease 2019 (COVID-19). Radiology 295(3), 715–721 (2020)
Shi, F., Wang, J., et al.: Review of artificial intelligence techniques in imaging data acquisition, segmentation, and diagnosis for COVID-19. IEEE Rev. Biomed. Eng. 14, 4–15 (2021)
Sharma, P., Shamout, F.E., Clifton, D.A.: Preserving patient privacy while training a predictive model of in-hospital mortality. In: 33rd Conference on Neural Information Processing Systems (NeurIPS 2019), Vancouver, Canada (2019)
Wang, X., Deng, X., Fu, Q., et al.: A weakly-supervised framework for COVID-19 classification and lesion localization from chest CT. IEEE Trans. Med. Imaging 39(8), 2615–2625 (2020)
Ferdinandus, F., Yuniarno, E.M., Purnama, I.K.E., Purnomo, M.H.: Covid-19 lung segmentation using U-Net CNN based on computed tomography image. In: 2022 IEEE 9th International Conference on Computational Intelligence and Virtual Environments for Measurement Systems and Applications (CIVEMSA) (2022)
Saood, A., Hatem, I.: COVID-19 lung CT image segmentation using deep learning methods: U-Net versus SegNet. BMC Med. Imaging 21(1), 19 (2021)
Xiao, H., Ran, Z., Mabu, S., et al.: SAUNet++: an automatic segmentation model of COVID-19 lesion from CT slices. Vis Comput. 39, 2291–2304 (2023)
Zhang, D., Huang, G., Zhang, Q., et al.: Cross-modality deep feature learning for brain tumor segmentation. Pattern Recogn. 110, 107562 (2021)
Cheng, Z., Qu, A., He, X.: Contour-aware semantic segmentation network with spatial attention mechanism for medical image. Vis. Comput. 38, 749–762 (2022)
Nazir, A., Cheema, M.N., Sheng, B., et al.: ECSU-Net: an embedded clustering sliced U-net coupled with fusing strategy for efficient intervertebral disc segmentation and classification. IEEE Trans. Image Process. 31, 880–893 (2022)
Bhattacharyya, D., Thirupathi Rao, N., Joshua, E.S.N., et al.: A bidirectional deep learning architecture for lung nodule semantic segmentation. Vis. Comput. 39, 5245–5261 (2022)
Jangam, E., Barreto, A.A.D., Annavarapu, C.S.R.: Automatic detection of COVID-19 from chest CT scan and chest X-Rays images using deep learning, transfer learning and stacking. Appl. Intell. 52(2), 2243–2259 (2022). https://doi.org/10.1007/s10489-021-02393-4
Chen, H., Jiang, Y., Loew, M., Ko, H.: Unsupervised domain adaptation based COVID-19 CT infection segmentation network. Appl. Intell. 52, 6340–6353 (2022). https://doi.org/10.1007/s10489-021-02691-x
Karthik, R., Menaka, R., Hariharan, M., Won, D.: Contour-enhanced attention CNN for CT-based COVID-19 segmentation. Pattern Recogn. 125, 108538 (2022)
Zhao, X., Zhang, P., et al.: Prior attention network for multi-lesion segmentation in medical images. IEEE Trans. Med. Image 41(12), 3812–3823 (2022). https://doi.org/10.1109/TMI.2022.3197180
Li, X., Jiang, Y., et al.: Lightweight attention convolutional neural network for retinal vessel image segmentation. IEEE Trans. Ind. Inf. 17(3), 1958–1967 (2021)
Xia, H., Ma, M., Li, H., Song, S.: MC-Net: multi-scale context-attention network for medical CT image segmentation. Appl. Intell. 52, 1508–1519 (2021)
Goodfellow, I., et al.: Generative adversarial nets. In: NIPS, Canada (2014)
Ho, J., et al.: Denoising diffusion probabilistic models. In: Advances in Neural Information Processing Systems, vol. 33 (2020)
Dhariwal, P., Nichol, A.: Diffusion models beats GANs on image synthesis. In: Advances in Neural Information Processing Systems, vol. 34 (2021)
Yang, L., Zhang, Z., Song, Y.: Diffusion models: a comprehensive survey of methods and applications. Comput. Vis. Pattern Recogn. (2022). https://doi.org/10.48550/arXiv.2209.00796
Luo, S., Hu, W.: Diffusion probabilistic models for 3d point cloud generation. In: IEEE Conference on Computer Vision and Pattern Recognition, pp. 2836–2844 (2021). https://doi.org/10.1109/CVPR46437.2021.00286
Kazerouni, A., Aghdam, E.K., et al.: Diffusion model for medical image analysis: a comprehensive survey. Image Video Process. (2022)
Wolleb, J., et al.: Diffusion models for implicit image segmentation ensembles. Comput. Vis. Pattern Recogn. (2023). https://doi.org/10.48550/arXiv.2112.03145
Kim, B., Oh, Y., Ye, J.C.: Diffusion adversarial representation learning for self-supervised vessel segmentation. Comput. Vis. Pattern Recogn. (2023). https://doi.org/10.48550/arXiv.2209.14566
Wu, J., Fang, H., et al.: MedSegDiff: medical image segmentation with diffusion probabilistic model. Comput. Vis. Pattern Recogn. (2023). https://doi.org/10.48550/arXiv.2211.00611
Fan, D.-P., Zhou, T., et al.: Inf-Net: automatic COVID-19 lung infection segmentation from CT images. IEEE Trans. Med. Imaging 39(8), 2626–2637 (2020)
Valanarasu, J.M.J., Oza, P., Hacihaliloglu, I., Patel, V.M.: Medical transformer: gated axial-attention for medical image segmentation. In: de Bruijne, M., et al. (eds.) MICCAI 2021. LNCS, vol. 12901, pp. 36–46. Springer, Cham (2021). https://doi.org/10.1007/978-3-030-87193-2_4
Wang, H., Zhu, Y., Green, B., Adam, H., Yuille, A., Chen, L.-C.: Axial-DeepLab: stand-alone axial-attention for panoptic segmentation. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12349, pp. 108–126. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58548-8_7
Federau, C., Christensen, S., Scherrer, N., Ospel, J.M., Schulze-Zachau, V., Schmidt, N., et al.: Improved segmentation and detection sensitivity of diffusion-weighted stroke lesions with synthetically enhanced deep learning. Radiol.: Artif. Intell. 2(5), e190217 (2020)
Luo, X., Chen, J., Song, T., Wang, G.: Semi-supervised medical image segmentation through dual-task consistency. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, p. 10 (2021)
Guo, X., et al: Accelerating diffusion models via pre-segmentation diffusion sampling for medical image segmentation. Image Video Process. (2022). https://doi.org/10.48550/arXiv.2210.17408
Tang, F., et al.: Multi-level global context cross consistency model for semi-supervised ultrasound image segmentation with diffusion model. Comput. Vis. Pattern Recogn. (2023). https://doi.org/10.48550/arXiv.2305.09447
Liang, S., Nie, R., Cao, J., et al.: FCF: feature complement fusion network for detecting COVID-19 through CT scan images. Appl. Soft Comput. 125, 109111 (2022)
Huang, Z., Li, L., Zhang, X., et al.: A coarse-refine segmentation network for COVID-19 CT images. IET Image Process. 16, 333–343 (2022)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Shi, A. et al. (2024). LS-Net: COVID-19 Lesion Segmentation from CT Image via Diffusion Probabilistic Model. In: Sheng, B., Bi, L., Kim, J., Magnenat-Thalmann, N., Thalmann, D. (eds) Advances in Computer Graphics. CGI 2023. Lecture Notes in Computer Science, vol 14498. Springer, Cham. https://doi.org/10.1007/978-3-031-50078-7_13
Download citation
DOI: https://doi.org/10.1007/978-3-031-50078-7_13
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-50077-0
Online ISBN: 978-3-031-50078-7
eBook Packages: Computer ScienceComputer Science (R0)