arXiv reaDer
A Data Augmented Approach to Transfer Learning for Covid-19 Detection
早期のCovid-19検出は、その広がりを防ぐための効果的な治療と隔離計画に役立ちます。最近、X線、超音波、およびCTスキャンを使用したCovid-19検出に、転移学習が使用されています。これらの提案された方法に固有の主な制限の1つは、Covid-19診断と疾患進行の信頼性に影響を与えるラベル付きデータセットサイズの制限です。この作業では、コントラスト制限適応ヒストグラム均等化(CLAHE)を使用して、事前トレーニング済みの深層学習モデルの最後のレイヤーをトレーニングし、Covid-の転移学習のバイアスを軽減することで、制限付きX線画像データを拡張する方法を示します。 19検出。 AlexNet、ZFNet、VGG-16、ResNet-18、GoogLeNetなど、事前にトレーニングされたさまざまな深層学習モデルを転送し、CLAHE拡張データセットを使用して最後のレイヤーを微調整します。実験結果は、事前にトレーニングされたさまざまな深層学習モデルに対するCLAHEベースの拡張により、モデルの効率が大幅に向上することを示しています。 CLAHEベースの拡張画像を使用した事前トレーニング済みのVCG-16モデルは、15エポックを使用して95%の感度を実現します。 AlexNetの作品は、拡張されていないデータでトレーニングされたときに優れた感度を示します。他のモデルは、拡張されていないデータでトレーニングされた場合、60%未満の値を示します。私たちの結果は、サンプルのバイアスが、CLAHEベースの拡張を使用することによって大幅に改善される転移学習のパフォーマンスに悪影響を与える可能性があることを明らかにしています。
Covid-19 detection at an early stage can aid in an effective treatment and isolation plan to prevent its spread. Recently, transfer learning has been used for Covid-19 detection using X-ray, ultrasound, and CT scans. One of the major limitations inherent to these proposed methods is limited labeled dataset size that affects the reliability of Covid-19 diagnosis and disease progression. In this work, we demonstrate that how we can augment limited X-ray images data by using Contrast limited adaptive histogram equalization (CLAHE) to train the last layer of the pre-trained deep learning models to mitigate the bias of transfer learning for Covid-19 detection. We transfer learned various pre-trained deep learning models including AlexNet, ZFNet, VGG-16, ResNet-18, and GoogLeNet, and fine-tune the last layer by using CLAHE-augmented dataset. The experiment results reveal that the CLAHE-based augmentation to various pre-trained deep learning models significantly improves the model efficiency. The pre-trained VCG-16 model with CLAHEbased augmented images achieves a sensitivity of 95% using 15 epochs. AlexNet works show good sensitivity when trained on non-augmented data. Other models demonstrate a value of less than 60% when trained on non-augmented data. Our results reveal that the sample bias can negatively impact the performance of transfer learning which is significantly improved by using CLAHE-based augmentation.
updated: Thu Aug 05 2021 22:23:23 GMT+0000 (UTC)
published: Thu Aug 05 2021 22:23:23 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト