arXiv reaDer
UncertaintyFuseNet:COVID-19検出用のアンサンブルモンテカルロドロップアウトを備えたロバストな不確実性を意識した階層的特徴融合モデル
UncertaintyFuseNet: Robust Uncertainty-aware Hierarchical Feature Fusion Model with Ensemble Monte Carlo Dropout for COVID-19 Detection
COVID-19(コロナウイルス病2019)のパンデミックは、人間の健康と福祉に対する主要な世界的脅威となっています。したがって、胸部コンピュータ断層撮影(CT)とX線データを使用してCOVID-19を他の疾患と正確に区別できるコンピュータ支援検出(CAD)システムの開発が当面の優先事項です。このような自動システムは通常、従来の機械学習または深層学習の方法に基づいています。 COVID-19ケース分類でCTスキャンまたはX線画像のいずれかを使用した既存の研究のほとんどとは異なり、UncertaintyFuseNetと呼ばれるシンプルで効率的な深層学習機能融合モデルを提示します。これは両方の大規模なデータセットを正確に分類できます。これらのタイプの画像の。既存の研究のほとんどがそれを見落としていたとしても、モデルの予測の不確実性を学習プロセスで考慮に入れるべきであると私たちは主張します。効果的なEnsembleMC Dropout(EMCD)手法を使用して、特徴融合モデルの予測の不確実性を定量化します。新しいモデルの結果を既存のアプローチと比較するために包括的なシミュレーション研究が実施され、適合率、再現率、Fメジャー、精度、およびROC曲線の観点から競合モデルのパフォーマンスが評価されました。得られた結果は、考慮されたCTスキャンとX線データセットに対してそれぞれ99.08%と96.35%の予測精度を提供したモデルの効率を証明しています。さらに、私たちのUncertaintyFuseNetモデルは、一般的にノイズに対してロバストであり、以前は見られなかったデータでうまく機能しました。実装のソースコードは、https://github.com/moloud1987/UncertaintyFuseNet-for-COVID-19-Classificationから無料で入手できます。
The COVID-19 (Coronavirus disease 2019) pandemic has become a major global threat to human health and well-being. Thus, the development of computer-aided detection (CAD) systems that are capable to accurately distinguish COVID-19 from other diseases using chest computed tomography (CT) and X-ray data is of immediate priority. Such automatic systems are usually based on traditional machine learning or deep learning methods. Differently from most of existing studies, which used either CT scan or X-ray images in COVID-19-case classification, we present a simple but efficient deep learning feature fusion model, called UncertaintyFuseNet, which is able to classify accurately large datasets of both of these types of images. We argue that the uncertainty of the model's predictions should be taken into account in the learning process, even though most of existing studies have overlooked it. We quantify the prediction uncertainty in our feature fusion model using effective Ensemble MC Dropout (EMCD) technique. A comprehensive simulation study has been conducted to compare the results of our new model to the existing approaches, evaluating the performance of competing models in terms of Precision, Recall, F-Measure, Accuracy and ROC curves. The obtained results prove the efficiency of our model which provided the prediction accuracy of 99.08% and 96.35% for the considered CT scan and X-ray datasets, respectively. Moreover, our UncertaintyFuseNet model was generally robust to noise and performed well with previously unseen data. The source code of our implementation is freely available at: https://github.com/moloud1987/UncertaintyFuseNet-for-COVID-19-Classification.
updated: Sun Jan 30 2022 08:45:27 GMT+0000 (UTC)
published: Tue May 18 2021 15:20:34 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アソシエイト