论文标题
COVID-MTL:用于自动诊断和严重程度评估的多任务学习和随机加权损失-19
COVID-MTL: Multitask Learning with Shift3D and Random-weighted Loss for Automated Diagnosis and Severity Assessment of COVID-19
论文作者
论文摘要
迫切需要使用自动化方法来协助Covid-19的准确有效评估。放射学和核酸测试(NAT)是互补的COVID-19诊断方法。在本文中,我们提出了能够自动化和同时检测(针对放射学和NAT)以及COVID-19的严重性评估的端到端多任务学习(MTL)框架(COVID-MTL)。 Covid-Mtl通过我们新颖的随机加权损失函数并行学习了不同的Covid-19任务,该任务在Dirichlet分布下分配了学习权重,以防止任务优势。我们的新3D实时增强算法(Shift3D)通过将体积输入的低级特征表示在三个维度上转移,从而引入了3D CNN组件的空间差异;因此,与单任务模型相比,MTL框架能够加速收敛并提高关节学习绩效。通过仅使用胸部CT扫描,对Covid-MTL进行了930 CT扫描的训练,并在399例病例上进行了测试。 Covid-MTL的AUC达到了0.939和0.846,用于检测COVID-19的10.23%和79.20%的精度分别针对放射学和NAT,这表现优于最先进的模型。同时,Covid-MTL产生的AUC为0.800 $ \ pm $ 0.020和0.813 $ \ pm $ 0.021(带有转移学习),用于对控制/可疑,轻度/常规和严重/严重/严重的案件进行分类。为了破译识别机制,我们还确定了与COVID-19的阳性和严重程度显着相关的高通量肺特征(P <0.001)。
There is an urgent need for automated methods to assist accurate and effective assessment of COVID-19. Radiology and nucleic acid test (NAT) are complementary COVID-19 diagnosis methods. In this paper, we present an end-to-end multitask learning (MTL) framework (COVID-MTL) that is capable of automated and simultaneous detection (against both radiology and NAT) and severity assessment of COVID-19. COVID-MTL learns different COVID-19 tasks in parallel through our novel random-weighted loss function, which assigns learning weights under Dirichlet distribution to prevent task dominance; our new 3D real-time augmentation algorithm (Shift3D) introduces space variances for 3D CNN components by shifting low-level feature representations of volumetric inputs in three dimensions; thereby, the MTL framework is able to accelerate convergence and improve joint learning performance compared to single-task models. By only using chest CT scans, COVID-MTL was trained on 930 CT scans and tested on separate 399 cases. COVID-MTL achieved AUCs of 0.939 and 0.846, and accuracies of 90.23% and 79.20% for detection of COVID-19 against radiology and NAT, respectively, which outperformed the state-of-the-art models. Meanwhile, COVID-MTL yielded AUC of 0.800 $\pm$ 0.020 and 0.813 $\pm$ 0.021 (with transfer learning) for classifying control/suspected, mild/regular, and severe/critically-ill cases. To decipher the recognition mechanism, we also identified high-throughput lung features that were significantly related (P < 0.001) to the positivity and severity of COVID-19.