Developing efficient Bayesian computation algorithms for imaging inverse problems is challenging due to the dimensionality involved and because Bayesian imaging models are often not smooth. Current state-of-the-art methods often address these difficulties by replacing the posterior density with a smooth approximation that is amenable to efficient exploration by using Langevin Markov chain Monte Carlo (MCMC) methods. An alternative approach is based on data augmentation and relaxation, where auxiliary variables are introduced in order to construct an approximate augmented posterior distribution that is amenable to efficient exploration by Gibbs sampling. This paper proposes a new accelerated proximal MCMC method called latent space SK-ROCK (ls SK-ROCK), which tightly combines the benefits of the two aforementioned strategies. Additionally, instead of viewing the augmented posterior distribution as an approximation of the original model, we propose to consider it as a generalisation of this model. Following on from this, we empirically show that there is a range of values for the relaxation parameter for which the accuracy of the model improves, and propose a stochastic optimisation algorithm to automatically identify the optimal amount of relaxation for a given problem. In this regime, ls SK-ROCK converges faster than competing approaches from the state of the art, and also achieves better accuracy since the underlying augmented Bayesian model has a higher Bayesian evidence. The proposed methodology is demonstrated with a range of numerical experiments related to image deblurring and inpainting, as well as with comparisons with alternative approaches from the state of the art. An open-source implementation of the proposed MCMC methods is available from https://github.com/luisvargasmieles/ls-MCMC.
翻译:针对成像逆问题开发高效的贝叶斯计算方法具有挑战性,这源于问题的高维特性以及贝叶斯成像模型通常非光滑的特性。当前最先进的方法通常通过将后验密度替换为适用于Langevin马尔可夫链蒙特卡洛方法高效探索的光滑近似来解决这些困难。另一种替代方法基于数据增广与松弛技术,通过引入辅助变量构建适用于吉布斯采样高效探索的近似增广后验分布。本文提出一种新型加速近端MCMC方法——潜空间SK-ROCK(ls SK-ROCK),该方法紧密融合了上述两种策略的优势。此外,我们不再将增广后验分布视为原始模型的近似,而是将其看作该模型的推广。基于此,我们通过实验证明存在一个松弛参数取值范围,在该范围内模型精度得到提升,并提出了随机优化算法以自动识别给定问题的最优松弛量。在该参数区间内,ls SK-ROCK不仅比当前最先进的竞争方法收敛更快,同时由于基础增广贝叶斯模型具有更高的贝叶斯证据,其精度也更优。通过一系列图像去模糊与图像修复数值实验,以及与现有先进方法的对比,验证了所提出方法的有效性。所提出MCMC方法的开源实现可从 https://github.com/luisvargasmieles/ls-MCMC 获取。