arXiv reaDer
確率的セグメンテーションネットワーク:空間的に相関する偶然性不確実性のモデリング
Stochastic Segmentation Networks: Modelling Spatially Correlated Aleatoric Uncertainty
画像の分割では、多くの場合、特定の入力に対して複数のもっともらしい解があります。たとえば、医療用画像処理では、オブジェクトの境界の正確な位置について専門家が意見を異にすることがよくあります。この固有の不確実性を推定し、複数のもっともらしい仮説を予測することは、多くのアプリケーションで大きな関心事ですが、この能力は現在のほとんどのディープラーニングメソッドには欠けています。このペーパーでは、確率的セグメンテーションネットワーク(SSN)、任意の画像セグメンテーションネットワークアーキテクチャで偶然不確実性をモデル化するための効率的な確率的手法を紹介します。ピクセルごとの推定値を生成するアプローチとは対照的に、SSNはラベルマップ全体の共同分布をモデル化するため、単一の画像に対して複数の空間的にコヒーレントな仮説を生成できます。ロジット空間で低ランクの多変量正規分布を使用して、画像が与えられたラベルマップの確率をモデル化することにより、基盤となるアーキテクチャを変更せずにニューラルネットワークで効率的に計算できる空間的に一貫した確率分布を取得します。 2D CTの肺結節や3DマルチモーダルMRIスキャンの脳腫瘍など、実際の医療データのセグメンテーションでメソッドをテストしました。 SSNは、あいまいな画像の相関する不確実性をモデリングするための最先端の性能よりもはるかにシンプルで、柔軟性が高く、効率的です。
In image segmentation, there is often more than one plausible solution for a given input. In medical imaging, for example, experts will often disagree about the exact location of object boundaries. Estimating this inherent uncertainty and predicting multiple plausible hypotheses is of great interest in many applications, yet this ability is lacking in most current deep learning methods. In this paper, we introduce stochastic segmentation networks (SSNs), an efficient probabilistic method for modelling aleatoric uncertainty with any image segmentation network architecture. In contrast to approaches that produce pixel-wise estimates, SSNs model joint distributions over entire label maps and thus can generate multiple spatially coherent hypotheses for a single image. By using a low-rank multivariate normal distribution over the logit space to model the probability of the label map given the image, we obtain a spatially consistent probability distribution that can be efficiently computed by a neural network without any changes to the underlying architecture. We tested our method on the segmentation of real-world medical data, including lung nodules in 2D CT and brain tumours in 3D multimodal MRI scans. SSNs outperform state-of-the-art for modelling correlated uncertainty in ambiguous images while being much simpler, more flexible, and more efficient.
updated: Tue Dec 22 2020 16:28:58 GMT+0000 (UTC)
published: Wed Jun 10 2020 18:06:41 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アソシエイト