arXiv reaDer
転移学習を使用したCOVID-19関連肺浸潤のセグメンテーションのための3DU-Net:手頃な価格のハードウェアでの最先端の結果
3D U-Net for segmentation of COVID-19 associated pulmonary infiltrates using transfer learning: State-of-the-art results on affordable hardware
肺浸潤のセグメンテーションは、COVID-19の重症度を評価するのに役立ちますが、手動のセグメンテーションは労力と時間がかかります。ニューラルネットワークを使用して肺浸潤をセグメント化すると、このタスクの自動化が可能になります。ただし、コンピュータ断層撮影(CT)データから3D U-Netをトレーニングすることは、時間とリソースを大量に消費します。したがって、この作業では、転送学習を使用して、限られたハードウェアでより短い時間で最先端のセグメンテーションモデルをトレーニングする方法に関するソリューションを開発およびテストしました。最近公開されたRSNAInternational COVID-19 Open Radiology Database(RICORD)を使用して、Kinetics-400データセットでエンコーダーとして事前トレーニングされた18層の3DResNetを使用して完全な3次元U-Netアーキテクチャをトレーニングします。次に、モデルの一般化は、胸部CTを受けたCOVID-19患者の2つの公開されているデータセット(コロナケースとMosMedデータセット)でテストされました。私たちのモデルは、以前に公開された3D U-Netアーキテクチャと同等のパフォーマンスを発揮し、チューニングデータセットで0.679、Coronacasesデータセットで0.648、MosMedデータセットで0.405の平均ダイススコアを達成しました。特に、これらの結果は、以前の研究で使用されたGPUよりも使用可能なメモリが少ない単一のGPUでの短いトレーニング時間で達成されました。
Segmentation of pulmonary infiltrates can help assess severity of COVID-19, but manual segmentation is labor and time-intensive. Using neural networks to segment pulmonary infiltrates would enable automation of this task. However, training a 3D U-Net from computed tomography (CT) data is time- and resource-intensive. In this work, we therefore developed and tested a solution on how transfer learning can be used to train state-of-the-art segmentation models on limited hardware and in shorter time. We use the recently published RSNA International COVID-19 Open Radiology Database (RICORD) to train a fully three-dimensional U-Net architecture using an 18-layer 3D ResNet, pretrained on the Kinetics-400 dataset as encoder. The generalization of the model was then tested on two openly available datasets of patients with COVID-19, who received chest CTs (Corona Cases and MosMed datasets). Our model performed comparable to previously published 3D U-Net architectures, achieving a mean Dice score of 0.679 on the tuning dataset, 0.648 on the Coronacases dataset and 0.405 on the MosMed dataset. Notably, these results were achieved with shorter training time on a single GPU with less memory available than the GPUs used in previous studies.
updated: Mon Jan 25 2021 09:37:32 GMT+0000 (UTC)
published: Mon Jan 25 2021 09:37:32 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アソシエイト