Ke Rihuan
IEEE Trans Image Process. 2024;33:2908-2923. doi: 10.1109/TIP.2024.3355818. Epub 2024 Apr 16.
With recent deep learning based approaches showing promising results in removing noise from images, the best denoising performance has been reported in a supervised learning setup that requires a large set of paired noisy images and ground truth data for training. The strong data requirement can be mitigated by unsupervised learning techniques, however, accurate modelling of images or noise variances is still crucial for high-quality solutions. The learning problem is ill-posed for unknown noise distributions. This paper investigates the tasks of image denoising and noise variance estimation in a single, joint learning framework. To address the ill-posedness of the problem, we present deep variation prior (DVP), which states that the variation of a properly learnt denoiser with respect to the change of noise satisfies some smoothness properties, as a key criterion for good denoisers. Building upon DVP and under the assumption that the noise is zero mean and pixel-wise independent conditioned on the image, an unsupervised deep learning framework, that simultaneously learns a denoiser and estimates noise variances, is developed. Our method does not require any clean training images or an external step of noise estimation, and instead, approximates the minimum mean squared error denoisers using only a set of noisy images. With the two underlying tasks being considered in a single framework, we allow them to be optimised for each other. The experimental results show a denoising quality comparable to that of supervised learning and accurate noise variance estimates.
随着基于深度学习的方法在图像去噪方面显示出有前景的结果,在有监督学习设置中已报告了最佳去噪性能,该设置需要大量成对的噪声图像和真实数据用于训练。虽然无监督学习技术可以减轻对数据的强烈需求,但是对于高质量的解决方案而言,对图像或噪声方差进行精确建模仍然至关重要。对于未知的噪声分布,学习问题是不适定的。本文在一个单一的联合学习框架中研究图像去噪和噪声方差估计任务。为了解决该问题的不适定性,我们提出深度变分先验(DVP),即一个经过适当学习的去噪器相对于噪声变化的变化满足一些平滑特性,作为良好去噪器的关键标准。基于DVP并在噪声为零均值且在图像条件下像素-wise独立的假设下,开发了一个无监督深度学习框架,该框架同时学习一个去噪器并估计噪声方差。我们的方法不需要任何干净的训练图像或外部噪声估计步骤,相反,仅使用一组噪声图像来近似最小均方误差去噪器。在一个单一框架中考虑这两个基本任务时,我们允许它们相互优化。实验结果表明去噪质量与有监督学习相当,并且噪声方差估计准确。