提出一种收敛保证的通用图像修复方法,解决传统插件式算法不稳定问题。
Convergent Primal-Dual Plug-and-Play Image Restoration: A General Algorithm and Applications
- 结合原始对偶分裂法,构建可证明收敛的插件式图像修复框架。
- 在非二次数据保真项和约束条件下仍保持收敛性,适用于广泛修复任务。
- 理论严谨且计算高效,适合需要稳定性的实际图像恢复场景。
我们提出一种具有理论收敛保证的通用深度插件式(PnP)算法。尽管PnP策略通过利用高斯去噪器蕴含的强大先验,在多种图像修复任务中表现优异,但现有方法通常因设计上的随意性,在现实假设下缺乏理论收敛性保障,导致行为不一致。即使提供收敛性保证,也往往局限于特定场景或需高昂计算成本处理非二次数据保真项与额外约束,而这些正是许多图像修复任务的关键成分。为此,我们将PnP范式与原始对偶分裂(PDS)——一种高效求解广泛凸优化问题的近端分裂方法——相结合,开发出一个通用且收敛的PnP框架。具体而言,我们在合理假设下建立了所提PnP算法的收敛条件。此外,我们表明该算法所求解的问题并非标准凸优化问题,而是一个更一般的单调包含问题,并给出了其解集的数学表征。本方法能高效处理一大类图像修复问题,并具备理论收敛保证。在具体图像修复任务上的数值实验验证了理论结果的实际有效性。
原文摘要 · Abstract (English)
We propose a general deep plug-and-play (PnP) algorithm with a theoretical convergence guarantee. PnP strategies have demonstrated outstanding performance in various image restoration tasks by exploiting the powerful priors underlying Gaussian denoisers. However, existing PnP methods often lack theoretical convergence guarantees under realistic assumptions due to their ad-hoc nature, resulting in inconsistent behavior. Moreover, even when convergence guarantees are provided, they are typically designed for specific settings or require a considerable computational cost in handling non-quadratic data-fidelity terms and additional constraints, which are key components in many image restoration scenarios. To tackle these challenges, we integrate the PnP paradigm with primal-dual splitting (PDS), an efficient proximal splitting methodology for solving a wide range of convex optimization problems, and develop a general convergent PnP framework. Specifically, we establish theoretical conditions for the convergence of the proposed PnP algorithm under a reasonable assumption. Furthermore, we show that the problem solved by the proposed PnP algorithm is not a standard convex optimization problem but a more general monotone inclusion problem, where we provide a mathematical representation of the solution set. Our approach efficiently handles a broad class of image restoration problems with guaranteed theoretical convergence. Numerical experiments on specific image restoration tasks validate the practicality and effectiveness of our theoretical results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。