arXiv reaDer
事前自動エンコーディングを使用した結合事後最大化による逆問題の解決
Solving Inverse Problems by Joint Posterior Maximization with Autoencoding Prior
この作業では、事前に変分オートエンコーダー(VAE)が使用されている、イメージングにおける不適切な逆問題を解決する問題に対処します。具体的には、事前に一度トレーニングし、再トレーニングせずに多くの異なる対数凹劣化モデルに再利用できる分離されたケースを検討します。この問題に対する以前のMAPベースのアプローチは、高度に非凸最適化アルゴリズムにつながりますが、私たちのアプローチは、代替の最適化アルゴリズムに自然につながるジョイント(空間潜在)MAPを計算し、確率的エンコーダーを使用して計算を高速化します。結果として得られる手法(JPMAP)は、自動エンコード事前確率を使用してジョイント事後最大化を実行します。提案された目的関数が両凸に非常に近いという理論的および実験的証拠を示します。実際、それは、最適化スキームが停留点に収束することを保証するのに十分な弱い両凸特性を満たします。また、生成モデルの品質に影響を与えることなく、エンコーダーが分布外の画像に適切に一般化されるようにするために、ノイズ除去基準を使用してVAEを正しくトレーニングすることの重要性を強調します。この単純な変更は、手順全体に堅牢性を提供するための鍵です。最後に、共同MAP手法がより一般的なMAPアプローチにどのように関連するかを示し、JPMAPアルゴリズムを利用してより堅牢なMAP推定を提供する継続スキームを提案します。実験結果はまた、他の非凸MAPアプローチと比較して、JPMAPアプローチによって得られたソリューションの品質が高いことを示しています。これらのアプローチは、偽の局所最適点でスタックすることがよくあります。
In this work we address the problem of solving ill-posed inverse problems in imaging where the prior is a variational autoencoder (VAE). Specifically we consider the decoupled case where the prior is trained once and can be reused for many different log-concave degradation models without retraining. Whereas previous MAP-based approaches to this problem lead to highly non-convex optimization algorithms, our approach computes the joint (space-latent) MAP that naturally leads to alternate optimization algorithms and to the use of a stochastic encoder to accelerate computations. The resulting technique (JPMAP) performs Joint Posterior Maximization using an Autoencoding Prior. We show theoretical and experimental evidence that the proposed objective function is quite close to bi-convex. Indeed it satisfies a weak bi-convexity property which is sufficient to guarantee that our optimization scheme converges to a stationary point. We also highlight the importance of correctly training the VAE using a denoising criterion, in order to ensure that the encoder generalizes well to out-of-distribution images, without affecting the quality of the generative model. This simple modification is key to providing robustness to the whole procedure. Finally we show how our joint MAP methodology relates to more common MAP approaches, and we propose a continuation scheme that makes use of our JPMAP algorithm to provide more robust MAP estimates. Experimental results also show the higher quality of the solutions obtained by our JPMAP approach with respect to other non-convex MAP approaches which more often get stuck in spurious local optima.
updated: Tue Mar 02 2021 11:18:34 GMT+0000 (UTC)
published: Tue Mar 02 2021 11:18:34 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アソシエイト