D2A U-Net: Automatic Segmentation of COVID-19 Lesions from CT Slices
with Dilated Convolution and Dual Attention Mechanism
- URL: http://arxiv.org/abs/2102.05210v1
- Date: Wed, 10 Feb 2021 01:21:59 GMT
- Title: D2A U-Net: Automatic Segmentation of COVID-19 Lesions from CT Slices
with Dilated Convolution and Dual Attention Mechanism
- Authors: Xiangyu Zhao, Peng Zhang, Fan Song, Guangda Fan, Yangyang Sun, Yujia
Wang, Zheyuan Tian, Luqi Zhang, Guanglei Zhang
- Abstract summary: We propose a dilated dual attention U-Net (D2A U-Net) for COVID-19 lesion segmentation in CT slices based on dilated convolution and a novel dual attention mechanism.
Our experiment results have shown that by introducing dilated convolution and dual attention mechanism, the number of false positives is significantly reduced.
- Score: 9.84838467721235
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Coronavirus Disease 2019 (COVID-19) has caused great casualties and becomes
almost the most urgent public health events worldwide. Computed tomography (CT)
is a significant screening tool for COVID-19 infection, and automated
segmentation of lung infection in COVID-19 CT images will greatly assist
diagnosis and health care of patients. However, accurate and automatic
segmentation of COVID-19 lung infections remains to be challenging. In this
paper we propose a dilated dual attention U-Net (D2A U-Net) for COVID-19 lesion
segmentation in CT slices based on dilated convolution and a novel dual
attention mechanism to address the issues above. We introduce a dilated
convolution module in model decoder to achieve large receptive field, which
refines decoding process and contributes to segmentation accuracy. Also, we
present a dual attention mechanism composed of two attention modules which are
inserted to skip connection and model decoder respectively. The dual attention
mechanism is utilized to refine feature maps and reduce semantic gap between
different levels of the model. The proposed method has been evaluated on
open-source dataset and outperforms cutting edges methods in semantic
segmentation. Our proposed D2A U-Net with pretrained encoder achieves a Dice
score of 0.7298 and recall score of 0.7071. Besides, we also build a simplified
D2A U-Net without pretrained encoder to provide a fair comparison with other
models trained from scratch, which still outperforms popular U-Net family
models with a Dice score of 0.7047 and recall score of 0.6626. Our experiment
results have shown that by introducing dilated convolution and dual attention
mechanism, the number of false positives is significantly reduced, which
improves sensitivity to COVID-19 lesions and subsequently brings significant
increase to Dice score.
Related papers
- EDAC: Efficient Deployment of Audio Classification Models For COVID-19
Detection [0.0]
The global spread of COVID-19 had severe consequences for public health and the world economy.
Various researchers made use of machine learning methods in an attempt to detect COVID-19.
The solutions leverage various input features, such as CT scans or cough audio signals, with state-of-the-art results arising from deep neural network architectures.
To address this, we first recreated two models that use cough audio recordings to detect COVID-19.
arXiv Detail & Related papers (2023-09-11T10:07:51Z) - AttResDU-Net: Medical Image Segmentation Using Attention-based Residual
Double U-Net [0.0]
This paper proposes an attention-based residual Double U-Net architecture (AttResDU-Net) that improves on the existing medical image segmentation networks.
We conducted experiments on three datasets: CVC Clinic-DB, ISIC 2018, and the 2018 Data Science Bowl datasets and achieved Dice Coefficient scores of 94.35%, 91.68%, and 92.45% respectively.
arXiv Detail & Related papers (2023-06-25T14:28:08Z) - Dual Multi-scale Mean Teacher Network for Semi-supervised Infection
Segmentation in Chest CT Volume for COVID-19 [76.51091445670596]
Automated detecting lung infections from computed tomography (CT) data plays an important role for combating COVID-19.
Most current COVID-19 infection segmentation methods mainly relied on 2D CT images, which lack 3D sequential constraint.
Existing 3D CT segmentation methods focus on single-scale representations, which do not achieve the multiple level receptive field sizes on 3D volume.
arXiv Detail & Related papers (2022-11-10T13:11:21Z) - Dual-Consistency Semi-Supervised Learning with Uncertainty
Quantification for COVID-19 Lesion Segmentation from CT Images [49.1861463923357]
We propose an uncertainty-guided dual-consistency learning network (UDC-Net) for semi-supervised COVID-19 lesion segmentation from CT images.
Our proposed UDC-Net improves the fully supervised method by 6.3% in Dice and outperforms other competitive semi-supervised approaches by significant margins.
arXiv Detail & Related papers (2021-04-07T16:23:35Z) - COVID-Net CT-2: Enhanced Deep Neural Networks for Detection of COVID-19
from Chest CT Images Through Bigger, More Diverse Learning [70.92379567261304]
We introduce COVID-Net CT-2, enhanced deep neural networks for COVID-19 detection from chest CT images.
We leverage explainability to investigate the decision-making behaviour of COVID-Net CT-2.
Results are promising and suggest the strong potential of deep neural networks as an effective tool for computer-aided COVID-19 assessment.
arXiv Detail & Related papers (2021-01-19T03:04:09Z) - Exploiting Shared Knowledge from Non-COVID Lesions for
Annotation-Efficient COVID-19 CT Lung Infection Segmentation [10.667692828593125]
We propose a relation-driven collaborative learning model for COVID-19 lung infection segmentation.
We exploit shared knowledge between COVID and non-COVID lesions to regularize the relation consistency between extracted features.
Our method achieves superior segmentation performance compared with existing methods in the absence of sufficient high-quality COVID-19 annotations.
arXiv Detail & Related papers (2020-12-31T11:40:29Z) - CHS-Net: A Deep learning approach for hierarchical segmentation of
COVID-19 infected CT images [0.6091702876917281]
The pandemic of novel severe acute respiratory syndrome coronavirus 2 (SARS-CoV-2) also known as COVID-19 has been spreading worldwide.
Medical imaging such as computed tomography (CT), X-ray, etc., plays a significant role in diagnosing the patients by presenting the excellent details about the structure of the organs.
Deep learning technologies have displayed its strength in analyzing such scans to aid in the faster diagnosis of the diseases and viruses such as COVID-19.
arXiv Detail & Related papers (2020-12-13T15:02:05Z) - Classification of COVID-19 in CT Scans using Multi-Source Transfer
Learning [91.3755431537592]
We propose the use of Multi-Source Transfer Learning to improve upon traditional Transfer Learning for the classification of COVID-19 from CT scans.
With our multi-source fine-tuning approach, our models outperformed baseline models fine-tuned with ImageNet.
Our best performing model was able to achieve an accuracy of 0.893 and a Recall score of 0.897, outperforming its baseline Recall score by 9.3%.
arXiv Detail & Related papers (2020-09-22T11:53:06Z) - Automated Chest CT Image Segmentation of COVID-19 Lung Infection based
on 3D U-Net [0.0]
The coronavirus disease 2019 (COVID-19) affects billions of lives around the world and has a significant impact on public healthcare.
We propose an innovative automated segmentation pipeline for COVID-19 infected regions.
Our method focuses on on-the-fly generation of unique and random image patches for training by performing several preprocessing methods.
arXiv Detail & Related papers (2020-06-24T17:29:26Z) - Synergistic Learning of Lung Lobe Segmentation and Hierarchical
Multi-Instance Classification for Automated Severity Assessment of COVID-19
in CT Images [61.862364277007934]
We propose a synergistic learning framework for automated severity assessment of COVID-19 in 3D CT images.
A multi-task deep network (called M$2$UNet) is then developed to assess the severity of COVID-19 patients.
Our M$2$UNet consists of a patch-level encoder, a segmentation sub-network for lung lobe segmentation, and a classification sub-network for severity assessment.
arXiv Detail & Related papers (2020-05-08T03:16:15Z) - Inf-Net: Automatic COVID-19 Lung Infection Segmentation from CT Images [152.34988415258988]
Automated detection of lung infections from computed tomography (CT) images offers a great potential to augment the traditional healthcare strategy for tackling COVID-19.
segmenting infected regions from CT slices faces several challenges, including high variation in infection characteristics, and low intensity contrast between infections and normal tissues.
To address these challenges, a novel COVID-19 Deep Lung Infection Network (Inf-Net) is proposed to automatically identify infected regions from chest CT slices.
arXiv Detail & Related papers (2020-04-22T07:30:56Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.