arXiv reaDer
胸部X線を使用したCOVID-19検出のための深い畳み込みニューラルネットワーク
A Deep Convolutional Neural Network for COVID-19 Detection Using Chest X-Rays
目的:高密度畳み込みネットワークに基づく画像分類器を提示し、COVID-19、肺炎、正常の3つのラベルに従って胸部X線画像を分類するための学習を転送します。方法:ImageNetで事前トレーニングされたニューラルネットワークを微調整し、NIH ChestX-ray14データセットを中間ステップとして使用して、2回転送学習アプローチを適用しました。また、出力ニューロン保持と呼ばれる新規性を提案しました。これは、2回転送学習手法を変更します。モデルの手口を明確にするために、レイヤーごとの関連性の伝播(LRP)を使用してヒートマップを生成しました。結果:テストデータセットで100%のテスト精度に到達することができました。 2回の伝達学習と出力ニューロンの維持は、主にトレーニングプロセスの開始時に、パフォーマンスを向上させる有望な結果を示しました。 LRPは、X線上の単語がネットワークの予測に影響を与える可能性があることを明らかにしましたが、これが精度に与える影響はごくわずかであることがわかりました。結論:優れた一般化をさらに確実にするためには、臨床研究とより大きなデータセットが依然として必要ですが、私たちが達成した最先端のパフォーマンスは、人工知能の助けを借りて、胸部X線が安価で正確な補助方法になる可能性があることを示していますCOVID-19診断用。 LRPによって生成されたヒートマップは、ディープニューラルネットワークの解釈可能性を改善し、診断に関する将来の研究のための分析パスを示します。パフォーマンスの向上を維持する出力ニューロンによる2回の伝達学習。
Purpose: We present image classifiers based on Dense Convolutional Networks and transfer learning to classify chest X-ray images according to three labels: COVID-19, pneumonia and normal. Methods: We fine-tuned neural networks pretrained on ImageNet and applied a twice transfer learning approach, using NIH ChestX-ray14 dataset as an intermediate step. We also suggested a novelty called output neuron keeping, which changes the twice transfer learning technique. In order to clarify the modus operandi of the models, we used Layer-wise Relevance Propagation (LRP) to generate heatmaps. Results: We were able to reach test accuracy of 100% on our test dataset. Twice transfer learning and output neuron keeping showed promising results improving performances, mainly in the beginning of the training process. Although LRP revealed that words on the X-rays can influence the networks' predictions, we discovered this had only a very small effect on accuracy. Conclusion: Although clinical studies and larger datasets are still needed to further ensure good generalization, the state-of-the-art performances we achieved show that, with the help of artificial intelligence, chest X-rays can become a cheap and accurate auxiliary method for COVID-19 diagnosis. Heatmaps generated by LRP improve the interpretability of the deep neural networks and indicate an analytical path for future research on diagnosis. Twice transfer learning with output neuron keeping improved performances.
updated: Fri Dec 11 2020 06:48:36 GMT+0000 (UTC)
published: Thu Apr 30 2020 13:20:42 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アソシエイト