D2A U-Net: Automatic segmentation of COVID-19 CT slices based on dual attention and hybrid dilated convolution

Xiangyu Zhao,Peng Zhang,Fan Song,Guangda Fan,Yangyang Sun,Yujia Wang,Zheyuan Tian,Luqi Zhang,Guanglei Zhang
DOI: https://doi.org/10.1016/j.compbiomed.2021.104526
IF: 7.7
2021-08-01
Computers in Biology and Medicine
Abstract:<p>Coronavirus Disease 2019 (COVID-19) has become one of the most urgent public health events worldwide due to its high infectivity and mortality. Computed tomography (CT) is a significant screening tool for COVID-19 infection, and automatic segmentation of lung infection in COVID-19 CT images can assist diagnosis and health care of patients. However, accurate and automatic segmentation of COVID-19 lung infections is faced with a few challenges, including blurred edges of infection and relatively low sensitivity. To address the issues above, a novel <em>dilated dual attention U-Net</em> based on the dual attention strategy and hybrid dilated convolutions, namely <em>D2A U-Net</em>, is proposed for COVID-19 lesion segmentation in CT slices. In our <em>D2A U-Net</em>, the dual attention strategy composed of two attention modules is utilized to refine feature maps and reduce the semantic gap between different levels of feature maps. Moreover, the hybrid dilated convolutions are introduced to the model decoder to achieve larger receptive fields, which refines the decoding process. The proposed method is evaluated on an open-source dataset and achieves a Dice score of 0.7298 and recall score of 0.7071, which outperforms the popular cutting-edge methods in the semantic segmentation. The proposed network is expected to be a potential AI-based approach used for the diagnosis and prognosis of COVID-19 patients.</p>
engineering, biomedical,computer science, interdisciplinary applications,mathematical & computational biology,biology
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve The paper aims to address the following issues: 1. **Improving the accuracy of automatic segmentation of COVID-19 lung infections**: For the automatic segmentation of COVID-19 lung infection areas in CT images, existing methods face several challenges, including blurred infection edges and relatively low sensitivity. To tackle these challenges, the authors propose a new network architecture—D2A U-Net. 2. **Introducing a Dual Attention Strategy**: By combining the Gated Attention Module (GAM) and the Decoder Attention Module (DAM), the paper aims to refine feature maps and reduce the semantic gap between feature maps at different levels. 3. **Utilizing Hybrid Dilated Convolution**: The paper introduces hybrid dilated convolution in the model decoder to achieve a larger receptive field and improve the decoding process, thereby enhancing segmentation performance, especially in segmenting blurred lesions. With these improvements, the proposed D2A U-Net outperforms current popular state-of-the-art methods on public datasets, achieving higher Dice scores and recall rates in semantic segmentation tasks. This network is expected to become a potential AI-based method for the diagnosis and prognosis evaluation of COVID-19 patients.