Automatic COVID ‐19 CT segmentation using U‐Net integrated spatial and channel attention mechanism - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue International Journal of Imaging Systems and Technology Année : 2021

Automatic COVID ‐19 CT segmentation using U‐Net integrated spatial and channel attention mechanism

Automatic COVID-19 CT segmentation using U-Net integrated spatial and channel attention mechanism.

Tongxue Zhou
  • Fonction : Auteur
Stéphane Canu
  • Fonction : Auteur
Su Ruan

Résumé

The coronavirus disease (COVID-19) pandemic has led to a devastating effect on the global public health. Computed Tomography (CT) is an effective tool in the screening of COVID-19. It is of great importance to rapidly and accurately segment COVID-19 from CT to help diagnostic and patient monitoring. In this paper, we propose a U-Net based segmentation network using attention mechanism. As not all the features extracted from the encoders are useful for segmentation, we propose to incorporate an attention mechanism including a spatial attention module and a channel attention module, to a U-Net architecture to re-weight the feature representation spatially and channel-wise to capture rich contextual relationships for better feature representation. In addition, the focal Tversky loss is introduced to deal with small lesion segmentation. The experiment results, evaluated on a COVID-19 CT segmentation dataset where 473 CT slices are available, demonstrate the proposed method can achieve an accurate and rapid segmentation result on COVID-19. The method takes only 0.29 second to segment a single CT slice. The obtained Dice Score and Hausdorff Distance are 83.1% and 18.8, respectively.

Dates et versions

hal-03226497 , version 1 (14-05-2021)

Identifiants

Citer

Tongxue Zhou, Stéphane Canu, Su Ruan. Automatic COVID ‐19 CT segmentation using U‐Net integrated spatial and channel attention mechanism. International Journal of Imaging Systems and Technology, 2021, 31 (1), pp.16-27. ⟨10.1002/ima.22527⟩. ⟨hal-03226497⟩
23 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More