Two-stage hybrid network for segmentation of COVID-19 pneumonia lesions in CT images: a multicenter study

Med Biol Eng Comput. 2022 Sep;60(9):2721-2736. doi: 10.1007/s11517-022-02619-8. Epub 2022 Jul 19.

Abstract

COVID-19 has been spreading continuously since its outbreak, and the detection of its manifestations in the lung via chest computed tomography (CT) imaging is essential to investigate the diagnosis and prognosis of COVID-19 as an indispensable step. Automatic and accurate segmentation of infected lesions is highly required for fast and accurate diagnosis and further assessment of COVID-19 pneumonia. However, the two-dimensional methods generally neglect the intraslice context, while the three-dimensional methods usually have high GPU memory consumption and calculation cost. To address these limitations, we propose a two-stage hybrid UNet to automatically segment infected regions, which is evaluated on the multicenter data obtained from seven hospitals. Moreover, we train a 3D-ResNet for COVID-19 pneumonia screening. In segmentation tasks, the Dice coefficient reaches 97.23% for lung segmentation and 84.58% for lesion segmentation. In classification tasks, our model can identify COVID-19 pneumonia with an area under the receiver-operating characteristic curve value of 0.92, an accuracy of 92.44%, a sensitivity of 93.94%, and a specificity of 92.45%. In comparison with other state-of-the-art methods, the proposed approach could be implemented as an efficient assisting tool for radiologists in COVID-19 diagnosis from CT images.

Keywords: COVID-19; Computed tomography; Infected lesion segmentation; Screening.

Publication types

  • Multicenter Study

MeSH terms

  • COVID-19 Testing
  • COVID-19* / diagnostic imaging
  • Humans
  • Lung / diagnostic imaging
  • SARS-CoV-2
  • Tomography, X-Ray Computed / methods