arXiv reaDer
Efficient and Visualizable Convolutional Neural Networks for COVID-19 Classification Using Chest CT
COVID-19の症例が急速に増加しているため、ディープラーニングは有望な診断手法として浮上しています。ただし、COVID-19患者を特徴付ける最も正確なモデルを特定することは困難です。これは、さまざまなタイプのデータと取得プロセスで得られた結果を比較することは簡単ではないためです。この論文では、EfficientNetファミリーを最初に検討した、COVID-19陽性、健康、または他の肺肺感染症に苦しんでいる患者を胸部CTスキャンに基づいて分類する際の、20の畳み込み中性ネットワークのパフォーマンスを設計、評価、比較しました。 COVID-19診断のために、モデルのパフォーマンスを視覚化するために中間活性化マップを採用します。すべてのモデルは、「CTスキャンのCOVIDマルチクラスデータセット」というタイトルのデータセットからの4173胸部CT画像と、COVID-19陽性、健康、またはその他に苦しんでいる患者の2168、758、および1247画像を使用して、Pythonでトレーニングおよび評価されます。それぞれ肺感染症。 EfficientNet-B5は、F1スコアが0.9769 +/- 0.0046、精度が0.9759 +/- 0.0048、感度が0.9788 +/- 0.0055、特異度が0.9730 +/- 0.0057、精度が0.9751 + /の最良のモデルとして特定されました。 -0.0051。代替の2クラスのデータセットでは、EfficientNetB5は0.9845 +/- 0.0109の精度、0.9599 +/- 0.0251のF1スコア、0.9682 +/- 0.0099の感度、0.9883 +/- 0.0150の特異度、および0.9526 + /の精度を取得しました。 -0.0523。中間アクティベーションマップと勾配加重クラスアクティベーションマッピングは、人工知能支援放射線ツールの有望なユースケースを示唆する、地上クラスの不透明度と統合に対するモデルの認識の人間が解釈できる証拠を提供しました。 GPUで0.1秒未満、CPUで0.5秒未満の予測速度で、提案されたモデルはCOVID-19の迅速でスケーラブルで正確な診断を提供します。
With COVID-19 cases rising rapidly, deep learning has emerged as a promising diagnosis technique. However, identifying the most accurate models to characterize COVID-19 patients is challenging because comparing results obtained with different types of data and acquisition processes is non-trivial. In this paper we designed, evaluated, and compared the performance of 20 convolutional neutral networks in classifying patients as COVID-19 positive, healthy, or suffering from other pulmonary lung infections based on Chest CT scans, serving as the first to consider the EfficientNet family for COVID-19 diagnosis and employ intermediate activation maps for visualizing model performance. All models are trained and evaluated in Python using 4173 Chest CT images from the dataset entitled "A COVID multiclass dataset of CT scans," with 2168, 758, and 1247 images of patients that are COVID-19 positive, healthy, or suffering from other pulmonary infections, respectively. EfficientNet-B5 was identified as the best model with an F1 score of 0.9769+/-0.0046, accuracy of 0.9759+/-0.0048, sensitivity of 0.9788+/-0.0055, specificity of 0.9730+/-0.0057, and precision of 0.9751 +/- 0.0051. On an alternate 2-class dataset, EfficientNetB5 obtained an accuracy of 0.9845+/-0.0109, F1 score of 0.9599+/-0.0251, sensitivity of 0.9682+/-0.0099, specificity of 0.9883+/-0.0150, and precision of 0.9526 +/- 0.0523. Intermediate activation maps and Gradient-weighted Class Activation Mappings offered human-interpretable evidence of the model's perception of ground-class opacities and consolidations, hinting towards a promising use-case of artificial intelligence-assisted radiology tools. With a prediction speed of under 0.1 seconds on GPUs and 0.5 seconds on CPUs, our proposed model offers a rapid, scalable, and accurate diagnostic for COVID-19.
updated: Mon Jul 19 2021 05:29:28 GMT+0000 (UTC)
published: Tue Dec 22 2020 07:09:48 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト