Inf-Net: Automatic COVID-19 Lung Infection Segmentation from CT Images
arXiv:2004.14133
Abstract
Coronavirus Disease 2019 (COVID-19) spread globally in early 2020, causing the world to face an existential health crisis. Automated detection of lung infections from computed tomography (CT) images offers a great potential to augment the traditional healthcare strategy for tackling COVID-19. However, segmenting infected regions from CT slices faces several challenges, including high variation in infection characteristics, and low intensity contrast between infections and normal tissues. Further, collecting a large amount of data is impractical within a short time period, inhibiting the training of a deep model. To address these challenges, a novel COVID-19 Lung Infection Segmentation Deep Network (Inf-Net) is proposed to automatically identify infected regions from chest CT slices. In our Inf-Net, a parallel partial decoder is used to aggregate the high-level features and generate a global map. Then, the implicit reverse attention and explicit edge-attention are utilized to model the boundaries and enhance the representations. Moreover, to alleviate the shortage of labeled data, we present a semi-supervised segmentation framework based on a randomly selected propagation strategy, which only requires a few labeled images and leverages primarily unlabeled data. Our semi-supervised framework can improve the learning ability and achieve a higher performance. Extensive experiments on our COVID-SemiSeg and real CT volumes demonstrate that the proposed Inf-Net outperforms most cutting-edge segmentation models and advances the state-of-the-art performance.
arXiv commentTo appear in IEEE TMI. The code is released in: https://github.com/DengPingFan/Inf-Net
References in corpus (13)
- Attention U-Net: Learning Where to Look for the Pancreas
- Res2Net: A New Multi-scale Backbone Architecture
- Can AI help in screening Viral and COVID-19 pneumonia?
- Deep Learning for Anomaly Detection: A Survey
- COVID-19 Image Data Collection
- COVID-CT-Dataset: A CT Scan Dataset about COVID-19
- Semi-Supervised Learning with Ladder Networks
- Embracing Imperfect Datasets: A Review of Deep Learning Solutions for Medical Image Segmentation
- Diagnosis of Coronavirus Disease 2019 (COVID-19) with Structured Latent Multi-View Representation Learning
- Severity Assessment of Coronavirus Disease 2019 (COVID-19) Using Quantitative Features from Chest CT Images
- Artificial Intelligence Forecasting of Covid-19 in China
- Harmony-Search and Otsu based System for Coronavirus Disease (COVID-19) Detection using Lung CT Scan Images
- Parting with Illusions about Deep Active Learning