论文标题
对比度跨站点学习,重新设计的网络进行了COVID-19 CT分类
Contrastive Cross-site Learning with Redesigned Net for COVID-19 CT Classification
论文作者
论文摘要
2019年冠状病毒疾病大流行(Covid-19)导致了一场全球公共卫生危机,蔓延了数百个国家。随着新感染的持续增长,非常需要开发用于COT-19鉴定的自动化工具,以帮助临床诊断并减少图像解释的繁琐工作量。为了扩大用于开发机器学习方法的数据集,从不同的医学系统中汇总案例以学习可靠且可推广的模型基本上很有帮助。本文提出了一个新颖的联合学习框架,以通过具有分布差异的异质数据集有效地学习来执行准确的COVID-19识别。我们通过重新设计网络体系结构和学习策略方面的最近提出的Covid-NET来建立强大的主链,以提高预测准确性和学习效率。除了改进的主链外,我们还通过在潜在空间中进行单独的特征归一化来进一步明确处理跨站点的转移。此外,我们建议使用对比度训练目标来增强语义嵌入的域不变性,以提高每个数据集上的分类性能。我们使用由CT图像组成的两个公共大规模COVID-19诊断数据集开发和评估我们的方法。广泛的实验表明,我们的方法一致地改善了两个数据集的性能,在每个数据集上的原始covid-net的表现分别优于AUC的原始covid-net,而AUC的表现分别超过了12.16%和14.23%,也超过了现有的现有的最新最新的多站点学习方法。
The pandemic of coronavirus disease 2019 (COVID-19) has lead to a global public health crisis spreading hundreds of countries. With the continuous growth of new infections, developing automated tools for COVID-19 identification with CT image is highly desired to assist the clinical diagnosis and reduce the tedious workload of image interpretation. To enlarge the datasets for developing machine learning methods, it is essentially helpful to aggregate the cases from different medical systems for learning robust and generalizable models. This paper proposes a novel joint learning framework to perform accurate COVID-19 identification by effectively learning with heterogeneous datasets with distribution discrepancy. We build a powerful backbone by redesigning the recently proposed COVID-Net in aspects of network architecture and learning strategy to improve the prediction accuracy and learning efficiency. On top of our improved backbone, we further explicitly tackle the cross-site domain shift by conducting separate feature normalization in latent space. Moreover, we propose to use a contrastive training objective to enhance the domain invariance of semantic embeddings for boosting the classification performance on each dataset. We develop and evaluate our method with two public large-scale COVID-19 diagnosis datasets made up of CT images. Extensive experiments show that our approach consistently improves the performances on both datasets, outperforming the original COVID-Net trained on each dataset by 12.16% and 14.23% in AUC respectively, also exceeding existing state-of-the-art multi-site learning methods.