A general approach for automatic segmentation of pneumonia, pulmonary nodule, and tuberculosis in CT images

Abstract

Proposing a general segmentation approach for lung lesions, including pulmonary nodules, pneumonia, and tuberculosis, in CT images will improve efficiency in radiology. However, the performance of generative adversarial networks ishampered by the limited availability of annotated samples and the catastrophicforgetting of the discriminator, whereas the universality of traditional morphology-based methods is insufficient for segmenting diverse lung lesions. A cascaded dual-attention network with a context-aware pyramid feature extraction module was designed to address these challenges. A self-supervised rotation loss was designed to mitigate discriminator forgetting. The proposed model achieved Dice coefficients of 70.92, 73.55, and 68.52% on multi-center pneumonia, lung nodule, and tuberculosis test datasets, respectively. No significant decrease in accuracy was observed (p>0.10) when a small training sample size was used. The cyclic training of the discriminator was reduced with self-supervised rotation loss (p<0.01). The proposed approach is promising for segmenting multiple lung lesion types in CT images.