aam Advances in Applied Mathematics 2324-7991 2324-8009 beplay体育官网网页版等您来挑战! 10.12677/aam.2024.136262 aam-89755 Articles 数学与物理 基于深度图像先验的椒盐噪声图像去噪
Salt-and-Pepper Noise Image Denoising Based on Deep Image Prior
胡锦华 赵泽华 宋苏奇 许建楼 河南科技大学数学与统计学院,河南 洛阳 06 06 2024 13 06 2734 2741 24 5 :2024 19 5 :2024 19 6 :2024 Copyright © 2024 beplay安卓登录 All rights reserved. 2024 This work is licensed under the Creative Commons Attribution International License (CC BY). http://creativecommons.org/licenses/by/4.0/ 为了有效地去除图像中的椒盐噪声,本文利用低秩和深度图像先验,提出了一种基于加权核范数的优化模型。为了有效地求解优化模型,本文利用双线性分解,采用交替方向法将原问题分解成几个优化的子问题,对每个子问题给出相应的优化算法。数值实验表明,相比其它先进的方法,假设的新模型取得更好的去噪效果。
To effectively remove salt-and-pepper noise from images, this paper proposes an optimization model based on weighted nuclear norm with the prior knowledge of low rank and depth image prior. To efficiently solve the optimization model, the paper utilizes bilinear decomposition and employs the Alternating Direction Method to decompose the original problem into several optimized sub-problems, for each of which corresponding optimization algorithms are provided. Numerical experiments demonstrate that compared to other advanced methods, the proposed new model achieves better denoising results.
图像去噪,椒盐噪声,加权核范数,交替方向法,深度图像先验
Image Denoising
Salt-and-Pepper Noise Weighted Nuclear Norm Alternating Direction Method Deep Image Prior
1. 引言

随着网络的快速发展和通讯技术的更新换代,人们可以通过视频、文字、语音、图像等多种途径获取信息。相较于通过语音与文字等传统方式所获取的信息,利用图像进行信息传递则具有更加直接且全面的优势。由此可得,图像是一种重要的信息载体,通过图像处理可以帮助人们深入了解信息的内涵。然而,图像中常常存在各种噪声,这些噪声会干扰人们对图像的准确理解。因此,图像去噪的任务就是消除图像中的噪声,以提高人们对图像的认知水平,为后续处理提供更好的基础 [1]

图像复原其目的是从所得到的带噪图像中恢复出原图像即求原图像的某种最优意义下的估计值。在图像去噪的过程中,其核心任务便是要从已经遭受污染的图像中,成功清除掉那些无用的噪声,同时还要尽可能地保留住原始图像中所包含的所有重要特征 [2] 。图像去噪的方法分为两大类:基于传统的数学模型的方法和基于深度学习的方法。在传统的图像处理方法中,由于自然图像的低阶特性,基于低秩矩阵恢复方法得到广泛应用,旨在从退化的图像观测中找到潜在的低维子空间。

在机器学习 [3] 的领域中,出现许多优秀的降维方法。例如主成分分析算法(Principle Component Analysis, PCA)、线性判别分析、非负矩阵分解等等。其中,PCA是非常典型的线性降维方法 [4] 。值得注意的是,PCA在对高维数据进行降维处理时,会尽可能地保留住原始数据中的重要特征,并且还能揭示出高维数据矩阵背后所隐藏的低维结构。但是对于一些严重离群的噪声来说,PCA方法往往无法清除图像数据中的噪声。因此Wright等人 [5] 通过改良PCA方法在鲁棒性方面的缺陷,提出了一个改进模型——鲁棒主成分分析模型(Robust Principal Component Analysis, RPCA),该模型如下:

min A , E r a n k ( A ) + λ E 0 , s .t . D = A + E ,(1)

在上公式中,D为数据矩阵,由结构矩阵A和噪声矩阵E组成, 0 为矩阵的 L 0 范数, λ 为权衡因子,其取值大于0。式(1)是一个非凸的NP-hard问题,为了有效地求解式(1),Wright等人 [5] 对(1)式进行凸松弛,研究如下模型:

