医療画像セグメンテーションアプリケーションの不確実性を定量化することは、重要な意思決定に関連していることが多いため、不可欠です。画像セグメンテーションアーキテクチャの不確実性を定量化するために、たとえば入力画像を条件とする密度セグメンテーションモデルを学習するなど、説得力のある試みがなされてきました。この分野での典型的な作業は、これらの学習された密度を厳密にガウス分布に制限します。この論文では、正規化フロー(NF)を導入することにより、より柔軟なアプローチを使用することを提案します。これにより、学習された密度がより複雑になり、不確実性のより正確なモデリングが容易になります。確率的U-Netを採用し、事後密度をNFで補強して表現力を高めることにより、この仮説を証明します。マルチアノテーション付きおよびシングルアノテーション付きのLIDC-IDRIおよびKvasir-SEGセグメンテーションデータセットに対する定性的および定量的(GEDおよびIoU)評価は、それぞれ明らかな改善を示しています。これは、偶然性の不確実性の定量化と最大14%の予測パフォーマンスの向上で主に明らかです。この結果は、密度モデリングを通じてセグメンテーションのあいまいさを捉えようとするアーキテクチャでは、より柔軟な密度モデルを真剣に検討する必要があることを強く示しています。この改善されたモデリングの利点は、注釈とセグメンテーションに対する人間の信頼を高め、実際のテクノロジーの熱心な採用を可能にします。
Quantifying uncertainty in medical image segmentation applications is essential, as it is often connected to vital decision-making. Compelling attempts have been made in quantifying the uncertainty in image segmentation architectures, e.g. to learn a density segmentation model conditioned on the input image. Typical work in this field restricts these learnt densities to be strictly Gaussian. In this paper, we propose to use a more flexible approach by introducing Normalizing Flows (NFs), which enables the learnt densities to be more complex and facilitate more accurate modeling for uncertainty. We prove this hypothesis by adopting the Probabilistic U-Net and augmenting the posterior density with an NF, allowing it to be more expressive. Our qualitative as well as quantitative (GED and IoU) evaluations on the multi-annotated and single-annotated LIDC-IDRI and Kvasir-SEG segmentation datasets, respectively, show a clear improvement. This is mostly apparent in the quantification of aleatoric uncertainty and the increased predictive performance of up to 14 percent. This result strongly indicates that a more flexible density model should be seriously considered in architectures that attempt to capture segmentation ambiguity through density modeling. The benefit of this improved modeling will increase human confidence in annotation and segmentation, and enable eager adoption of the technology in practice.