arXiv reaDer
プラグアンドプレイ分割ギブス サンプラー: ベイジアン推論に深い生成事前分布を埋め込む
Plug-and-Play split Gibbs sampler: embedding deep generative priors in Bayesian inference
このホワイト ペーパーでは、変数分割を利用して事後分布から効率的にサンプリングする、確率的プラグ アンド プレイ (PnP) サンプリング アルゴリズムを紹介します。分割ギブス サンプリング (SGS) に基づくアルゴリズムは、交互方向乗数法 (ADMM) から着想を得ています。事後サンプリングの困難なタスクを 2 つの単純なサンプリング問題に分割します。最初の問題は尤度関数に依存しますが、2 番目の問題は、深い生成モデルによって容易に実行できるベイジアン ノイズ除去問題として解釈されます。具体的には、説明の目的で、提案された方法は、最先端の拡散ベースの生成モデルを使用して、この論文で実装されています。その決定論的な PnP ベースの対応物と同様に、提案された方法は、事前に訓練された生成モデルにエンコードされる事前分布の明示的な選択を必要としないという大きな利点を示します。ただし、一般に点推定のみを提供する最適化手法 (PnP-ADMM など) とは異なり、提案されたアプローチでは、従来のベイジアン推定量に妥当な追加の計算コストで信頼区間を含めることができます。一般的に研究されている画像処理の問題に関する実験は、提案されたサンプリング戦略の効率を示しています。そのパフォーマンスは、最近の最先端の最適化およびサンプリング方法と比較されます。
This paper introduces a stochastic plug-and-play (PnP) sampling algorithm that leverages variable splitting to efficiently sample from a posterior distribution. The algorithm based on split Gibbs sampling (SGS) draws inspiration from the alternating direction method of multipliers (ADMM). It divides the challenging task of posterior sampling into two simpler sampling problems. The first problem depends on the likelihood function, while the second is interpreted as a Bayesian denoising problem that can be readily carried out by a deep generative model. Specifically, for an illustrative purpose, the proposed method is implemented in this paper using state-of-the-art diffusion-based generative models. Akin to its deterministic PnP-based counterparts, the proposed method exhibits the great advantage of not requiring an explicit choice of the prior distribution, which is rather encoded into a pre-trained generative model. However, unlike optimization methods (e.g., PnP-ADMM) which generally provide only point estimates, the proposed approach allows conventional Bayesian estimators to be accompanied by confidence intervals at a reasonable additional computational cost. Experiments on commonly studied image processing problems illustrate the efficiency of the proposed sampling strategy. Its performance is compared to recent state-of-the-art optimization and sampling methods.
updated: Fri Apr 21 2023 17:17:51 GMT+0000 (UTC)
published: Fri Apr 21 2023 17:17:51 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アソシエイト