min A , E A + λ E 1 , s .t . D = A + E ,(2)

其中 1 为矩阵的 L 1 范数, 为矩阵的核范数。

对于RPCA模型的求解算法,众多研究人员投身其中,提出了增广拉格朗日乘子法(Augmented Lagrange Multipliers, ALM) [6] 、加速近端梯度算法法 [7] 、迭代阈值算法 [8] 等。运用以上算法时,需要求解核范数最小化问题且进行奇异值分解(Singular Value Decomposition, SVD) [9]

随着对该问题的研究发展,人们发现加权核范数相较于核范数来说,考虑到了矩阵中每个奇异值所代表的数据信息具有不同的重要程度,因此加权核范数的应用给不同的奇异值分配不一样的阈值,更能体现图像数据中的差异,基于此问题,Gu等人 [10] 将式(2)中的核范数替换为加权核范数,构建了加权核范数优化模型,具体模型如下:

min A , E A w , + λ E 1 , s .t . D = A + E ,(3)

其中, w , 为矩阵的加权核范数。该模型对矩阵中的每个奇异值调配各自的权值,自适应调整矩阵奇异值的目的,弥补了核范数的不足,使得通过模型求解的矩阵更具低秩性。

除了基于低秩先验的方法外,基于深度学习的方法在图像恢复问题中也引起了广泛的关注。在文献 [11] 中,设计了一种深度卷积神经网络(Convolutional Neural Network, CNN)来从噪声观测中恢复多通道图像。它能够在不破坏图像高维结构的情况下学习多通道滤波器。Xu等人 [12] 采用张量核范数(TNN)来表征这种张量图像数据集的全局结构,提出了一个隐式正则化器来插入卷积神经网络去噪器,该方法能够很好地恢复图像的整体结构和细节。然而,图像结构表示的有效性主要取决于训练数据的数量和质量。在一些实际应用中,如高光谱、遥感和医学图像恢复,训练数据可能是有限的,甚至无法收集。因此Ulyanov等人 [13] 提出了一种无监督深度学习方法,称为深度图像先验(deep image prior, DIP)用于图像恢复。该项模型的最大优点是不需要大量的样本,只用一张带噪声的图像利用卷积神经网络的自适应的学习能力实现了一个可以生成图片的滤波器组,该方法通常比基于低秩先验的方法更能代表图像细节。然而,在被异常值和缺失条目污染的挑战性场景下,由于网络对图像观测中被污染的分量过度拟合,其性能会在过度迭代下下降。虽然单独使用低秩先验可以表示图像对噪声和离群值的全局相关性,但必须在噪声和离群值的鲁棒性以及图像细节的保存之间进行权衡。因此,文献 [14] 将深度图像先验与低秩先验相互插入并应用到图像恢复中,深度先验信息可以通过预定于的卷积神经网络隐式捕获,其网络参数通过拟合退化图像进行迭代优化,而不是从训练数据中学习,提出了基于深度先验的重加权低秩矩阵分解模型(Reweighted low-rank Matrix Factorization with Deep prior, RMFD)。

虽然RMFD模型结合了图像低秩先验和深度先验,继承了二者的优势,从而获得较好的恢复精度,但是其中的DIP性能任有提升空间,且在求解RMFD模型时,作者采用了相同的正则化参数 μ ,影响了模型求解的恢复精度,针对以上问题,本文提出了一个基于加权核范数的优化模型,引入一个显式变分先验项,为图像处理提供更多信息,提高DIP的性能,并设置不同的正则化参数 μ 1 , μ 2 , μ 3 。实验结果表明提出的新模型取得较好的效果。

2. 本文的新模型和算法 2.1. 基础理论

定理1 [14] :对于任意秩 ( X ) r 的矩阵 X n 1 × n 2 ,且可以分解为 X = U V U n 1 × r V r × n 2 ,有

