小波域图像去噪

基本模型

考虑加性噪声模型
y=x+n y=x+n
其中, y为观测图像, x 为真实图像,n为噪声

等号两边进行小波变换,因为小波变换的叠加性可得
Y=X+N Y=X+N
其中 Y 为 y 的小波变换,X 为 x 的小波变换, N 为 n 的小波变换

从统计学的角度来讲,我们要得到真是图像 X 的最有估计 X^\hat{X},即
X^=argmaxXP(XY) \hat{X} = arg \max\limits_X{P(X|Y)}
采用贝叶斯模型,即
P(XY)=P(YX)P(X)P(Y) P(X|Y)=\frac{P(Y|X)P(X)}{P(Y)}

X^=argmaxXP(YX)P(X)P(Y)=argminX{logP(YX)logP(X)} \hat{X} = arg \max\limits_X{\frac{P(Y|X)P(X)}{P(Y)}}=arg \min\limits_X \{ -\log P(Y|X)-\log P(X) \}

P(YX)=P((X+N)X)=P(XX)+P(NX)=1+P(N) P(Y|X)=P((X+N)|X)=P(X|X)+P(N|X)=1+P(N)
因此
(1)X^=argmaxXP(YX)P(X)P(Y)=argminX{logP(N)logP(X)} \hat{X} = arg \max\limits_X{\frac{P(Y|X)P(X)}{P(Y)}}=arg \min\limits_X \{ -\log P(N)-\log P(X) \} \tag{1}
首先考虑(1)式的第一项

假设噪声 n 服从零均值高斯白噪声,即
P(nij)exp{ni,j22σn2} P(n_{ij})\propto exp \{ -\frac{n_{i,j}^2}{2 \sigma_n^2} \}
对于正交小波变换,白噪声在小波变换前后性质不变,方差不变,即
(2)P(Ni,j)exp{ni,j22σN2}=exp{(Yi,jXi,j)22σN2} P(N_{i,j})\propto exp \{ -\frac{n_{i,j}^2}{2 \sigma_N^2} \}= exp \{ -\frac{(Y_{i,j}-X_{i,j})^2}{2 \sigma_N^2} \} \tag{2}
在考虑(1)式的第二项,即图像在小波域的统计概率模型,包括广义高斯分布、广义拉普拉斯分布、隐马尔可夫模型等。根据图像的特点,不同的先验统计模型,会得到不同的去噪算法。

这里我们选用最为简单的高斯分布,即
(3)P(Xi,j)exp{Xi,j22σX2} P(X_{i,j}) \propto exp \{ -\frac{X_{i,j}^2}{2 \sigma_X^2} \} \tag{3}
将(2)、(3)带入到(1)可得:
X^i,j=argminXi,j{(Yi,jXi,j)2σN2+Xi,j2σX2} \hat{X}_{i,j} = arg \min\limits_{X_{i,j}} \{ \frac{(Y_{i,j}-X_{i,j})^2}{ \sigma_N^2} +\frac{X_{i,j}^2}{ \sigma_X^2} \}

J=(Yi,jXi,j)2σN2+Xi,j2σX2 J= \frac{(Y_{i,j}-X_{i,j})^2}{ \sigma_N^2} +\frac{X_{i,j}^2}{ \sigma_X^2}

0=JXi,j=2Xi,jYi,jσN2+2Xi,jσN2 0=\frac{\partial J}{\partial X_{i,j}}=\frac{2(X_{i,j}-Y_{i,j})}{\sigma_N^2}+\frac{2X_{i,j}}{\sigma_N^2}
可得
(4)X^i,j=σX2σN2+σX2Yi,j=WYi,j \hat{X}_{i,j}= \frac{\sigma_X^2}{\sigma_N^2+\sigma_X^2}Y_{i,j}=WY_{i,j} \tag{4}
显然,当选取小波域统计模型为高斯分布时,最优估计的小波域系数为观测图像小波域系数的线性倍数,即维纳滤波。

图像的小波域统计模型为广义高斯分布时的情形

现在考虑小波统计模型为广义高斯分布时的情形


(5)P(X)exp{(XσX)P} P(X) \propto exp \{ -(\frac{|X|}{\sigma_X})^P\} \tag{5}

P=1\2时

J1/2=(Yi,jXi,j)2σN2+Xi,j1/2σX0=J1Xi,j=0{Xi,j+14σXσN2x=Yi,jXi,j0Xi,j+14σXσN2x=Yi,jXi,j<0 J_{1/2}= \frac{(Y_{i,j}-X_{i,j})^2}{ \sigma_N^2} +\frac{|X_{i,j}|^{1/2}}{ \sigma_X}\\ 0=\frac{\partial J_1}{\partial X_{i,j}}=0 \Rightarrow \begin{cases} X_{i,j}+\frac{1}{4\sigma_X\sigma_N^2\sqrt{x}}=Y_{i,j} \qquad X_{i,j}\geq 0\\ X_{i,j}+\frac{1}{4\sigma_X\sigma_N^2\sqrt{-x}}=Y_{i,j} \qquad X_{i,j}<0 \end{cases}

p=1时

J1=(Yi,jXi,j)2σN2+Xi,jσX0=J1Xi,j={2Xi,jYi,jσN2+1σXXi,j02Xi,jYi,jσN21σXXi,j<0 J_1= \frac{(Y_{i,j}-X_{i,j})^2}{ \sigma_N^2} +\frac{|X_{i,j}|}{ \sigma_X}\\ 0=\frac{\partial J_1}{\partial X_{i,j}}=\begin{cases} \frac{2(X_{i,j}-Y_{i,j})}{\sigma_N^2}+\frac{1}{\sigma_X} \qquad X_{i,j} \geq 0 \\ \frac{2(X_{i,j}-Y_{i,j})}{\sigma_N^2}-\frac{1}{\sigma_X} \qquad X_{i,j}<0 \end{cases}

可得
(6)X^i,j={Yi,jσN22σXXi,j0Yi,j+σN22σXXi,j<0 \hat{X}_{i,j}=\begin{cases} Y_{i,j}-\frac{\sigma_N^2}{2\sigma_X} \qquad X_{i,j} \geq 0\\ Y_{i,j}+\frac{\sigma_N^2}{2\sigma_X} \qquad X_{i,j} <0 \end{cases} \tag{6}

P=2时

​ 很方便,可以验证最优估计的形式为线性形式,如式(4)。

为了直观地看出取不同P值时,小波域最优估计系数与观测系数的关系,我们做如下图

小波域图像降噪

其中P=1/2时,根据对勾函数的性质可以得到最有估计与观测值的关系如图(a);P=1时,显然根据式(6)可得图(b);P=2时,根据式(4)可得图(c)。

事实上,图(a)(b) (c)分别对应了小波域图像去噪收缩模型的硬阈值,软阈值及全局维纳滤波。

在下篇文章中,将针对全局维纳滤波的缺点,介绍局部维纳滤波的方法。

相关文章: