arXiv reaDer
COVID-MTL:COVID-19の自動診断と重症度評価のためのShift3Dとランダム加重損失によるマルチタスク学習
COVID-MTL: Multitask Learning with Shift3D and Random-weighted Loss for Automated Diagnosis and Severity Assessment of COVID-19
COVID-19の正確で効果的な評価を支援するための自動化された方法が緊急に必要とされています。放射線医学および核酸検査(NAT)は、補完的なCOVID-19診断方法です。この論文では、COVID-19の自動化された同時検出(放射線とNATの両方に対して)と重症度評価が可能なエンドツーエンドのマルチタスク学習(MTL)フレームワーク(COVID-MTL)を紹介します。 COVID-MTLは、新しいランダム加重損失関数を介して、さまざまなCOVID-19タスクを並行して学習します。この関数は、ディリクレ分布の下で学習加重を割り当てて、タスクの優位性を防ぎます。新しい3Dリアルタイム拡張アルゴリズム(Shift3D)は、体積入力の低レベルの特徴表現を3次元でシフトすることにより、3DCNNコンポーネントの空間分散を導入します。これにより、MTLフレームワークは、単一タスクモデルと比較して、収束を加速し、共同学習のパフォーマンスを向上させることができます。胸部CTスキャンのみを使用することにより、COVID-MTLは930のCTスキャンでトレーニングされ、別々の399のケースでテストされました。 COVID-MTLは、放射線学およびNATに対するCOVID-19の検出で、それぞれ0.939および0.846のAUC、および90.23%および79.20%の精度を達成し、最先端のモデルを上回りました。一方、COVID-MTLは、コントロール/疑わしい、軽度/通常、および重度/重症の症例を分類するために、0.800±0.020および0.813±0.021(転移学習あり)のAUCをもたらしました。認識メカニズムを解読するために、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 ± 0.020 and 0.813 ± 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.
updated: Thu Dec 31 2020 14:27:16 GMT+0000 (UTC)
published: Thu Dec 10 2020 08:30:46 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト