この論文は、事前知識がトレーニングデータの形で利用可能である逆問題の画像化においてベイズ推定を実行するための新しい方法論を提案します。多様体仮説に従い、生成モデリングアプローチを採用して、周囲空間の部分多様体でサポートされるデータ駆動型事前分布を構築します。これは、変分オートエンコーダーまたは生成的敵対的ネットワークを使用してトレーニングデータから学習できます。簡単に検証できる条件下で、関連する事後分布と事後モーメントの存在と適切性を確立し、ベイズ推定量と不確実性の定量化分析の厳密な基盤を提供します。ベイジアン計算は、マニフォールドで事前調整されたクランクニコルソンアルゴリズムの並列強化バージョンを使用して実行されます。これは、これらのデータ駆動モデルの非凸性に対してエルゴードで堅牢であることが示されています。点推定と不確実性の定量化分析に加えて、モデルの仕様ミステストを導出して、データ駆動型の事前分布が信頼できない状況を自動的に検出し、トレーニングデータから潜在空間の次元を直接識別する方法を説明します。提案されたアプローチは、MNISTデータセットを使用した一連の実験で説明されており、最先端の代替画像再構成アプローチよりも優れています。モデルの精度分析は、データ駆動型モデルによって報告されたベイズ確率も、頻度確率の定義の下で非常に正確であることを示唆しています。
This paper proposes a new methodology for performing Bayesian inference in imaging inverse problems where the prior knowledge is available in the form of training data. Following the manifold hypothesis and adopting a generative modelling approach, we construct a data-driven prior that is supported on a sub-manifold of the ambient space, which we can learn from the training data by using a variational autoencoder or a generative adversarial network. We establish the existence and well-posedness of the associated posterior distribution and posterior moments under easily verifiable conditions, providing a rigorous underpinning for Bayesian estimators and uncertainty quantification analyses. Bayesian computation is performed by using a parallel tempered version of the preconditioned Crank-Nicolson algorithm on the manifold, which is shown to be ergodic and robust to the non-convex nature of these data-driven models. In addition to point estimators and uncertainty quantification analyses, we derive a model misspecification test to automatically detect situations where the data-driven prior is unreliable, and explain how to identify the dimension of the latent space directly from the training data. The proposed approach is illustrated with a range of experiments with the MNIST dataset, where it outperforms alternative image reconstruction approaches from the state of the art. A model accuracy analysis suggests that the Bayesian probabilities reported by the data-driven models are also remarkably accurate under a frequentist definition of probability.