arXiv reaDer
SUPER-Net: エンコーダー/デコーダー ネットワークにおける不確実性伝播を伴う信頼できる医用画像セグメンテーション
SUPER-Net: Trustworthy Medical Image Segmentation with Uncertainty Propagation in Encoder-Decoder Networks
ディープ ラーニング (DL) は、その精度、効率性、客観性により、ヘルスケア業界の再構築に大きな期待を寄せています。ただし、ノイズの多い分布外の入力に対する DL モデルの脆弱性は、診療所での展開を困難にしています。ほとんどのモデルは、モデルの不確実性や信頼性に関する追加情報なしで点推定値を生成します。このホワイト ペーパーでは、セグメンテーション ニューラル ネットワークにおける不確実性定量化のための新しいベイジアン DL フレームワークを紹介します。 SUPER-Net は、テイラー級数近似を使用して、非線形層全体のモデル パラメーターの事後分布の最初の 2 つのモーメント (平均と共分散) を解析的に伝播します。特に、SUPER-Net は、高価なポストホック モンテカルロ サンプリングやモデル アンサンブルを使用せずに、平均と共分散を同時に学習します。出力は、セグメント化された画像と、予測分布の共分散行列に対応するピクセルごとの不確実性マップの 2 つの同時マップで構成されます。さまざまなノイズの多い敵対的な条件下で、磁気共鳴イメージングおよびコンピューター断層撮影スキャンの医用画像セグメンテーションに関するSUPER-Netの広範な評価を実施します。複数のベンチマーク データセットに関する私たちの実験は、SUPER-Net が最先端のセグメンテーション モデルよりもノイズや敵対的攻撃に対してより堅牢であることを示しています。さらに、提案されたSUPER-Netの不確実性マップは、ノイズ、アーティファクト、または敵対的攻撃で破損したテスト入力画像のパッチに低い信頼性(または同等に高い不確実性)を関連付けます。おそらくもっと重要なことは、モデルは、特にノイズや敵対的な例のために誤った予測を行う場合に、セグメンテーションの決定の自己評価の能力を発揮することです。
Deep Learning (DL) holds great promise in reshaping the healthcare industry owing to 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 models produce point estimates without further information about model uncertainty or confidence. This paper introduces a new Bayesian DL framework for uncertainty quantification in segmentation neural networks: SUPER-Net: trustworthy medical image Segmentation with Uncertainty Propagation in Encoder-decodeR Networks. SUPER-Net analytically propagates, using Taylor series approximations, the first two moments (mean and covariance) of the posterior distribution of the model parameters across the nonlinear layers. In particular, SUPER-Net simultaneously learns the mean and covariance without expensive post-hoc Monte Carlo sampling or model ensembling. The output consists of two simultaneous maps: the segmented image and its pixelwise uncertainty map, which corresponds to the covariance matrix of the predictive distribution. We conduct an extensive evaluation of SUPER-Net on medical image segmentation of Magnetic Resonances Imaging and Computed Tomography scans under various noisy and adversarial conditions. Our experiments on multiple benchmark datasets demonstrate that SUPER-Net is more robust to noise and adversarial attacks than state-of-the-art segmentation models. Moreover, the uncertainty map of the proposed SUPER-Net associates low confidence (or equivalently high uncertainty) to patches in the test input images that are corrupted with noise, artifacts, or adversarial attacks. Perhaps more importantly, the model exhibits the ability of self-assessment of its segmentation decisions, notably when making erroneous predictions due to noise or adversarial examples.
updated: Sat Jan 21 2023 17:26:07 GMT+0000 (UTC)
published: Wed Nov 10 2021 22:46:05 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アソシエイト