基于交互势函数和均场参数估计的分层MRF模型的纹理图像分割
李一兵1, 杨鹏1, 叶方1, 刘丹丹2
1.哈尔滨工程大学 信息与通信工程学院,哈尔滨 150001
2.黑龙江科技大学 电气与控制工程学院,哈尔滨 150022
叶方(1980-),女,副教授,博士.研究方向:超宽带无线通信,认知无线电,图像处理与模式识别.E-mail:yefang0815@sina.cn

作者简介:李一兵(1967-),男,教授,博士生导师.研究方向:图像处理,认知无线电,信息融合技术.E-mail:liyibing@hrbeu.edu.cn

摘要

针对分层马尔科夫模型在用期望最大(EM)算法进行参数估计时,隐变量之间相互作用导致求期望值较难的问题,将均场理论应用到GMRF模型的系数估计中,使得模型的参数可以在不使用窗函数的情况下仅通过简单的线性方程即可求出。而对于固定势函数和变权重势函数不能表达图像区域间节点交互关系的缺点,提出了一种基于贝叶斯传播算法的交互势函数。试验结果表明:本文算法分割后的图像不仅具有良好的区域性,而且区域内部平滑,改善了传统小波域分层马尔科夫模型在分割区域内部存在混分的现象。

关键词: 通信技术; 分层MRF模型; 均场理论; 线性方程; 交互势函数; 图像分割
中图分类号:TN911.73 文献标志码:A 文章编号:1671-5497(2015)06-2075-05
Texture image segmentation using hierarchical MRF model based on the interactive potential function and mean-field parameter estimation
LI Yi-bing1, YANG Peng1, YE Fang1, LIU Dan-dan2
1.College of Information and Communication Engineering, Harbin Engineering University, Harbin 150001,China
2.College of Electrical and Control Engineering, Heilongjiang University of Science and Technology, Harbin 150022, China
Abstract

To solve the difficult problem of expectations caused by interaction between hidden variables when using Expectation-maximization (EM) algorithm to estimate parameters for hierarchical Markov Random Fields (MRF) model, the mean-field theory is introduced into Gaussian-MRF (GMRF) model. Parameters can be estimated easily through simple linear equation in case of without window function. An interactive potential function based on Bayesian belief propagation algorithm is proposed to change the situation that the fixed or variable weighted potential function can not express the interaction of image regions. Experiments demonstrate that the proposed method not only has good regional classification but also smoothly internal region. In addition, the mixed and confused phenomenon of traditional hierarchical MRF is improved in wavelet domain.

Keyword: communication; hierarchical MRF; mean-field theory; linear equation; interactive potential function; image segmentation
0 引 言

MRF模型是已被广泛应用的随机模型[1, 2, 3, 4]。为解决在边界处的不同类别的纹理将导致图像特征不能精确地被提取出来这个问题, 在已经提出了许多的随机模型中, 最值得注意的研究是基于分层MRF模型[5]。直到现在, 在分层MRF模型中, 最流行的算法仍是期望最大(EM)算法。Zhang等[5]提出的分层MRF包括两个马尔可夫随机场模型。一个是隐藏区域分布建模; 另一个是纹理建模。在这项技术中, 隐藏区域使用EM估计产生的纹理分割。然而, 在这种情况下, 当某一区域比其他区域相对较小或者初始估计不正确, 那么就会产生错误分割。通常用GMRF对特征场进行建模时, 一般采用EM算法[4]。然而, 由于不同部位的隐变量之间的相互作用, EM算法的期望是很难计算的。在本文中, 采用平均场理论[1]应用到分层GMRF的建模中, 并且采用唯一的MRF线性解决方案进行参数估计。

