Automatic COVID-19 CT segmentation using U-Net integrated spatial and channel attention mechanism
© 2020 Wiley Periodicals LLC.
Publié dans: | International journal of imaging systems and technology. - 1998. - 31(2021), 1 vom: 24. März, Seite 16-27 |
---|---|
Auteur principal: | |
Autres auteurs: | , |
Format: | Article en ligne |
Langue: | English |
Publié: |
2021
|
Accès à la collection: | International journal of imaging systems and technology |
Sujets: | Journal Article COVID‐19 CT attention mechanism deep learning focal tversky loss segmentation |
Résumé: | © 2020 Wiley Periodicals LLC. 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 |
---|---|
Description: | Date Revised 05.06.2023 published: Print-Electronic Citation Status PubMed-not-MEDLINE |
ISSN: | 0899-9457 |
DOI: | 10.1002/ima.22527 |