论文标题

BigPrior:朝着解耦学习事先的幻觉和图像修复中的数据保真度

BIGPrior: Towards Decoupling Learned Prior Hallucination and Data Fidelity in Image Restoration

论文作者

Helou, Majed El, Süsstrunk, Sabine

论文摘要

经典的图像恢复算法使用多种先验,无论是隐式还是显式。他们的先验是手工设计的,相应的权重被启发。因此,深度学习方法通​​常会产生出色的图像恢复质量。但是,深层网络能够引起强大且几乎可以预测的幻觉。网络隐含地学会在先进的图像之前忠于观察到的数据;因此,不可能将原始数据和幻觉数据的分离。这限制了他们在图像恢复中广泛采用。此外,通常是幻觉的部分是降级模型过度拟合的受害者。 我们提出了一种基于网络优先的幻觉和数据保真度项的方法。我们将框架称为生成先验的贝叶斯整合(BigPrior)。我们的方法植根于贝叶斯框架,并紧密连接到经典的修复方法。实际上,它可以看作是大型经典恢复算法的概括。我们使用网络反转来从生成网络中提取图像先验信息。我们表明,在图像着色,介入和降解上,我们的框架始终改善反转结果。我们的方法虽然部分依赖于生成网络倒置的质量,但与最先进的监督和特定于任务的恢复方法具有竞争力。它还提供了一个额外的度量标准,该指标阐明了每个像素相对于数据保真度的先验依赖程度。

Classic image-restoration algorithms use a variety of priors, either implicitly or explicitly. Their priors are hand-designed and their corresponding weights are heuristically assigned. Hence, deep learning methods often produce superior image restoration quality. Deep networks are, however, capable of inducing strong and hardly predictable hallucinations. Networks implicitly learn to be jointly faithful to the observed data while learning an image prior; and the separation of original data and hallucinated data downstream is then not possible. This limits their wide-spread adoption in image restoration. Furthermore, it is often the hallucinated part that is victim to degradation-model overfitting. We present an approach with decoupled network-prior based hallucination and data fidelity terms. We refer to our framework as the Bayesian Integration of a Generative Prior (BIGPrior). Our method is rooted in a Bayesian framework and tightly connected to classic restoration methods. In fact, it can be viewed as a generalization of a large family of classic restoration algorithms. We use network inversion to extract image prior information from a generative network. We show that, on image colorization, inpainting and denoising, our framework consistently improves the inversion results. Our method, though partly reliant on the quality of the generative network inversion, is competitive with state-of-the-art supervised and task-specific restoration methods. It also provides an additional metric that sets forth the degree of prior reliance per pixel relative to data fidelity.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源