arXiv reaDer
評価者固有のベイジアン ニューラル ネットワークを介した医療画像セグメンテーションにおける評価者間不確実性の定量化
Inter-Rater Uncertainty Quantification in Medical Image Segmentation via Rater-Specific Bayesian Neural Networks
自動医療画像セグメンテーションには本質的にある程度の不確実性が伴います。この不確実性に寄与する重要な要因の 1 つは、主に画像の外観の変化により、対象となるターゲット領域の境界を決定する際に生じる可能性がある曖昧さです。これに加えて、この分野の専門家の間でも、特定の解剖学的構造の正確な定義に関して異なる意見が現れる可能性があります。この研究は、評価者間不確実性として知られるセグメンテーションの不確実性のモデリングに特に取り組んでいます。その主な目的は、医療画像の複数の専門家が同じ画像を解釈して注釈を付けるときに発生する可能性のあるセグメンテーション結果のばらつきを調査および分析することです。医療画像セグメンテーションにおける評価者間の不確実性を推定するための、新しいベイジアン ニューラル ネットワーク ベースのアーキテクチャを導入します。私たちのアプローチには 3 つの重要な進歩があります。まず、不確実性推定に特化した 1 つのエンコーダとマルチ デコーダのアーキテクチャを導入し、関係する各専門家の評価者固有の表現を取得できるようにします。次に、新しいアーキテクチャにベイジアン モデリングを提案します。これにより、特にアノテーションが限られたシナリオで評価者間分布を効率的に取得できるようになります。最後に、アテンション モジュールを各デコーダに統合することで、評価者固有の表現を強化します。このモジュールは、各評価者にとって焦点を絞った洗練されたセグメンテーション結果を容易にします。当社は、技術革新を厳密に検証するために、合成データセットと現実世界のデータセットを使用して広範な評価を実施します。私たちの手法は、さまざまな不確実性の側面を含む 2 つの評価指標を考慮して、公開されている QUBIQ データセットに関する 7 つの多様なタスクのうち 5 つにおいて既存のベースライン手法を上回っています。コード、モデル、および新しいデータセットは、GitHub リポジトリ: https://github.com/HaoWang420/bOEMD-net から入手できます。
Automated medical image segmentation inherently involves a certain degree of uncertainty. One key factor contributing to this uncertainty is the ambiguity that can arise in determining the boundaries of a target region of interest, primarily due to variations in image appearance. On top of this, even among experts in the field, different opinions can emerge regarding the precise definition of specific anatomical structures. This work specifically addresses the modeling of segmentation uncertainty, known as inter-rater uncertainty. Its primary objective is to explore and analyze the variability in segmentation outcomes that can occur when multiple experts in medical imaging interpret and annotate the same images. We introduce a novel Bayesian neural network-based architecture to estimate inter-rater uncertainty in medical image segmentation. Our approach has three key advancements. Firstly, we introduce a one-encoder-multi-decoder architecture specifically tailored for uncertainty estimation, enabling us to capture the rater-specific representation of each expert involved. Secondly, we propose Bayesian modeling for the new architecture, allowing efficient capture of the inter-rater distribution, particularly in scenarios with limited annotations. Lastly, we enhance the rater-specific representation by integrating an attention module into each decoder. This module facilitates focused and refined segmentation results for each rater. We conduct extensive evaluations using synthetic and real-world datasets to validate our technical innovations rigorously. Our method surpasses existing baseline methods in five out of seven diverse tasks on the publicly available QUBIQ dataset, considering two evaluation metrics encompassing different uncertainty aspects. Our codes, models, and the new dataset are available through our GitHub repository: https://github.com/HaoWang420/bOEMD-net .
updated: Fri Aug 25 2023 05:09:32 GMT+0000 (UTC)
published: Wed Jun 28 2023 20:52:51 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アソシエイト