arXiv reaDer
BayeSeg: 解釈可能な一般化可能性を備えた医用画像セグメンテーションのためのベイジアン モデリング
BayeSeg: Bayesian Modeling for Medical Image Segmentation with Interpretable Generalizability
多様な医用画像システムから引き起こされるクロスドメイン分布のシフトにより、多くのディープ ラーニング セグメンテーション手法は、目に見えないデータに対してうまく機能せず、現実世界への適用性が制限されます。最近の研究では、ドメイン一般化でドメイン不変表現を抽出する利点が示されています。ただし、ドメイン不変の機能の解釈可能性は依然として大きな課題です。この問題に対処するために、医療画像セグメンテーションのモデルの一般化可能性を高めるために、画像とラベル統計のベイジアン モデリングを通じて解釈可能なベイジアン フレームワーク (BayeSeg) を提案します。具体的には、最初に画像を空間相関変数と空間バリアント変数に分解し、階層的なベイズ事前確率を割り当てて、ドメイン安定形状とドメイン固有の外観情報をそれぞれモデル化するよう明示的に強制します。次に、セグメンテーションを、形状のみに関連する局所的に滑らかな変数としてモデル化します。最後に、変分ベイジアン フレームワークを開発して、これらの説明可能な変数の事後分布を推測します。このフレームワークはニューラル ネットワークで実装されるため、ディープ ベイジアン セグメンテーションと呼ばれます。前立腺セグメンテーションおよび心臓セグメンテーション タスクに関する定量的および定性的な実験結果は、提案された方法の有効性を示しています。さらに、事後を説明することによって BayeSeg の解釈可能性を調査し、さらなるアブレーション研究を通じて一般化能力に影響を与える特定の要因を分析しました。私たちのコードは、原稿が出版のために受け入れられると、https://zmiclab.github.io/projects.html 経由でリリースされます。
Due to the cross-domain distribution shift aroused from diverse medical imaging systems, many deep learning segmentation methods fail to perform well on unseen data, which limits their real-world applicability. Recent works have shown the benefits of extracting domain-invariant representations on domain generalization. However, the interpretability of domain-invariant features remains a great challenge. To address this problem, we propose an interpretable Bayesian framework (BayeSeg) through Bayesian modeling of image and label statistics to enhance model generalizability for medical image segmentation. Specifically, we first decompose an image into a spatial-correlated variable and a spatial-variant variable, assigning hierarchical Bayesian priors to explicitly force them to model the domain-stable shape and domain-specific appearance information respectively. Then, we model the segmentation as a locally smooth variable only related to the shape. Finally, we develop a variational Bayesian framework to infer the posterior distributions of these explainable variables. The framework is implemented with neural networks, and thus is referred to as deep Bayesian segmentation. Quantitative and qualitative experimental results on prostate segmentation and cardiac segmentation tasks have shown the effectiveness of our proposed method. Moreover, we investigated the interpretability of BayeSeg by explaining the posteriors and analyzed certain factors that affect the generalization ability through further ablation studies. Our code will be released via https://zmiclab.github.io/projects.html, once the manuscript is accepted for publication.
updated: Fri Mar 03 2023 04:48:37 GMT+0000 (UTC)
published: Fri Mar 03 2023 04:48:37 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アソシエイト