arXiv reaDer
NPBDREG:微分同相脳MRI登録のためのノンパラメトリックベイズ深層学習ベースのアプローチ
NPBDREG: A Non-parametric Bayesian Deep-Learning Based Approach for Diffeomorphic Brain MRI Registration
ディープニューラルネットワーク(DNN)ベースの画像レジストレーションアルゴリズムの不確実性の定量化は、実際の医療アプリケーションと研究指向の処理パイプラインの安全な展開、および一般化機能の向上に重要な役割を果たします。変分エンコーダ-デコーダアーキテクチャや推論時間ドロップアウトアプローチなど、現在利用可能な不確実性推定のアプローチでは、特定のネットワークアーキテクチャが必要であり、潜在空間のパラメトリック分布を想定しているため、予測される変形の事後分布の特性が最適ではなくなる可能性があります。 -田畑。 Adamオプティマイザーと確率勾配ランジュバンダイナミクス(SGLD)を組み合わせて、事後サンプリングを通じて真の事後分布を特徴付けることにより、教師なしDNNベースの変形可能な画像レジストレーションのための完全にノンパラメトリックなベイズフレームワークであるNPBDREGを紹介します。 NPBDREGは、真の事後分布を特徴付ける原理的なノンパラメトリック手法を提供します。これにより、理論的に十分に根拠があり、計算効率の高い方法で、不確実性の推定と信頼度の測定が改善されます。 4つの公的に利用可能なデータベース(MGH10、CMUC12、ISBR18、LPBA40)からの390の画像ペアを使用した脳MRI画像登録で、ベースラインの確率的VoxelMorph教師なしモデル(PrVXM)と比較したNPBDREGの付加価値を示しました。 NPBDREGは、PrVXMと比較して登録精度がわずかに向上し(ダイススコア0.73対0.68、p≪0.01)、混合構造ノイズによって破損したデータの一般化機能が向上します(たとえば、ダイススコア0.729対αの0.686)。 = 0.2)そして何よりもまず、予測された不確実性と分布外データとの有意に優れた相関関係(r> 0.95対r <0.5)。
Quantification of uncertainty in deep-neural-networks (DNN) based image registration algorithms plays an important role in the safe deployment of real-world medical applications and research-oriented processing pipelines, and in improving generalization capabilities. Currently available approaches for uncertainty estimation, including the variational encoder-decoder architecture and the inference-time dropout approach, require specific network architectures and assume parametric distribution of the latent space which may result in sub-optimal characterization of the posterior distribution for the predicted deformation-fields. We introduce the NPBDREG, a fully non-parametric Bayesian framework for unsupervised DNN-based deformable image registration by combining an Adam optimizer with stochastic gradient Langevin dynamics (SGLD) to characterize the true posterior distribution through posterior sampling. The NPBDREG provides a principled non-parametric way to characterize the true posterior distribution, thus providing improved uncertainty estimates and confidence measures in a theoretically well-founded and computationally efficient way. We demonstrated the added-value of NPBDREG, compared to the baseline probabilistic VoxelMorph unsupervised model (PrVXM), on brain MRI images registration using 390 image pairs from four publicly available databases: MGH10, CMUC12, ISBR18 and LPBA40. The NPBDREG shows a slight improvement in the registration accuracy compared to PrVXM (Dice score of 0.73 vs. 0.68, p ≪0.01), a better generalization capability for data corrupted by a mixed structure noise (e.g Dice score of 0.729 vs. 0.686 for α=0.2) and last but foremost, a significantly better correlation of the predicted uncertainty with out-of-distribution data (r>0.95 vs. r<0.5).
updated: Sun Aug 29 2021 06:51:53 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アソシエイト