arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01648stat.MLcs.CVcs.LGeess.IVmath.OC

通过基于自编码先验的联合后验最大化求解逆问题

Solving Inverse Problems by Joint Posterior Maximization with Autoencoding Prior

CNRS(法国国家科学研究中心)

Mario González, Andrés Almansa, Pauline Tan

AI总结:

本文提出JPMAP方法,通过计算联合(空间-潜在)MAP并使用去噪准则训练的VAE先验,高效求解成像中的不适定逆问题,避免非凸优化且收敛性更好,解的质量更高。

摘要:

在这项工作中,我们解决了成像中不适定逆问题的求解问题,其中先验是一个变分自编码器(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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