在以往的马尔科夫模型分割中, 为了方便起见, 选取固定的值作为势函数, 可是考虑到势函数对整体分割结果的影响:当势函数很小时, 会得到较为精细的分割, 在结果中更多的细节会被保留下来, 那么图像的真实边界也更为清晰, 可是较小的势函数的区域性却不是很好, 当逐渐增加势函数时, 区域性得到了很好的表现, 但当势函数过大时, 图像边界会出现严重的变形现象[6]。所以在以往选取势函数时, 都是根据经验值进行试验, 这样不仅费时费力, 而且这样选择的固定值势函数很难做到既保留边界细节又有良好的区域性。变权重方法有一定改进效果, 它将特征场加一个权重[7], 使得特征场模型更为接近小波系数向量场。可是在这个权重函数中, 只考虑了尺度、分类数、迭代次数, 并没有考虑到图像各层之间的空间交互关系。

为了克服这一缺点, 本文将基于贝叶斯置信传播算法的交互势函数结合到势函数的选取中, 这样就能充分考虑节点之间的交互信息, 使分割结果更加准确。

1 分层MRF模型

分层MRF模型经常用在纹理图像的分析中。Zhang等[1, 5]提出的分层MRF模型可简单表示为如下方程:

p(y(i, j), z(i, j)θ)=p(y(i, j)z(i, j), θ)p(z(i, j)θ)(1)

式中:y(i, j)为像素(i, j)的灰度值; θ 为MRF模型的参数; z(i, j)为像素点 i, j的标号。

这种分层MRF可以通过不完全数据的概念被提取出来, 而这个不完全概念将标号区域认为是一种非可观测区域, 这时一般要用到EM算法进行分析。这个算法可以不用窗函数, 并且特征的提取和分割是同时进行的。然而, 该算法不能在每个像素点都产生纹理特征, 并且每个区域只能提取一个纹理特征。因此, 当一个区域相对其他区域较小或者初始化分割不准确的时候, 这种算法将会产生错误的纹理分割。

在文献[4]中的分层MRF模型中, 提出了代表图像强度的RSI(Random spatial interaction)模型也是MRF模型, RSI的概率密度方程p Θ(i, j)可以根据马尔科夫性质而用下式表示:

pΘ(i, j)ΘS(i, j)=pΘ(i, j)ΘN(i, j)2

式中:S为整个图像像素点的集合; N(i, j)为像素点(i, j)的邻域集合。

设符号< · > 代表期望。为了获得参数集Θ (i, j)的期望, 在计算EM算法步骤中的Θ (i, j)时, 要考虑到 ΘN(i, j)Θ (i, j)的影响。然而, 在一般情况下, 由于< Θ (i, j)> 是复杂指数的形式, 要精确计算< Θ (i, j)> 是不能直接实现的[2]。本文为了解决这一问题, 将平均场理论[1]应用到分层MRF模型中。因此, ΘN(i, j)的作用被认为是< Θ (i, j)> 的作用。提出的RSI的概率密度方程可用下式表示:

pΘ(i, j)ΘN(i, j)=pΘ(i, j)< ΘN(i, j)> 3

2 基于线性方程的模型参数估计

GMRF是分析纹理图像的一种常用模型, GMRF模型可以为RSI模型的均值提供简单的积分方程, 因此, GMRF模型被应用到分层马尔科夫模型中, 它的参数可以通过简单的线性方程来得到很好的估计。

首先, GMRF参数向量Θ (i, j)被定义为:

Θ(i, j)=θ(i, j)1, 0θ(i, j)0, 1θ(i, j)1, 1θ(i, j)(1, -1)T4

式中: θ(i, j)(m, n)代表GMRF模型的交互系数, 它表示定义在像素点(i, j)和(i-m, j-n)的随机场的关联。由于邻域结构是对称的, 那么像素点(i-m, j-n)和(i+m, j+n)是一样的, 即 θ(i, j)(m, n)= θ(i, j)-m, -n)。因此, 式(4)中的向量的数目可变为4个, 那么RSI模型的交互参数Θ (i, j)可表示为:

Λ(i, j)=Λ(i, j)1, 0Λ(i, j)0, 1Λ(i, j)1, 1Λ(i, j)(1, -1)T5

式中:

Λ(i, j)(m, n)=λ(i, j)(m, n)(1, 0)λ(i, j)(m, n)(1, -1)T6

(m, n)∈ {(1, 0), (0, 1), (1, 1), (1, -1)}。

分层GMRF模型的概率分布为:

