马尔科夫随机场与灰度图像去噪（转化为最大流）

jitwo

已于 2022-12-28 17:32:04 修改

阅读量527

点赞数 1

文章标签：人工智能

于 2022-12-28 17:28:51 首次发布

本文链接：https://blog.csdn.net/jitwo/article/details/128471420

版权

马尔科夫随机场定义

图片的椒盐噪声可以利用马尔科夫随机场进行滤除。

图像的马尔科夫随机场可形式化地定义如下：

1.位置集合 $S=\{1,2,\cdots N\}$ 。位置集合中的每一个元素对应图像中的一个像素；

2.一组随机变量 $\{w_n\}^N_{n=1}$ 。其中每一个随机变量与位置集合中的一个元素相关联；

3.每一个位置元素都有一个邻居集合 ${N_n\}_{n-1}^N$ .

马尔科夫随机场满足马尔可夫性：
$P_r(w_n|w_{S\\n}) = P_r(w_n|w_{N_n} )$

因此，马尔科夫随机场可以将变量的联合概率描述为势函数的乘积，即
$P_r(w) = \frac{1}{Z}\prod_{j=1}^J \phi_j[\mathbf{w}_{C_j} ],$

其中 $\phi_j[•]\ge0$ 表示第j个势函数。势函数的设置与变量 $C_j \subset \{1,\cdots,N\}$ 有关。

将模型表示为Gibbs分布如下：
$\operatorname{Pr}(\mathbf{w})=\frac{1}{Z} \exp \left[-\sum_{j=1}^{J} \psi_{j}\left[\mathbf{w}_{C_{j}}\right]\right]$

其中 $\psi[•] = − log[\phi[•]]$ 表示代价函数，可以返回正值或负值。

图像去噪的模型表示

在滤除椒盐噪声时，

待去噪图像表示为一组变量 $\mathbf{x}=\{x_1,x_2,\cdots,x_n\}$ ，其中每一个代表图像所包含的所有灰度值。我们的目的为求解添加噪声前的原始图像 $\mathbf{w}={w_1,w_2,\cdots,w_n}$ ，其中的每一个变量也代表灰度值。
模型示意图如图1.
马尔科夫随机场示意图

马尔科夫随机场的团由图像的网格结构中的相邻像素对组成，即每个像素与其上、下、左、右的像素点组成的像素对。因此，
$\operatorname{Pr}\left(w_{1 \ldots N}\right)=\frac{1}{Z} \exp \left[-\sum_{(m, n) \in \mathcal{C}} \psi\left[w_{m}, w_{n}, \boldsymbol{\theta}\right]\right]$

$\psi\left[w_{m}=j, w_{n}=k, \boldsymbol{\theta}\right]=\theta_{j k}$

其中参数 $\theta_{jk}$ 定义了代价函数。当j和k的差值较小时， $\theta_{jk}$ 也较小。因此该代价函数鼓励大多数平滑的去噪方案。

图像去噪需要求解的是

$\begin{aligned} \hat{w}_{1 \ldots N} & =\underset{w_{1} \ldots N}{\operatorname{argmax}}\left[\operatorname{Pr}\left(w_{1 \ldots N} \mid \mathbf{x}_{1 \ldots N}\right)\right] \\ & =\underset{w_{1 \ldots N}}{\operatorname{argmax}}\left[\prod_{n=1}^{N} \operatorname{Pr}\left(x_{n} \mid w_{n}\right) \operatorname{Pr}\left(w_{1 \ldots N}\right)\right] \\ & =\underset{w_{1 \ldots N}}{\operatorname{argmax}}\left[\sum_{n=1}^{N} \log \left[\operatorname{Pr}\left(x_{n} \mid w_{n}\right)\right]+\log \left[\operatorname{Pr}\left(w_{1 \ldots N}\right)\right]\right] \end{aligned}$

应用贝叶斯法则并转换到对数域后，上述可以写为

$\begin{aligned} \hat{w}_{1 \ldots N} & =\underset{w_{1} \ldots N}{\operatorname{argmax}}\left[\sum_{n=1}^{N} \log \left[\operatorname{Pr}\left(x_{n} \mid w_{n}\right)\right]-\sum_{(m, n) \in \mathcal{C}} \psi\left[w_{m}, w_{n}, \boldsymbol{\theta}\right]\right] \\ & =\underset{w_{1 \ldots N}}{\operatorname{argmin}}\left[\sum_{n=1}^{N}-\log \left[\operatorname{Pr}\left(x_{n} \mid w_{n}\right)\right]+\sum_{(m, n) \in \mathcal{C}} \psi\left[w_{m}, w_{n}, \boldsymbol{\theta}\right]\right] \\ & =\underset{w_{1 \ldots N}}{\operatorname{argmin}}\left[\sum_{n=1}^{N} U_{n}\left(w_{n}\right)+\sum_{(m, n) \in \mathcal{C}} P_{m n}\left(w_{m}, w_{n}\right)\right], \end{aligned}$

其中 $U_{n}\left(w_{n}\right)$ 表示像素n处的一元项，而 $P_{m n}\left(w_{m}, w_{n}\right)$ 表示图像中相邻位置 $w_m$ 和 $w_n$ 的灰度值（标签）为m和n的代价函数。

最大流定义

关于代价函数的优化，我们可以将马尔科夫随机场的推理过程转换为最大流问题。

我们首先考虑二值化图像的情况。二值化图像的代价函数可以设置为以下函数。

$P_{mn}(w_m,w_n) = \begin{cases} \theta_{00} & w_m=0,w_n=0 \\ \theta_{10} & w_m=1,w_n=0 \\ \theta_{01} & w_m=0, w_n=1 \\ \theta_{11} & w_m=1, w_n=1 \\ \end{cases}$

其中 $\theta_{10},\theta_{01} > 0$ .

之后我们将MAP模型转换为一个图模型。图像中的每一个像素对应图模型中的一个顶点，图像中的每一对相邻像素点之间都构建一对有向边。之后，在图模型中额外构建两个顶点：源点和汇点。源点到除汇点外的所有顶点拥有一条有向边，除源点外的所有顶点到汇点拥有一条有向边。将图模型简化为只有3个像素点的情况，模型表示如图2.

在这里插入图片描述

当最小割切割到相邻像素顶点之间的边时，需要加上相应的代价函数。之后再加上所有像素点的一元项即为所求算式的最优解。求解实例如图3.

对于灰度图像，我们可以定义合适的代价函数，鼓励平滑的情况。但我们定义的代价函数需要满足非凸条件才能转化为最大流图模型进行优化计算。
{灰度图像代价函数定义

如图4，a) 二次函数
$P_{mn}(w_m,w_n)=\kappa(w_m=w_n)^2$ 是凸函数，不能作为图模型的代价函数；b) 截断二次函数
$P_{mn}(w_m,w_n)=\min(\kappa_1(w_m=w_n)^2,\kappa_2)$ 是非凸函数，可以作为图模型的代价函数；c) Potts模型
$P_{mn}(w_m,w_n)=\kappa(1-\delta(w_m-w_n))$ 为非凸函数。

MAP模型求解的不足

在本次实验的实际操作中我并没有使用MAP模型进行灰度图像的去噪任务，原因如下：
经实验，当待去噪图像的噪声比例过大时，MAP模型的先验信息过少，去噪效果较差；此外，
像素n处的一元项 $U_{n}\left(w_{n}\right)$ 难以找到合适的经验值进行实际的去噪任务，很容易出现将灰度图片平滑为全白或全黑的情况。