arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2101.04442cs.CVeess.IV
收藏

野外联合去马赛克和去噪:真值不确定性下的训练案例

Joint Demosaicking and Denoising in the Wild: The Case of Training Under Ground Truth Uncertainty

The Hong Kong University of Science and Technology(香港科技大学)

Jierun Chen, Song Wen, S. -H. Gary Chan

摘要:

图像去马赛克和去噪是数字相机流水线中的两个关键基本步骤,旨在从噪声亮度读数中重建干净的彩色图像。在本文中,我们提出并研究了Wild-JDD,一种新的学习框架,用于在野外进行联合去马赛克和去噪。在以前的作品,一般假设训练数据的真值是一个完美的现实反映,我们认为在这里更常见的不完美的情况下,真值的不确定性在野外。我们首先说明其表现为各种各样的文物,包括拉链效应,彩色云纹和残留噪声。然后,我们制定了一个两阶段的数据退化过程,以捕捉这样的真值的不确定性,共轭先验分布的基础分布。在此之后,我们推导出一个证据下限(ELBO)损失训练一个神经网络,近似的参数的共轭先验分布条件下的退化输入。最后,为了进一步提高对分布外输入的性能,我们设计了一个简单但有效的微调策略,将输入作为弱信息先验。考虑到真值的不确定性,Wild-JDD在优化过程中具有良好的可解释性。大量的实验验证,它优于国家的最先进的计划联合去马赛克和去噪任务的合成和现实的原始数据集。

英文摘要:

Image demosaicking and denoising are the two key fundamental steps in digital camera pipelines, aiming to reconstruct clean color images from noisy luminance readings. In this paper, we propose and study Wild-JDD, a novel learning framework for joint demosaicking and denoising in the wild. In contrast to previous works which generally assume the ground truth of training data is a perfect reflection of the reality, we consider here the more common imperfect case of ground truth uncertainty in the wild. We first illustrate its manifestation as various kinds of artifacts including zipper effect, color moire and residual noise. Then we formulate a two-stage data degradation process to capture such ground truth uncertainty, where a conjugate prior distribution is imposed upon a base distribution. After that, we derive an evidence lower bound (ELBO) loss to train a neural network that approximates the parameters of the conjugate prior distribution conditioned on the degraded input. Finally, to further enhance the performance for out-of-distribution input, we design a simple but effective fine-tuning strategy by taking the input as a weakly informative prior. Taking into account ground truth uncertainty, Wild-JDD enjoys good interpretability during optimization. Extensive experiments validate that it outperforms state-of-the-art schemes on joint demosaicking and denoising tasks on both synthetic and realistic raw datasets.

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

↑