Dense Regression Activation Maps For Lesion Segmentation in CT scans of COVID-19 patients

Abstract

Automatic lesion segmentation on thoracic CT enables rapid quantitative analysis of lung involvement in COVID-19 infections. However, obtaining a large amount of voxel-level annotations for training segmentation networks is prohibitively expensive. Therefore, we propose a weakly-supervised segmentation method based on dense regression activation maps (dRAMs). Most weakly-supervised segmentation approaches exploit class activation maps (CAMs) to localize objects. However, because CAMs were trained for classification, they do not align precisely with the object segmentations. Instead, we produce high-resolution activation maps using dense features from a segmentation network that was trained to estimate a per-lobe lesion percentage. In this way, the network can exploit knowledge regarding the required lesion volume. In addition, we propose an attention neural network module to refine dRAMs, optimized together with the main regression task. We evaluated our algorithm on 90 subjects. Results show our method achieved 70.2% Dice coefficient, substantially outperforming the CAM-based baseline at 48.6%.

References in corpus (11)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification
• Training Deep Neural Networks on Noisy Labels with Bootstrapping
• 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation
• Learning Pixel-level Semantic Affinity with Image-level Supervision for Weakly Supervised Semantic Segmentation
• Self-supervised Equivariant Attention Mechanism for Weakly Supervised Semantic Segmentation
• Object Region Mining with Adversarial Erasing: A Simple Classification to Semantic Segmentation Approach
• GASNet: Weakly-supervised Framework for COVID-19 Lesion Segmentation
• Weakly supervised segmentation from extreme points
• Inf-Net: Automatic COVID-19 Lung Infection Segmentation from CT Images
• Label-Free Segmentation of COVID-19 Lesions in Lung CT