An extended Perona-Malik model based on probabilistic models
基于概率模型扩展的Perona-Malik模型
Lars M. Mescheder, Dirk A. Lorenz
AI总结 本文基于高斯尺度混合模型扩展了Perona-Malik模型,通过EM算法推导出滞后扩散算法,并改进其以更好地捕捉恢复中的不确定性,同时提出计算可行的放松方法,实验显示改进算法在恢复纹理区域和模糊边缘方面表现更优。
详情
Perona-Malik模型在从噪声输入中恢复图像方面非常成功。本文将该模型重新诠释为高斯尺度混合物的语言,并推导出一些扩展。具体来说,我们展示了将EM算法应用于高斯尺度混合物导致滞后扩散算法用于计算Perona-Malik扩散方程的稳态点。此外,我们展示了这些高斯尺度混合物的均场近似如何导致一种改进的滞后扩散算法,更准确地捕捉恢复中的不确定性。由于这种改进在实践中难以计算,我们提出对均场目标进行放松以使算法计算可行。我们的数值实验表明,这种改进的滞后扩散算法在恢复纹理区域和模糊边缘方面通常比未改进的算法表现更好。作为高斯尺度混合框架的第二个应用,我们展示了如何通过高效采样过程获得概率模型,使计算条件均值和其他期望在算法上可行。同样,所得到的算法与滞后扩散算法有很强的相似性。最后,我们展示了在相同框架下,通过离散边缘先验可以得到概率版本的Mumford-Shah分割模型。
The Perona-Malik model has been very successful at restoring images from noisy input. In this paper, we reinterpret the Perona-Malik model in the language of Gaussian scale mixtures and derive some extensions of the model. Specifically, we show that the expectation-maximization (EM) algorithm applied to Gaussian scale mixtures leads to the lagged-diffusivity algorithm for computing stationary points of the Perona-Malik diffusion equations. Moreover, we show how mean field approximations to these Gaussian scale mixtures lead to a modification of the lagged-diffusivity algorithm that better captures the uncertainties in the restoration. Since this modification can be hard to compute in practice we propose relaxations to the mean field objective to make the algorithm computationally feasible. Our numerical experiments show that this modified lagged-diffusivity algorithm often performs better at restoring textured areas and fuzzy edges than the unmodified algorithm. As a second application of the Gaussian scale mixture framework, we show how an efficient sampling procedure can be obtained for the probabilistic model, making the computation of the conditional mean and other expectations algorithmically feasible. Again, the resulting algorithm has a strong resemblance to the lagged-diffusivity algorithm. Finally, we show that a probabilistic version of the Mumford-Shah segementation model can be obtained in the same framework with a discrete edge-prior.