arXiv reaDer
nnU-Net の効率的なベイジアン不確実性推定
Efficient Bayesian Uncertainty Estimation for nnU-Net
自己構成型の nnU-Net は、さまざまな医用画像セグメンテーションの課題において優れたパフォーマンスを達成しています。これは、最適なモデルであり、医療画像セグメンテーションの強力なベースラインと広く考えられています。ただし、その並外れたパフォーマンスにもかかわらず、nnU-Net は、その障害の可能性を示すための不確実性の尺度を提供しません。これは、データが異種であり、nnU-Net が通知なしに失敗する可能性がある大規模な画像セグメンテーション アプリケーションでは問題になる可能性があります。この作業では、医用画像セグメンテーションの nnU-Net 不確実性を推定する新しい方法を紹介します。ベイズの不確実性推定のための重み空間の事後サンプリングのための非常に効果的なスキームを提案します。モンテカルロ ドロップアウトや平均場ベイジアン ニューラル ネットワークなどの以前のベースライン手法とは異なり、提案された手法は変分アーキテクチャを必要とせず、元の nnU-Net アーキテクチャをそのまま維持するため、優れたパフォーマンスと使いやすさが維持されます。さらに、マルチモーダル事後モデルを周辺化することで、元の nnU-Net よりもセグメンテーションのパフォーマンスを向上させます。心臓 MRI のパブリック ACDC および M&M データセットにこの方法を適用し、さまざまなベースライン方法で不確実性推定の改善を実証しました。提案された方法は、セグメンテーション精度と品質管理の両方の観点から、医用画像セグメンテーションの nnU-Net をさらに強化します。
The self-configuring nnU-Net has achieved leading performance in a large range of medical image segmentation challenges. It is widely considered as the model of choice and a strong baseline for medical image segmentation. However, despite its extraordinary performance, nnU-Net does not supply a measure of uncertainty to indicate its possible failure. This can be problematic for large-scale image segmentation applications, where data are heterogeneous and nnU-Net may fail without notice. In this work, we introduce a novel method to estimate nnU-Net uncertainty for medical image segmentation. We propose a highly effective scheme for posterior sampling of weight space for Bayesian uncertainty estimation. Different from previous baseline methods such as Monte Carlo Dropout and mean-field Bayesian Neural Networks, our proposed method does not require a variational architecture and keeps the original nnU-Net architecture intact, thereby preserving its excellent performance and ease of use. Additionally, we boost the segmentation performance over the original nnU-Net via marginalizing multi-modal posterior models. We applied our method on the public ACDC and M&M datasets of cardiac MRI and demonstrated improved uncertainty estimation over a range of baseline methods. The proposed method further strengthens nnU-Net for medical image segmentation in terms of both segmentation accuracy and quality control.
updated: Tue Apr 30 2024 06:05:26 GMT+0000 (UTC)
published: Mon Dec 12 2022 23:12:19 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アソシエイト