配布されていない (OOD) 医用画像が頻繁に発生します。たとえば、サイトやスキャナーの違い、または画像の破損が原因です。 OOD 画像には、不適切な画像セグメンテーションのリスクが伴い、下流の診断や治療に悪影響を及ぼす可能性があります。このような誤ったセグメンテーションに対するロバスト性を確保するために、画像セグメンテーションにおける認識 (モデル) と偶然 (データ) の不確実性を共同でモデル化するラプラシアン セグメンテーション ネットワーク (LSN) を提案します。空間的に相関するロジット分布でデータの不確実性を捉えます。モデルの不確実性については、高次元の出力を持つスキップ接続を持つ大規模なニューラル ネットワークにスケーリングする重み事後確率の最初のラプラス近似を提案します。経験的に、空間ピクセル相関をモデル化することで、ラプラシアン セグメンテーション ネットワークが画像内に現れる分布外オブジェクトに高い認識論的不確実性を割り当てることができることを実証します。
Out of distribution (OOD) medical images are frequently encountered, e.g. because of site- or scanner differences, or image corruption. OOD images come with a risk of incorrect image segmentation, potentially negatively affecting downstream diagnoses or treatment. To ensure robustness to such incorrect segmentations, we propose Laplacian Segmentation Networks (LSN) that jointly model epistemic (model) and aleatoric (data) uncertainty in image segmentation. We capture data uncertainty with a spatially correlated logit distribution. For model uncertainty, we propose the first Laplace approximation of the weight posterior that scales to large neural networks with skip connections that have high-dimensional outputs. Empirically, we demonstrate that modelling spatial pixel correlation allows the Laplacian Segmentation Network to successfully assign high epistemic uncertainty to out-of-distribution objects appearing within images.