A noise-robust framework for automatic segmentation of COVID-19 pneumonia lesions from CT images

G Wang, X Liu, C Li, Z Xu, J Ruan, H Zhu… - … on Medical Imaging, 2020 - ieeexplore.ieee.org
G Wang, X Liu, C Li, Z Xu, J Ruan, H Zhu, T Meng, K Li, N Huang, S Zhang
IEEE Transactions on Medical Imaging, 2020ieeexplore.ieee.org
Segmentation of pneumonia lesions from CT scans of COVID-19 patients is important for
accurate diagnosis and follow-up. Deep learning has a potential to automate this task but
requires a large set of high-quality annotations that are difficult to collect. Learning from
noisy training labels that are easier to obtain has a potential to alleviate this problem. To this
end, we propose a novel noise-robust framework to learn from noisy labels for the
segmentation task. We first introduce a noise-robust Dice loss that is a generalization of Dice …
Segmentation of pneumonia lesions from CT scans of COVID-19 patients is important for accurate diagnosis and follow-up. Deep learning has a potential to automate this task but requires a large set of high-quality annotations that are difficult to collect. Learning from noisy training labels that are easier to obtain has a potential to alleviate this problem. To this end, we propose a novel noise-robust framework to learn from noisy labels for the segmentation task. We first introduce a noise-robust Dice loss that is a generalization of Dice loss for segmentation and Mean Absolute Error (MAE) loss for robustness against noise, then propose a novel COVID-19 Pneumonia Lesion segmentation network (COPLE-Net) to better deal with the lesions with various scales and appearances. The noise-robust Dice loss and COPLE-Net are combined with an adaptive self-ensembling framework for training, where an Exponential Moving Average (EMA) of a student model is used as a teacher model that is adaptively updated by suppressing the contribution of the student to EMA when the student has a large training loss. The student model is also adaptive by learning from the teacher only when the teacher outperforms the student. Experimental results showed that: (1) our noise-robust Dice loss outperforms existing noise-robust loss functions, (2) the proposed COPLE-Net achieves higher performance than state-of-the-art image segmentation networks, and (3) our framework with adaptive self-ensembling significantly outperforms a standard training process and surpasses other noise-robust training approaches in the scenario of learning from noisy labels for COVID-19 pneumonia lesion segmentation.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果