arXiv reaDer
ディープラーニングの 2 段階アプローチを使用した COVID-19 の分類
COVID-19 Classification Using Deep Learning Two-Stage Approach
この論文では、深層学習ベースのアプローチ、つまり事前トレーニング済みの畳み込みニューラル ネットワーク (VGG16 および VGG19) の微調整、および開発された CNN モデルのエンドツーエンドのトレーニングを使用して、X 線画像を次のように分類しています。 COVID-19、正常、不透明、肺炎のケースを含む 4 つの異なるクラス。 20,000 を超える X 線スキャンを含むデータセットが Kaggle から取得され、この実験で使用されました。ワンショット分類アプローチと比較するために、2 段階分類アプローチが実装されました。私たちの仮説は、2 段階モデルは 1 回限りのモデルよりも優れたパフォーマンスを達成できるというものでした。私たちの結果は、VGG16 が 5 倍のトレーニングでワンショット アプローチを使用して 95% の精度を達成したことを示しています。今後の作業は、2 段階分類モデル Covid-TSC のより堅牢な実装に焦点を当てます。主な改善点は、データがステージ 1 の出力からステージ 2 の入力に流れるようにすることです。ステージ 1 とステージ 2 のモデルは、Covid-19 データセットで微調整された VGG16 モデルです。
In this paper, deep-learning-based approaches namely fine-tuning of pretrained convolutional neural networks (VGG16 and VGG19), and end-to-end training of a developed CNN model, have been used in order to classify X-Ray images into four different classes that include COVID-19, normal, opacity and pneumonia cases. A dataset containing more than 20,000 X-ray scans was retrieved from Kaggle and used in this experiment. A two-stage classification approach was implemented to be compared to the one-shot classification approach. Our hypothesis was that a two-stage model will be able to achieve better performance than a one-shot model. Our results show otherwise as VGG16 achieved 95% accuracy using one-shot approach over 5-fold of training. Future work will focus on a more robust implementation of the two-stage classification model Covid-TSC. The main improvement will be allowing data to flow from the output of stage-1 to the input of stage-2, where stage-1 and stage-2 models are VGG16 models fine-tuned on the Covid-19 dataset.
updated: Mon Nov 28 2022 23:03:29 GMT+0000 (UTC)
published: Mon Nov 28 2022 23:03:29 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アソシエイト