機械学習モデルは通常、ペアの (入力、出力) データを使用して、エンド ツー エンドで監視された設定でトレーニングされます。例には、(低解像度、高解像度の) 画像のペアでトレーニングする最近の超解像メソッドが含まれます。ただし、これらのエンド ツー エンドのアプローチでは、入力 (たとえば、夜間の画像と昼光) または関連する潜在変数 (たとえば、カメラのブレや手の動き) の分布シフトがあるたびに再トレーニングが必要です。この作業では、最先端の (SOTA) 生成モデル (ここでは StyleGAN2) を活用して、強力な画像事前分布を構築します。これにより、多くの下流の再構築タスクにベイズの定理を適用できます。私たちの方法である生成モデルによるベイジアン再構成 (BRGM) は、単一の事前訓練された生成モデルを使用して、さまざまな画像復元タスク、つまり超解像とインペインティングをさまざまな前方破損モデルと組み合わせることで解決します。ジェネレーター モデルの重みを固定し、再構築された画像を生成した入力潜在ベクトルに対してベイズ最大事後確率 (MAP) 推定を推定することにより、画像を再構築します。さらに、変分推論を使用して、潜在ベクトル上の事後分布を近似し、そこから複数のソリューションをサンプリングします。 (i) Flick Faces High Quality データセットからの 60,000 枚の画像 (ii) MIMIC III からの 240,000 の胸部 X 線、および (iii) 7,329 回のスキャンによる 5 つの脳 MRI データセットの組み合わせコレクション。 3 つすべてのデータセットにわたって、データセット固有のハイパーパラメーター調整なしで、私たちのシンプルなアプローチは、超解像およびインペインティングに関する現在のタスク固有の最先端の方法に匹敵するパフォーマンスを実現し、より一般化可能でトレーニングを必要としません。 .ソース コードと事前トレーニング済みモデルは、オンラインで入手できます: https://razvanmarinescu.github.io/brgm/。
Machine learning models are commonly trained end-to-end and in a supervised setting, using paired (input, output) data. Examples include recent super-resolution methods that train on pairs of (low-resolution, high-resolution) images. However, these end-to-end approaches require re-training every time there is a distribution shift in the inputs (e.g., night images vs daylight) or relevant latent variables (e.g., camera blur or hand motion). In this work, we leverage state-of-the-art (SOTA) generative models (here StyleGAN2) for building powerful image priors, which enable application of Bayes' theorem for many downstream reconstruction tasks. Our method, Bayesian Reconstruction through Generative Models (BRGM), uses a single pre-trained generator model to solve different image restoration tasks, i.e., super-resolution and in-painting, by combining it with different forward corruption models. We keep the weights of the generator model fixed, and reconstruct the image by estimating the Bayesian maximum a-posteriori (MAP) estimate over the input latent vector that generated the reconstructed image. We further use variational inference to approximate the posterior distribution over the latent vectors, from which we sample multiple solutions. We demonstrate BRGM on three large and diverse datasets: (i) 60,000 images from the Flick Faces High Quality dataset (ii) 240,000 chest X-rays from MIMIC III and (iii) a combined collection of 5 brain MRI datasets with 7,329 scans. Across all three datasets and without any dataset-specific hyperparameter tuning, our simple approach yields performance competitive with current task-specific state-of-the-art methods on super-resolution and in-painting, while being more generalisable and without requiring any training. Our source code and pre-trained models are available online: https://razvanmarinescu.github.io/brgm/.