X w , = min U , V ; X = U V 1 2 ( U w , F 2 + V w , F 2 ) ,(4)

其中 U w , F ( i = 1 min { n 1 , r } w i σ i 2 ( U ) ) 1 / 2 V w , F ( i = 1 min { n 1 , r } w i σ i 2 ( V ) ) 1 / 2 。对于权值w的选择,对 U w , F 2 V w , F 2 采用下式:

w i = c ( σ i ( U k ) σ i ( V k ) + ε ) p 1 ,(5)

而对于 X w , 的权值w选择来说,采用下式:

w i = c ( σ i ( X ) + ε ) p 1 ,(6)

在式(5)和式(6)中 c > 0 0 < p < 1 均为一个常数。

定义1 [15] :设矩阵 X n 1 × n 2 的秩为r,其奇异值分解为

X = U r Λ r V r T ,(7)

其中 U r m × r V r n × r 均为正交矩阵; Λ r = d i a g ( σ 1 , σ 2 , , σ r ) σ i 为矩阵X的奇异值且满足 0 < σ r σ 2 σ 1 。对于 τ > 0 ,定义

D τ ( X ) = U r d i a g ( max ( 0 , σ i τ ) ) V r ,(8)

D τ 为奇异值阈值收缩算子。

定理2 [15] :对 τ > 0 ,矩阵 Y m × n ,奇异值阈值收缩算子满足

D τ ( Y ) = arg min X τ X + 1 2 X Y F 2 。(9)

2.2. 提出的新模型

Ulyanov等人 [13] 将深度图像先验引入到图像处理模型中,构建了最小化问题的组合模型,模型如下:

arg min θ 1 2 H f θ ( z ) g 2 2 (10)

在上式中 f θ ( · ) 表示一个CNN生成器, g n 表示一个自然图像,z为一个随机输入的向量。

在文献 [16] 中,作者基于式(10),在该模型中增加了全变分先验项来提高深度图像先验的性能,具体模型如下:

arg min θ 1 2 H f θ ( z ) g 2 2 + i = 1 N μ i ( D f θ ( z ) ) i 2 (11)

同时,作者采用交替方向乘子法(Alternating direction multiplier method, ADMM)代替标准的次梯度方法。由于ADMM的模块化结构允许通过简单地修改正则相关的子步骤来嵌入任何先验(显式或隐式)信息,所以ADMM框架更加灵活。同时,带有显示先验项的数学模型在数字图像处理中具有多重优势。首先,它有助于减少图像中的噪声和模糊,通过约束恢复过程,使结果更接近真实场景。其次,显示先验项提高了图像恢复的质量,突出边缘和细节,从而增强了图像的可视化效果。此外,引入先验信息可以改善图像分割的准确性和稳定性,加速算法的收敛速度,并节省计算时间。

随着深度图像先验的发展,相比于传统图像先验,该技术能为计算机提供更加丰富图像信息,使得模型回复效果更佳,由此文献 [14] 深度图像先验与低秩先验相互插入并应用到图像恢复中,提出了基于深度先验的重加权低秩矩阵分解模型,具体模型如下:

min θ λ f θ ( z 0 ) w , + f θ ( z 0 ) Y 1 ,(12)

在上式中, f θ ( · ) 表示一个CNN生成器,其具有单个输出通道。 θ 为网络参数, λ > 0 是一个正则化参数 z 0 n 1 × n 2 为随机网络输入, Y n 1 × n 2 表示一个自然图像。

为了保护图像边缘,受到文献 [16] 的启发,本文假设新的去噪模型如下:

min θ λ f θ ( z 0 ) w + f θ ( z 0 ) Y 1 + α D f θ ( z 0 ) 1 ,(13)

其中, f θ ( · ) 表示一个CNN生成器, θ 为网络参数, z 0 n 1 × n 2 为随机网络输入, D f θ ( z 0 ) i = ( ( D h f θ ( z 0 ) , D v f θ ( z 0 ) ) i ) i = 1 , , n f θ ( z 0 ) 在像素i处计算的离散梯度, D h f θ ( z 0 ) D v f θ ( z 0 ) 分别为沿水平和竖直方向的一阶有限差分离散算子, Y n 1 × n 2 表示一个自然图像, λ , α > 0 表示正则化参数。相较于模型(12),该模型增加了显示变分先验项,该先验能够保护图像的边缘和细节,新模型对噪声图像的处理更加优秀,在视觉上使恢复的图像更加的自然平滑。在数值上,处理椒盐噪声图像时,该模型恢复图像的PSNR值较原模型中的恢复图像的PSNR值更高。总的来说,该模型继承了原模型采用深度图像先验的优势,同时,在显式先验项的帮助下,该模型恢复图像的效果更优。

2.3. 模型的算法

对于式(13),令 X = f θ ( z 0 ) = U V ,利用定理1,可得到下式:

min U , V , θ λ 2 ( U w , F 2 + V w , F 2 ) + f θ ( z 0 ) Y 1 + α D f θ ( z 0 ) 1 .(14)

对于上式,令 U = U ^ V = V ^ U V = f θ ( z 0 ) ,有增广拉格朗日方法得:

min U ^ , U , V ^ , V , θ λ 2 ( U ^ w , F 2 + V ^ w , F 2 ) + f θ ( z 0 ) Y 1 + α D f θ ( z 0 ) 1 + μ 1 2 U ^ U + Z 1 μ 1 F 2 + μ 2 2 V ^ V + Z 2 μ 2 F 2 + μ 3 2 U V f θ ( z 0 ) + Z 3 μ 3 F 2 (15)

其中, Z 1 , Z 2 , Z 3 均为拉格朗日参数, μ 1 , μ 2 , μ 3 均为正标量。

对式(15)采用交替方法可得:

U k + 1 = arg min U μ 1 2 U ^ k U + Z 1 k μ 1 F 2 + μ 3 2 U V k f θ k ( z 0 ) + Z 3 k μ 3 F 2 (16)

V k + 1 = arg min V μ 2 2 V ^ k V + Z 2 k μ 2 F 2 + μ 3 2 U k + 1 V f θ k ( z 0 ) + Z 3 k μ 3 F 2 (17)

U ^ k + 1 = arg min U ^ λ 2 U ^ w , F 2 + μ 1 2 U ^ U k + 1 + Z 1 k μ 1 F 2 (18)

V ^ k + 1 = arg min V ^ λ 2 V ^ w , F 2 + μ 2 2 V ^ V k + 1 + Z 2 k μ 2 F 2 (19)

θ k + 1 = arg min θ f θ ( z 0 ) Y 1 + α D f θ ( z 0 ) 1 + μ 3 2 U k + 1 V k + 1 f θ ( z 0 ) + Z 3 k μ 3 F 2 (20)

Z 1 k + 1 = Z 1 k + μ 1 ( U ^ k + 1 U k + 1 ) (21)

Z 2 k + 1 = Z 2 k + μ 2 ( V ^ k + 1 V k + 1 ) (22)

Z 3 k + 1 = Z 3 k + μ 3 ( U k + 1 V k + 1 f θ k + 1 ( z 0 ) ) (23)

对于上述讨论,详细的算法如下:

算法1 交替方向法去除椒盐噪声

输入:输入参数 λ , μ 1 , μ 2 , μ 3 ,误差 ε 0 < p < 1 c = 1

1. 初始化: Z 1 0 , Z 2 0 , Z 3 0 , t , U 0 = U ^ 0 , V 0 = V ^ 0

2. 通过式(16)和式(17)更新 U k + 1 V k + 1

3. 通过式(5)更新 w i = c ( σ i ( U ^ k ) σ i ( V ^ k ) + ε ) p 1 i = 1 , , r

4. 通过式(18)更新 U ^ k + 1

5. 通过式(5)更新 w i = c ( σ i ( U ^ k + 1 ) σ i ( V ^ k ) + ε ) p 1 i = 1 , , r

6. 通过式(19)更新 V ^ k + 1

7. 通过式(20)更新 θ k + 1

8. 分别通过(21)、(22)、(23)式更新 Z 1 k + 1 , Z 2 k + 1 , Z 3 k + 1

9. 达到终止条件输出 f θ k + 1 ( z 0 ) ,否者重复上边步骤进入循环

3. 数值实验

在本节中,给出了一些数值实验的结果,旨在评估所提出的方法在图像去噪问题上的有效性。采用10DB的椒盐噪声图对应的噪声进行测试,评价指标采用峰值信噪比(PSNR)。分析对比RMFD模型 [11] 以及本文的模型。在实验过程中所有的模型通过调整算法中的参数以达到较好的去噪效果。

本节实验使用的图像数据集为Miscellaneous volume of the USC-SIPI dataset [17] ,数据集中每张图像的尺寸均为256 × 256。下图为四幅示例图片的去噪示例图,从左至右图像依次为Green pepper、Forest、Women、Snow mountain。第一行中的四幅图像均为RMFD模型去噪结果图,第二行中的四幅图像均为本文模型去噪结果图。从 图1 中可以看出,在Women示例图中RMFD模型去噪效果不是很理想,图像较暗,以及较为模糊的效果,对图像高光部分的回复效果不如本文模型。在Forest示例图中可以看出,RMFD模型对于树干中的明暗关系过渡效果较差,不如本文模型对于树叶以及树干的回复效果,这说明新模型可以更好的保留树干图像的细节信息。从总体上看,本文使用的新模型在去除图像的椒盐噪声时,可以更好地保护图像中的边缘以及高光效果。同时从 表1 中可以看出本文模型的PSNR高于RMFD模型,且实验结果良好,说明了本文所使用的新模型对于图像去噪的有效性。

为了进一步说明本文模型在图像去噪方面的优势,以Women图像为例,给出PSNR随迭代次数的变化曲线图。其中迭代次数为横坐标,PSNR值为纵坐标见 图2 。总体上看,图像PSNR值总体呈上升趋势,在迭代次数的前50次中,恢复图像的PSNR值有较大的增长速度,说明本文模型具有较快的图像恢复速度,并且在迭代次数达到100次时所恢复图像的PSNR值已经超过25.0,具有较好的恢复效果。在400次到1000次迭代过程中,PSNR值在30.0上下浮动,且浮动变化较小,说明该模型在恢复去噪图像中具有较好的稳定性。

<xref></xref>Table 1. PSNR under different models of grayscale plotsTable 1. PSNR under different models of grayscale plots 表1. 灰度图在不同模型下的PSNR

测试图像

Green pepper

Forest

Women

Snow mountain

RMFD模型

28.5962

24.5575

29.6548

24.6758

本文模型

30.6482

26.1869

30.2984

25.7169

图1. 不同算法去噪效果。(a) RMFD去噪结果;(b) 本文算法去噪结果

Figure 2. The updated PSNR value for the noisy image (Woman)--图2. 噪声图片(妇女)迭代更新的PSNR值--

从上面的实验可以看出,相较于RMFD模型,本文方法在去除噪声的同时,也将图像的细节纹理及其边缘保护得更好,并且由实验数据可得本文的图像结果PSNR值更优,从折线图中看出本文模型具有较好的稳定性,更好地证明了本文模型的改进的成功性。

4. 结论

在本文中提出了一种基于加权核范数的优化模型,在模型中引入了深度图像先验与低秩先验,利用双线性分解,采用交替方向法将原问题分解成几个优化的子问题,对每个子问题给出相应的优化算法,并且添加了一个显示变分先验项提高DIP的性能,从而相对于原模型更灵活,对图像细节部分回复效果更优。数值实验表明,相比原模型的算法,新模型取得更有效的去噪效果以及较好的稳定性。

基金项目

本课题由河南省科技攻关计划项目(编号:232102210111);河南科技大学大学生训练计划项目资助(编号:2023221)。

NOTES

*通讯作者。

References 阮秋琦. 数字图像处理学[M]. 北京: 电子工业出版社, 2001. 蒋刚毅, 黄大江, 王旭, 等. 图像质量评价方法研究进展[J]. 电子与信息学报, 2010, 32(1): 219-226. 闫友彪, 陈元琰. 机器学习的主要策略综述[J]. 计算机应用研究, 2004, 21(7): 4-10, 13. Martinez, A.M. and Kak, A.C. (2001) PCA versus LDA. IEEE Transactions on Pattern Analysis and Machine Intelligence, 23, 228-233. >https://doi.org/10.1109/34.908974 Candès, E.J., Li, X., Ma, Y. and Wright, J. (2011) Robust Principal Component Analysis? Journal of the ACM, 58, 1-37. >https://doi.org/10.1145/1970392.1970395 Lin, Z., Chen, M. and Ma, Y. (2010) The Augmented Lagrange Multiplier Method for Exact Recovery of Corrupted Low-Rank Matrices. arXiv: 1009.5055. Chen, M., Ganesh, A., Lin, Z., et al. (2009) Fast Convex Optimization Algorithms for Exact Recovery of a Corrupted Low-Rank Matrix. Journal of the Marine Biological Association of the UK, 56, 707-722. 梁栋, 梁昭, 鲍文霞, 等. 基于非局部正则化稀疏表示的图像去噪算法[J]. 系统工程与电子计术, 2013, 35(5): 1104-1109.4 史加荣, 郑秀云, 魏宗田, 等. 低秩矩阵恢复算法综述[J]. 计算机应用研究, 2013, 30(6): 1601-1605. Gu, S., Zhang, L., Zuo, W. and Feng, X. (2014). Weighted Nuclear Norm Minimization with Application to Image Denoising. 2014 IEEE Conference on Computer Vision and Pattern Recognition, Columbus, 23-28 June 2014, 2862-2869. >https://doi.org/10.1109/cvpr.2014.366 Chang, Y., Yan, L., Fang, H., Zhong, S. and Liao, W. (2019) Hsi-denet: Hyperspectral Image Restoration via Convolutional Neural Network. IEEE Transactions on Geoscience and Remote Sensing, 57, 667-682. >https://doi.org/10.1109/tgrs.2018.2859203 Zhao, X., Xu, W., Jiang, T., Wang, Y. and Ng, M.K. (2020) Deep Plug-And-Play Prior for Low-Rank Tensor Completion. Neurocomputing, 400, 137-149. >https://doi.org/10.1016/j.neucom.2020.03.018 Ulyanov, D., Vedaldi, A. and Lempitsky, V. (2020) Deep Image Prior. International Journal of Computer Vision, 128, 1867-1888. >https://doi.org/10.1007/s11263-020-01303-4 Chen, L., Jiang, X., Liu, X. and Haardt, M. (2022) Reweighted Low-Rank Factorization with Deep Prior for Image Restoration. IEEE Transactions on Signal Processing, 70, 3514-3529. >https://doi.org/10.1109/tsp.2022.3183466 Cai, J., Candès, E.J. and Shen, Z. (2010) A Singular Value Thresholding Algorithm for Matrix Completion. SIAM Journal on Optimization, 20, 1956-1982. >https://doi.org/10.1137/080738970 Cascarano, P., Sebastiani, A., Comes, M.C., Franchini, G. and Porta, F. (2021). Combining Weighted Total Variation and Deep Image Prior for Natural and Medical Image Restoration via ADMM. 2021 21st International Conference on Computational Science and Its Applications (ICCSA), Cagliari, 13-16 September 2021, 39-46. >https://doi.org/10.1109/iccsa54496.2021.00016 The USC-SIPI Image Database. >http://sipi.usc.edu/database/
Baidu
map