ディープラーニング(DL)は、その精度、効率、および客観性を考慮して、医療システムの再構築に大きな期待を寄せています。ただし、ノイズの多い配布外の入力に対するDLモデルの脆弱性は、診療所での展開を妨げています。ほとんどのシステムは、モデルの不確実性や信頼性に関する詳細情報なしで点推定を生成します。この論文では、セグメンテーションニューラルネットワーク、特にエンコーダ-デコーダアーキテクチャにおける不確実性の定量化のための新しいベイズ深層学習フレームワークを紹介します。提案されたフレームワークは、1次のテイラー級数近似を使用して、証拠の下限を最大化することにより、トレーニングデータが与えられた場合のモデルパラメーターの分布の最初の2つのモーメント(平均と共分散)を伝播および学習します。出力は、セグメント化された画像とセグメンテーションの不確実性マップの2つのマップで構成されます。セグメンテーション決定の不確実性は、予測分布の共分散行列によって取得されます。磁気共鳴イメージングとコンピューター断層撮影スキャンからの医療画像セグメンテーションデータで提案されたフレームワークを評価します。複数のベンチマークデータセットでの実験は、提案されたフレームワークが、最先端のセグメンテーションモデルと比較して、ノイズや敵対的な攻撃に対してより堅牢であることを示しています。さらに、提案されたフレームワークの不確実性マップは、ノイズ、アーティファクト、または敵対的な攻撃で破損したテスト入力画像のパッチに低い信頼性(または同等に高い不確実性)を関連付けます。したがって、モデルは、不確実性マップでより高い値を提示することにより、誤った予測を行ったり、セグメンテーション構造の一部(腫瘍など)を見逃したりした場合に、セグメンテーションの決定を自己評価できます。
Deep Learning (DL) holds great promise in reshaping the healthcare systems given its precision, efficiency, and objectivity. However, the brittleness of DL models to noisy and out-of-distribution inputs is ailing their deployment in the clinic. Most systems produce point estimates without further information about model uncertainty or confidence. This paper introduces a new Bayesian deep learning framework for uncertainty quantification in segmentation neural networks, specifically encoder-decoder architectures. The proposed framework uses the first-order Taylor series approximation to propagate and learn the first two moments (mean and covariance) of the distribution of the model parameters given the training data by maximizing the evidence lower bound. The output consists of two maps: the segmented image and the uncertainty map of the segmentation. The uncertainty in the segmentation decisions is captured by the covariance matrix of the predictive distribution. We evaluate the proposed framework on medical image segmentation data from Magnetic Resonances Imaging and Computed Tomography scans. Our experiments on multiple benchmark datasets demonstrate that the proposed framework is more robust to noise and adversarial attacks as compared to state-of-the-art segmentation models. Moreover, the uncertainty map of the proposed framework associates low confidence (or equivalently high uncertainty) to patches in the test input images that are corrupted with noise, artifacts or adversarial attacks. Thus, the model can self-assess its segmentation decisions when it makes an erroneous prediction or misses part of the segmentation structures, e.g., tumor, by presenting higher values in the uncertainty map.