py(i, j), Θ(i, j)YN(i, j), < ΘN(i, j)> =12πσY2ΦΘ-1(2π)Nexp-y(i, j)-μ(i, j)2σY2×exp-Θ(i, j)-Σ(i, j)TΦΘ-1Θ(i, j)-Σ(i, j)27

式中:μ (i, j)y(i, j)的均值, μ (i, j)= Y-T(i, j)Θ (i, j)

那么, RSI的期望就可以通过式(7)计算出来的p Θ(i, j)y(i, j), YN(i, j), ΘN(i, j)进行计算, 那么代表Θ (i, j)的式(7)可由式(8)进行计算:

py(i, j), Θ(i, j)YN(i, j), < ΘN(i, j)> =ΦΘ-12πN+1σY2Ry(i, j)YN(i, j), < ΘN(i, j)> ×exp-Θ(i, j)-Ω(i, j)TΦ-1Θ(i, j)-Ω(i, j)28

式中:

Φ=σY2Y-(i, j)Y-T(i, j)+σY2ΦΘ-1-19

Ry(i, j)YN(i, j), < ΘN(i, j)> =exp-y(i, j)22σY2·

exp-12Σ(i, j)ΦΘ-1Φ-ΩT(i, j)Φ-1Ω(i, j)10

并且:

Ω(i, j)=1σY2Φy(i, j)Y-(i, j)+σY2ΦΘ-1Σ(i, j)11

那么p y(i, j)YN(i, j), ΘN(i, j)可以通过式(8)被定义成以下形式:

py(i, j)YN(i, j), < ΘN(i, j)> =ΦΘ-12πσY2Φ-1Ry(i, j)YN(i, j), < ΘN(i, j)> 12

根据式(8)和(12), p(Θ (i, j)|y(i, j), YN(i, j), < ΘN(i, j)> )可以通过式(13)表示:

pΘ(i, j)y(i, j), YN(i, j), < ΘN(i, j)> =Φ-1(2π)N·exp-Θ(i, j)-Ω(i, j)TΦ-1Θ(i, j)-Ω(i, j)213

这个方程是带有协方差矩阵Φ 和期望向量Ω (i, j)的联合高斯概率密度方程。因此, RSI的最终的期望方程可以通过式(9)和(11)改写为:

< ΘN(i, j)> =Ω(i, j)=Y-(i, j) Y-T(i, j)+σY2ΦΘ-1-1y(i, j)Y-(i, j)+σY2ΦΘ-1Σ(i, j)14

式中:σ Yy(i, j)的标准差; Φ Θ Θ 的协方差矩阵, 并且:

Y-(i, j)=y-(i, j)1, 0 y-(i, j)0, 1 y-(i, j)1, 1 y-(i, j)(1, -1)Ty-(i, j)m, n=y(i+m, j+n)+y(i-m, j-n)

式中:Σ (i, j)Θ (i, j)依赖 ΘN(i, j)的均值:

Σ(i, j)=ξ(i, j)1, 0ξ(i, j)0, 1ξ(i, j)1, 1ξ(i, j)(1, -1)Tξ(i, j)(m, n)=(k, t)λ(i, j)(m, n)(k, l)< θ(i+k, j+l)(m, n)> +< θ(i-k, j-l)(m, n)>

在式(14)中, RSI的期望可以通过简单的线性方程获得, 并且在每个像素点 i, j的线性方程的唯一解不必通过窗函数就可获得。然而, 为了计算每个点的RSI的期望, 首先RSI在邻域像素点的期望要被计算出来。因此, 式(14)被反复使用, 其中RSI的在邻域的期望已在前面给出。

3 交互势函数的建立

在以往的变权重思想中, 只考虑了尺度、分类数、迭代次数。并没有考虑到图像各层之间的空间交互关系。为了克服这一缺点, 本文将基于贝叶斯置信传播算法的交互势函数结合到势函数的选取中, 这样就能充分考虑节点之间的交互信息[8]

