arXiv reaDer
Uncertainty-aware GAN with Adaptive Loss for Robust MRI Image Enhancement
画像から画像への変換は、ソース画像とターゲット画像の間に一意の1対1のマッピングが存在しない可能性があるため、不適切な問題です。このコンテキストで提案された学習ベースの方法は、多くの場合、トレーニングデータと同様のテストデータのパフォーマンスを評価しますが、これは実用的でない場合があります。これには、特に医用画像などの重要な領域について、情報に基づいた意思決定を行うための予測の不確実性を定量化できる堅牢な方法が必要です。条件付き生成的敵対的ネットワーク(GAN)を採用した最近の研究では、ソース画像とターゲット画像の間の写実的な画像から画像へのマッピングを学習する際のパフォーマンスが向上していることが示されています。ただし、これらの方法は、(i)〜分布外(OOD)に対するモデルのロバスト性-ノイズの多いデータおよび(ii)〜不確実性の定量化に焦点を当てていません。この論文は、(i)残差にペナルティを課すために空間的に変化するノルムを自動的に調整するOODノイズの多いデータに対するロバスト性のための適応損失関数をモデル化し、(ii)予測におけるボクセルごとの不確実性を推定するGANベースのフレームワークを提案します。 。医用画像における2つの主要なアプリケーションで私たちの方法を示します:(i)〜アンダーサンプリングされた磁気共鳴画像(MRI)再構成(ii)〜MRIモダリティ伝播。 2つの異なる実世界のデータセットを使用した実験では、提案された方法が(i)OODノイズの多いテストデータに対して堅牢であり、精度が向上し、(ii)予測におけるボクセルレベルの不確実性を定量化することが示されています。
Image-to-image translation is an ill-posed problem as unique one-to-one mapping may not exist between the source and target images. Learning-based methods proposed in this context often evaluate the performance on test data that is similar to the training data, which may be impractical. This demands robust methods that can quantify uncertainty in the prediction for making informed decisions, especially for critical areas such as medical imaging. Recent works that employ conditional generative adversarial networks (GANs) have shown improved performance in learning photo-realistic image-to-image mappings between the source and the target images. However, these methods do not focus on (i)~robustness of the models to out-of-distribution (OOD)-noisy data and (ii)~uncertainty quantification. This paper proposes a GAN-based framework that (i)~models an adaptive loss function for robustness to OOD-noisy data that automatically tunes the spatially varying norm for penalizing the residuals and (ii)~estimates the per-voxel uncertainty in the predictions. We demonstrate our method on two key applications in medical imaging: (i)~undersampled magnetic resonance imaging (MRI) reconstruction (ii)~MRI modality propagation. Our experiments with two different real-world datasets show that the proposed method (i)~is robust to OOD-noisy test data and provides improved accuracy and (ii)~quantifies voxel-level uncertainty in the predictions.
updated: Thu Oct 07 2021 11:29:03 GMT+0000 (UTC)
published: Thu Oct 07 2021 11:29:03 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト