1.一种基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于由下述步骤组成:(1)构建模型训练数据集
灰度图像数据集为X,根据图像分辨率的大小,将图像数据集X中每张图像分割成像素为p×p的图像块,将图像块进行数据增强后构成图像块数据集PX1,对数据集PX1中的图像块加入混合噪声,得到混合噪声图像块集合NPX2,用混合噪声图像块集合NPX2中的图像块减去图像块集合PX1中的对应的图像块,得到只含有混合噪声的残差图像块,所有的残差图像块构成残差图像块数据集NPY,则(NPX2,NPY)组成用于监督训练的数据集;
(2)噪声学习神经网络模型构建
噪声学习神经网络模型的第一层为第一噪声特征抽取单元,中间层由N∈[5,16]个集成层串联组成,集成层由噪声特征强化模块和第一正则化单元构成,最后一层为第二噪声特征抽取单元,所述的第一噪声特征抽取单元将混合噪声图像块集合NPX2作为输入,所述的噪声特征强化模块用于强化和传递提取的噪声特征,所述的第一正则化单元用于加速和稳定训练过程,所述的第二噪声特征抽取单元用于产生预测的残差图像块;
所述的噪声特征强化模块由第一空洞卷积单元、第二正则化单元、带泄露线性整流单元、第二空洞卷积单元、第三正则化单元、指数映射单元、特征融合单元构成,第一空洞卷积单元的输出分两支,一支输出到第二正则化单元,另一支输出到特征融合单元,第二正则化单元输出到带泄露线性整流单元,带泄露线性整流单元输出到第二空洞卷积单元,第二空洞卷积单元输出到第三正则化单元,第三正则化单元输出到指数映射单元,指数映射单元输出到特征融合单元;
(3)噪声学习神经网络模型训练及初始参数的确定
(3.1)网络模型参数初始化:网络参数的初始值按照如下公式进行:
A=randn(n,n,in,out) (1)[θi,~,~]=SVD(A×α×chi) (2)
公式(1)中n表示卷积核的大小,in表示输入的特征图数量,out表示输出的特征图数量,randn表示生成n×n×in×out四维的具有高斯分布的矩阵,公式(2)中θi是当前卷积层的参数矩阵,符号~表示忽略该处的矩阵值,SVD表示执行奇异值分解操作,A是参数矩阵,α是参数变化率,chi表示当前卷积层的通道数;
(3.2)使用训练数据集(NPX2,NPY)对噪声学习神经网络模型进行训练,利用随机梯度下降算法来最小化损失函数L(θ),通过对损失函数L(θ)不断地迭代优化,直到迭代次数达到迭代次数上限,即完成噪声学习神经网络模型的训练和参数的优化,其中,噪声学习神经网络模型的损失函数L(θ)如下所示:公式(3)中K表示训练数据集(NPX2,NPY)中样本的个数, 表示执行l2范数操作,R(yi;θ)表示在噪声学习神经网络模型的参数为θ的条件下,预测出的第i张残差图像yi,NPYi表示第i张预测的残差图像对应的真实的残差图像,γ∈(0,1)是正则化系数,d(R(yi;θ),NPYi)表示距离函数;
所述距离函数d(R(yi;θ),NPYi)为:
d(R(yi;θ),NPYi)=(R(yi;θ)-NPYi)T×[βI-MTM]×(R(yi;θ)-NPYi) (4)公式(4)中I是单位矩阵,M是二进制对角矩阵,β是距离调整参数,β取值为δ×c,δ是控制因子,c为M的最大特征值,满足βI>||M||2=1;
(4)对混合噪声图像进行去噪
利用训练好的噪声学习神经网络模型对带有混合噪声的图像进行处理,输出残差图像,用带有混合噪声的图像减去残差图像可得到清晰图像。
2.根据权利要求1所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述的步骤(2)中噪声特征强化模块中,噪声特征强化模块的噪声特征强化公式为:outputi=f(inputi) (5)公式(5)中f(inputi)是噪声特征强化模块的特征强化函数,inputi和outputi分别是第i个噪声特征强化模块的输入和输出;
所述的噪声特征强化模块的特征强化函数f(inputi)为:
f(inputi)=exp{-(Dconv(LReLU(BN(Dconv(inputi)))))2}×Dconv(inputi) (6)公式(6)中Dconv是执行空洞卷积操作,BN是执行正则化操作,LReLU是执行带泄露线性整流操作,exp{.}是执行指数操作。
3.根据权利要求2所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述的步骤(2)中,噪声特征强化模块和第一正则化单元构成集成层的方式为:
boutputj=BN(f(inputj)) (7)公式(7)中的boutputj是第j个集成层的输出。
4.根据权利要求1所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述的步骤(3.1)中卷积核大小n∈{3,5,7,9};初始变化率α∈(0.001,0.01)。
5.根据权利要求1所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述步骤(3.2)中,控制因子δ∈(1,2)。
6.根据权利要求1所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述的步骤(1)中p∈{40,50,60}。
7.根据权利要求1所述的基于噪声学习神经网络模型的图像混合噪声消除方法,其特征在于:所述的步骤(1)和步骤(4)中混合噪声由高斯噪声级别σ∈(0,50]的高斯噪声、椒盐噪声比率s∈(0,0.4]的椒盐噪声、随机噪声比率r∈(0,0.3]的随机噪声中的两种或三种组成。