贝叶斯传播算法定义如下, 假设一个随机变量集合X={x1, x2, …, xQ}和对应的观测值集合Y={y1, y2, …, yQ}。集合X中任意两个相邻节点xi, xj统计相关并且条件独立于X中的其他节点[9]φ i(xi, yi)表示随机变量xi和观测值yi的关联势函数; ψ i, j(xi, xj)表示两个相邻节点的交互势函数, 分别表示为:

φi(xi, yi)=explngiyixi15ψi, j(xi, yi)=exp-Vc(xi, yi)16

对于区域的关联势函数φ i(xi, yi), 其观测值yi为区域的特征Fi, 因为本文分割算法通过对区域特征进行模糊C均值聚类来完成初始分割[10], 则本文将φ i(xi, yi)定义为:

φi(xi, yi)=μ(Fi, xi)(17)

式中:μ (Fi, xi)表示第i个区域对xi的隶属度。

对于两个相邻区域间的交互势函数ψ i, j(xi, xj), 定义如下:

ψ(xi, xj)=exp-(Fij/k)2, Fij< F0k, γ, FijF0k18

式中: Fij=Fi-Fj‖ , i=1, …, Q, jN(Ri)表示相邻区域的特征距离, 并将其进行归一化:

Fij=Fij-FminFmax-Fmin19

式中: Fmin= mini, jN(Ri)Fij; Fmax= maxi, jN(Ri)Fij; γ 为正常数, F0k= -k2×log(0.5), k=α +(t-1)× λ , k≤ 1.2。

这里α = 1Ei=1QjN(R)Fij表示图像中所有相邻区域特征差值的平均; E为分类数; λ t分别为步长和迭代次数。参数γ 的选择区间为0.4≤ γ ≤ 0.5; k的范围为α k≤ 1.2。

本文利用贝叶斯置信传播算法中表示节点i到节点j的信息传递, 也就是节点i当前的状态对节点j的影响函数mij(xj)作为分层MRF模型的势函数, 即β =mij(xj), 称之为交互势函数, 其中mij(xj)的表达式为:

mij(xj)=xiΩφi(xi, yi)ψjxi, xjkN(i)/jmki(xi)20

4 仿真试验及结果分析

为了验证本文算法的有效性, 在Matlab R2010a环境下, 将分层马尔科夫模型、变权重的分层MRF模型和本文提出的方法进行试验对比, 试验中采用的是对Brodatz纹理图像库的纹理图像进行合成之后的图像, 测试图像的尺寸均为256像素× 256像素, 灰度级为256。下面是试验对比图, 图1(a)、图2(a)、图3(a)是原始合成纹理图; 图1(b)、图2(b)、图3(b)是手工标注的纹理图像真实分割; 图1(c)、图2(c)、图3(c)是通过分层马尔科夫模型后的分割结果; 图1(d)、图2(d)、图3(d)是通过变权重分层马尔科夫模型后的分割结果; 图1(e)、图2(e)、图3(e)是通过本文算法后的分割结果。

图1 规则两区域合成纹理图分割对比Fig.1 Regular two synthetic texture image segmentations

图2 规则四区域合成纹理图分割对比Fig.2 Regular four synthetic texture image segmentations

图3 不规则四区域合成纹理图分割对比Fig.3 Reguless four syntheic texture image segmentations

本试验对区域的合成纹理图像进行仿真试验, 在选取合成纹理图的构成类型时, 不仅选取了规则图形的拼接, 也选择了不规则的图形。从试验对比图中可以看到, 以往的分层马尔科夫模型虽有良好的区域性, 但是它的分割图像各区域内部不够平滑, 有较多误分割的大斑点, 在变权重分层马尔科夫模型中, 大斑点的现象明显较少, 但是还是存在比较明显的斑点和细纹, 区域内部处理的也不够理想。而使用本文算法分割出来的图像, 有着良好的区域性和清晰的边界, 并且在分割结果图中每类纹理内部都没有多余的斑点和条纹, 分割效果较为理想, 验证了本文方法具有可行性和有效性。

表1从定量的角度也分析了上面的3组试验, 这里采用的定量指标为整体分类精度, 这个指标越大, 说明分割的精度越高, 分割效果越好。从表中数据可以看出, 通过与手工标注的真实分割图像对比, 验证了本文提出的算法较前两种方法更有效, 整体分类精度有10个百分点左右的提高。

表1 三种方法整体分类精度的比较 Table 1 Comparsion of overall classification accuracy(OCA)
5 结束语

针对分层马尔科夫图像分割模型在参数估计时使用EM算法的弊端, 引入均场理论, 并利用简单的线性方程来求得GMRF模型的参数估计值, 不仅简化了求均值的过程, 而且还避免了EM算法在某一区域相对小于其他区域或在初始估计时会产生错误分割的现象。然后通过贝叶斯传播算法中关联势函数和交互势函数的概念, 利用影响节点之间交互作用的函数作为本文算法的势函数, 很好地解决了以往固定势函数和变权重势函数没有利用图像节点之间的交互信息的问题。最后通过试验定量分析了本文算法的有效性。

The authors have declared that no competing interests exist.

参考文献
[1] Zhang Jun. The mean field theory in EM procedures for Markov rand om fields[J]. IEEE Trans Signal Processing, 1992, 40(10): 2570-2583. [本文引用:4]
[2] 李骜, 李一兵, 刘丹丹. 基于照度划分的多尺度图像增强新算法[J]. 吉林大学学报: 工学版, 2012, 42(2): 494-498.
Li Ao, Li Yi-bing, Liu Dan-dan. Multi-scale image enhancement algorithm based on illuminance partition[J]. Journal of Jilin University(Engineering and Technology Edition), 2012, 42(2): 494-498. [本文引用:2]
[3] Deng Hua-wu, Clausi D A. Unsupervised image segmentation using a simple MRF model with a new implementation scheme[J]. Pattern Recognition, 2004, 37(12): 2323-2335. [本文引用:1]
[4] Kim J H, Yun I D, Lee S U. Unsupervised segmentation of textured image using Markov rand om field in rand om spatial interaction[C]∥1998 International Conference on Image Processing, Chicago, IL, 1998: 756-760. [本文引用:2]
[5] Zhang Jun, Modestino J W, Langan D A. Maximum likelihood parameter estimation for unsupervised stochastic model-based image segmentation[J]. IEEE Trans Image Processing, 1994, 3(4): 404-420. [本文引用:3]
[6] 刘国英, . 基于Markov随机场的小波域图像建模及分割[M]. 北京: 科学出版社, 2010: 91-105. [本文引用:1]
[7] 刘雪娜, 侯宝明. 变权重MRF算法在图像自动无监督分割中的应用[J]. 计算机与现代化, 2012, 1(11): 78-80.
Liu Xue-na, Hou Bao-ming. Weighted MRF algorithm for automatic unsupervised image segmentation[J]. Computer and Modernization, 2012, 1(11): 78-80. [本文引用:1]
[8] 毕晓君, 彭伟. 基于改进贝叶斯优化算法的图像分割方法[J]. 应用科技, 2010(12): 19-22.
Bi Xiao-jun, Peng Wei. Image segmentation based on improved Bayesian optimization algorithm[J]. Applied Science And Technology, 2010(12): 19-22. [本文引用:1]
[9] 宋晓峰, 王爽, 刘芳. 基于区域MRF 和贝叶斯置信传播的SAR图像分割[J]. 电子学报, 2010, 38(12): 2810-2815.
Song Xiao-feng, Wang Shuang, Liu Fang. SAR image segmentation using Markov rand om field based on regions and bayes belief propagation[J]. Acta Electronica Sinica, 2010, 38(12): 2810-2815. [本文引用:1]
[10] 李骜, 李一兵, 刘丹丹, . 基于NSCT的多曝光工件图像Retinex增强方法[J]. 吉林大学学报: 工学版, 2012, 42(6): 1592-1596.
Li Ao, Li Yi-bing, Liu Dan-dan, et al. Retinex enhancement method of multi-exposure work piece images based on NSCT[J]. Journal of Jilin University (Engineering and Technology Edition), 2012, 42(6): 1592-1596. [本文引用:1]