Deep Active Learning For Fibrosis Segmentation Of Chest CT Scans From Covid-19 Patients

摘要

During the ongoing COVID-19 outbreak, it is critical to assess patients’ disease progression with COVID-19 pneumonia by computed tomography (CT). As most of the works focused on ground-glass opacity and consolidation segmentation of COVID-19 on CT images, lung fibrosis is relatively undervalued and less studied. Automatic segmentation and accurate measurement of lung fibrosis can potentially aid treatment planning for patients of post-COVID-19 pneumonia. However, the lack of sufficient training data hinders the fibrosis segmentation of CT images. Also, redundancy among CT images can reduce annotating efficiency. To address these issues, we propose deep active learning (AL) framework, which consists of a segmentation model called UNet-RGD, and a novel acquisition method named DeepRISS. The segmentation model consists of improved structures of residual blocks, channel gates, and dropout layers. The deep learning-based acquisition method combines uncertainty estimation and clustering for selecting representative and informative samples. Experimental results show that the AL framework can achieve state-of-the-art performance and effectively reduce the number of selected samples, saving the annotation cost by 25% to 44% compared to the non-selective approach.