arXiv reaDer
We Know Where We Don't Know: 3D Bayesian CNNs for Credible Geometric Uncertainty
 深層学習は、3Dコンピュータ断層撮影(CT)スキャンのセグメンテーションに正常に適用されています。これらのセグメンテーションの信頼性を確立するには、信頼できない予測を識別するために不確実性の定量化(UQ)が必要です。最近のUQアーキテクチャには、ディープガウスプロセスを近似するモンテカルロドロップアウトネットワーク(MCDN)と、重み空間の分布を学習するベイジアンニューラルネットワーク(BNN)が含まれます。 BNNはUQのMCDNよりも有利ですが、高次元では計算上不可能であると考えられており、どちらのアーキテクチャも解釈可能な幾何学的不確実性マップを作成していません。新しい3Dベイズ畳み込みニューラルネットワーク(BCNN)を提案します。これは、3Dデータに適用するための統計的に信頼できる幾何学的不確実性マップとスケールを生成する最初の深層学習法です。グラファイト電極とレーザー溶接金属のCTスキャンの実験結果を提示し、最近の不確実性メトリックでBCNNがMCDNよりも優れていることを示します。私たちのBCNNによって生成された幾何学的不確実性マップは、信頼区間として解釈可能なシグモイド値の分布をキャプチャします。これは、重要度の高い決定を深層学習に依存するアプリケーションにとって重要です。コードはで入手できます。
Deep learning has been successfully applied to the segmentation of 3D Computed Tomography (CT) scans. Establishing the credibility of these segmentations requires uncertainty quantification (UQ) to identify untrustworthy predictions. Recent UQ architectures include Monte Carlo dropout networks (MCDNs), which approximate deep Gaussian processes, and Bayesian neural networks (BNNs), which learn the distribution of the weight space. BNNs are advantageous over MCDNs for UQ but are thought to be computationally infeasible in high dimension, and neither architecture has produced interpretable geometric uncertainty maps. We propose a novel 3D Bayesian convolutional neural network (BCNN), the first deep learning method which generates statistically credible geometric uncertainty maps and scales for application to 3D data. We present experimental results on CT scans of graphite electrodes and laser-welded metals and show that our BCNN outperforms an MCDN in recent uncertainty metrics. The geometric uncertainty maps generated by our BCNN capture distributions of sigmoid values that are interpretable as confidence intervals, critical for applications that rely on deep learning for high-consequence decisions. Code available at
updated: Thu Apr 02 2020 00:35:12 GMT+0000 (UTC)
published: Wed Oct 23 2019 20:07:24 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト