在图像复原中,极大似然估计(Maximum Likelihood Estimation, MLE)提供了一种基于统计的方法来恢复原始图像。这种方法通过构建一个描述退化过程的概率模型,并找到最可能产生观测到的退化图像的参数值来进行图像复原。MLE 不仅考虑了退化过程本身,还考虑了噪声的统计特性。
极大似然估计应用于图像复原
假设我们有一个退化图像 g \bm{g} g,它是从原始图像 f \bm{f} f经过线性退化算子 H \bm{H} H和加性噪声 n \bm{n} n产生的:
g = H f + n \bm{g} = \bm{H}\bm{f} + \bm{n} g=Hf+n
这里:
- g \bm{g} g是退化后的观测图像。
- H \bm{H} H是退化矩阵,例如点扩散函数(PSF)的离散化版本。
- n \bm{n} n是噪声向量,通常假定为均值为0、方差为 σ 2 \sigma^2 σ2的高斯白噪声。
极大似然估计是寻找最有可能生成观测图像 g \bm{g} g的原始图像 f \bm{f} f。
构建似然函数
对于每个像素位置 i i i,观测值 g i g_i gi可以视为随机变量,其概率密度函数(PDF)由下式给出:
p ( g i ∣ f i ; H , σ 2 ) = 1 2 π σ 2 exp ( − ( g i − ( H f ) i ) 2 2 σ 2 ) p(g_i | f_i; \bm{H}, \sigma^2) = \frac{1}{\sqrt{2\pi\sigma^2}} \exp\left(-\frac{(g_i - (\bm{H}\bm{f})_i)^2}{2\sigma^2}\right) p(gi∣fi;H,σ2)=2πσ21exp(−2σ2(gi−(Hf)i)2)
假设噪声是独立同分布(i.i.d.)的高斯噪声。由于像素之间的噪声是独立的,我们可以将所有像素的概率相乘得到整个图像的联合概率:
L ( f ; g , H , σ 2 ) = ∏ i = 1 N p ( g i ∣ f i ; H , σ 2 ) L(\bm{f}; \bm{g}, \bm{H}, \sigma^2) = \prod_{i=1}^{N} p(g_i | f_i; \bm{H}, \sigma^2) L(f;g,H,σ2)=i=1∏Np(gi∣fi;H,σ2)
对数似然函数为:
ln L ( f ; g , H , σ 2 ) = − N 2 ln ( 2 π σ 2 ) − 1 2 σ 2 ∑ i = 1 N ( g i − ( H f ) i ) 2 \ln L(\bm{f}; \bm{g}, \bm{H}, \sigma^2) = -\frac{N}{2}\ln(2\pi\sigma^2) - \frac{1}{2\sigma^2} \sum_{i=1}^{N} (g_i - (\bm{H}\bm{f})_i)^2 lnL(f;g,H,σ2)=−2Nln(2πσ2)−2σ21i=1∑N(gi−(Hf)i)2
最大化对数似然函数
为了简化计算,我们通常最大化对数似然函数。注意到第一项与 f \bm{f} f无关,因此优化问题可以简化为最小化残差平方和(RSS),这实际上等价于最小二乘估计:
f ^ = arg min f ∑ i = 1 N ( g i − ( H f ) i ) 2 \hat{\bm{f}} = \arg\min_{\bm{f}} \sum_{i=1}^{N} (g_i - (\bm{H}\bm{f})_i)^2 f^=argfmini=1∑N(gi−(Hf)i)2
然而,如果噪声不是高斯分布或者退化模型更复杂,MLE 可能会采取不同的形式,并且需要更复杂的优化方法来求解。
正则化的极大似然估计
在实际应用中,直接使用 MLE 可能会导致病态问题或对噪声敏感。为了克服这些问题,可以在似然函数中添加正则化项,形成正则化的极大似然估计(Regularized Maximum Likelihood, RML)。例如,Tikhonov 正则化可以被添加到对数似然函数中,从而得到:
f ^ = arg min f [ ∑ i = 1 N ( g i − ( H f ) i ) 2 + λ ∥ R f ∥ 2 2 ] \hat{\bm{f}} = \arg\min_{\bm{f}} \left[ \sum_{i=1}^{N} (g_i - (\bm{H}\bm{f})_i)^2 + \lambda \|\bm{R}\bm{f}\|_2^2 \right] f^=argfmin[i=1∑N(gi−(Hf)i)2+λ∥Rf∥22]
这里 λ \lambda λ是正则化参数, R \bm{R} R是正则化矩阵,它可以根据先验知识选择,比如平滑度、总变差等。
迭代算法
对于大型图像数据集,解析解可能是不切实际的,因此常常使用迭代优化算法来求解上述优化问题。这些算法包括但不限于梯度下降法、牛顿法、共轭梯度法以及专门针对图像处理设计的算法如期望最大化(EM)算法。