A Weakly Supervised Region-Based Active Learning Method for COVID-19 Segmentation in CT Images

Abstract

One of the key challenges in the battle against the Coronavirus (COVID-19) pandemic is to detect and quantify the severity of the disease in a timely manner. Computed tomographies (CT) of the lungs are effective for assessing the state of the infection. Unfortunately, labeling CT scans can take a lot of time and effort, with up to 150 minutes per scan. We address this challenge introducing a scalable, fast, and accurate active learning system that accelerates the labeling of CT scan images. Conventionally, active learning methods require the labelers to annotate whole images with full supervision, but that can lead to wasted efforts as many of the annotations could be redundant. Thus, our system presents the annotator with unlabeled regions that promise high information content and low annotation cost. Further, the system allows annotators to label regions using point-level supervision, which is much cheaper to acquire than per-pixel annotations. Our experiments on open-source COVID-19 datasets show that using an entropy-based method to rank unlabeled regions yields to significantly better results than random labeling of these regions. Also, we show that labeling small regions of images is more efficient than labeling whole images. Finally, we show that with only 7\% of the labeling effort required to label the whole training set gives us around 90\% of the performance obtained by training the model on the fully annotated training set. Code is available at: \url{https://github.com/IssamLaradji/covid19_active_learning}.

References in corpus (9)

• Bayesian Active Learning for Classification and Preference Learning
• A Simple Baseline for Bayesian Uncertainty in Deep Learning
• COVID-19 Chest CT Image Segmentation -- A Deep Convolutional Neural Network Solution
• F3Net: Fusion, Feedback and Focus for Salient Object Detection
• CEREALS - Cost-Effective REgion-based Active Learning for Semantic Segmentation
• Reinforced active learning for image segmentation
• Active and Incremental Learning with Weak Supervision
• Instance Segmentation with Point Supervision
• An Adaptive Supervision Framework for Active Learning in Object Detection

Cited by in corpus (5)

• GASNet: Weakly-supervised Framework for COVID-19 Lesion Segmentation
• Counting Cows: Tracking Illegal Cattle Ranching From High-Resolution Satellite Imagery
• Affinity LCFCN: Learning to Segment Fish with Weak Supervision
• Diagnosis/Prognosis of COVID-19 Images: Challenges, Opportunities, and Applications
• A Survey of Machine Learning Techniques for Detecting and Diagnosing COVID-19 from Imaging