arXiv reaDer
NPBDREG:ノンパラメトリックベイズ深層学習ベースのアプローチを使用した微分同相脳MRI登録の不確実性評価
NPBDREG: Uncertainty Assessment in Diffeomorphic Brain MRI Registration using a Non-parametric Bayesian Deep-Learning Based Approach
ディープニューラルネットワーク(DNN)ベースの画像レジストレーションアルゴリズムの不確実性の定量化は、外科的計画、術中ガイダンス、疾患の進行や治療効果の長期モニタリングなどの臨床アプリケーション向けの画像レジストレーションアルゴリズムの展開において重要な役割を果たします。研究指向の処理パイプラインで。 DNNベースの画像レジストレーションアルゴリズムで不確実性を推定するために現在利用可能なアプローチは、レジストレーション潜在空間の想定されるパラメトリック分布のレジストレーションステムの不確実性の推定が潜在的に不正確であるため、最適ではない臨床的意思決定をもたらす可能性があります。 NPBDREGは、DNNベースの変形可能な画像レジストレーションにおける不確実性推定のための完全にノンパラメトリックなベイズフレームワークを紹介します。したがって、分布外データの存在と高度に相関する不確実性の推定値を提供する可能性があります。 4つの公的に利用可能なデータベース(MGH10、CMUC12、ISBR18、LPBA40)からの390の画像ペアを使用した脳MRI画像レジストレーションで、ベースラインの確率的VoxelMorphモデル(PrVXM)と比較したNPBDREGの付加価値を示しました。 NPBDREGは、予測された不確実性と分布外データとのより良い相関(r>0.95対r<0.5)、および登録精度の7.3%の改善(ダイススコア、0.74対0.69、p≪0.01)を示しています。 )、および見当合わせの滑らかさの18%の改善(変形フィールドの折り目のパーセンテージ、0.014対0.017、p≪0.01)。最後に、NPBDREGは、ベースラインのPrVXMアプローチと比較して、混合構造ノイズによって破損したデータに対してより優れた一般化機能を示しました(ダイススコア0.73対0.69、p≪0.01)。
Quantification of uncertainty in deep-neural-networks (DNN) based image registration algorithms plays a critical role in the deployment of image registration algorithms for clinical applications such as surgical planning, intraoperative guidance, and longitudinal monitoring of disease progression or treatment efficacy as well as in research-oriented processing pipelines. Currently available approaches for uncertainty estimation in DNN-based image registration algorithms may result in sub-optimal clinical decision making due to potentially inaccurate estimation of the uncertainty of the registration stems for the assumed parametric distribution of the registration latent space. We introduce NPBDREG, a fully non-parametric Bayesian framework for uncertainty estimation in DNN-based deformable image registration by combining an Adam optimizer with stochastic gradient Langevin dynamics (SGLD) to characterize the underlying posterior distribution through posterior sampling. Thus, it has the potential to provide uncertainty estimates that are highly correlated with the presence of out of distribution data. We demonstrated the added-value of NPBDREG, compared to the baseline probabilistic VoxelMorph model (PrVXM), on brain MRI image registration using 390 image pairs from four publicly available databases: MGH10, CMUC12, ISBR18 and LPBA40. The NPBDREG shows a better correlation of the predicted uncertainty with out-of-distribution data (r>0.95 vs. r<0.5) as well as a 7.3%improvement in the registration accuracy (Dice score, 0.74 vs. 0.69, p ≪0.01), and 18% improvement in registration smoothness (percentage of folds in the deformation field, 0.014 vs. 0.017, p ≪0.01). Finally, NPBDREG demonstrated a better generalization capability for data corrupted by a mixed structure noise (Dice score of 0.73 vs. 0.69, p ≪0.01) compared to the baseline PrVXM approach.
updated: Thu Jun 09 2022 06:40:18 GMT+0000 (UTC)
published: Sun Aug 15 2021 16:00:06 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アソシエイト