第28卷第12期 计算机应用研究 VoL 28 No.12 2011年l2月 Application Research of Computers Dee.2011 自适应全变分图像去噪模型及其快速求解 刘文 ,吴传生 ,许田 (1.武汉理工大学理学院数学系,武汉430070;2.武汉大学电子信息学院通信工程系,武汉430072) 摘要:在联合冲击滤波器和非线性各向异性扩散滤波器对含噪图像做预处理的基础上,利用边缘检测算子选 取自适应参数,构建能同时兼顾图像平滑去噪与边缘保留的自适应全变分模型,并基于Bregman迭代正则化方 法设计了其快速迭代求解算法。实验结果表明,自适应去噪模型及其求解算法在快速去除噪声的同时保留了图 像的边缘轮廓和纹理等细节信息,得到的复原图像在客观评价标准和主观视觉效果方面均有所提高。 关键词:图像去噪;全变分模型;Bregman迭代正则化;分裂Bregman迭代算法 中图分类号:TP391 文献标志码:A 文章编号:1001-3695(2011)12—4797—04 doi:10.3969/j.issn.1001—3695.2011.12.107 Adaptive total variation model for image denoising with fast solving algorithm LIU Wen ,WU Chuan—sheng ,XU Tian (1.Dept.ofMathematics,School ofSciences,Wuhan University ofTechnOlogy,Wuhan 430070,China;2.Dept.ofCommunication Engineer- ing,School of Electronic Information,Wuhan Unievrsity,Wuhan 430072,China) Abstract:This paper combined shock filter with anisotropic diffusion to preprocess the noisy images,and used the edge de— tection filters to choose the parameters adaptively based on the preprocessed images.Then introduced an adaptive total varia— tion regularization model for image denoising based on the chosen parameters.The proposed model could keep the balance be— tween noises smoothing and edges preserving adaptively.Furthermore,it proposed a fast iterative algorithm to solve the pro— posed adaptive model based on Bregman iteration regularization method.The numerical results show that the proposed model and fast algorithm can smooth the noises and preserve the edge and fine detail information properly with fast solving conver— gence rate,while the peak signal to noise ratio,mean structural similarity and subjective visual effect of the denoised images are improved obviously. Key words:image denoising;total variation model;Bregman iteration regularization;split Bregman iteration algorithm 图像去噪是数字图像处理中一个重要的分支,它的主要目 点,利用变分原理,求解TV模型对应Euler.Lagrange偏微分 的是改善图像的质量,便于图像处理后续工作的进行。1992 方程是最常用的方法 。由于该PDEs存在非线性及图像数 年Rudin等人提出的全变分(total vairation,TV)模型是目前应 据量庞大的特点,受CFL条件的影响,常用的人工时间演化 用最为成功的图像去噪模型之一,但该模型易将噪声当成图像 方法…收敛速度很慢。为降低CFL条件的限制,Marouina等 边缘,且对应Euler—Lagrange偏微分方程的数值解倾向于逐段 人 通过在TV模型相应梯度下降流方程右边添加扩散系数 常数值,其稳态解中往往有明显的分片常数效应 ,2 J。Song 的方式构建了改进的TV去噪模型,该模型可较快地收敛至 为克服TV模型易产生分片常数效应的问题,提出了广义Tv 稳定解,但去噪图像中也易产生分片常数效应。为完全消除 去噪模型,但针对不同的含噪图像,系数P的客观选取严重影 CFL条件的限制,Vogel等人 提出固定点迭代算法,这种方 响着图像的去噪效果。鉴于此,张红英等人 基于图像的梯 法仅仅线性收敛,依然存在收敛速度慢的问题。Goldstein等 度信息引入全变分自适应图像去噪模型,在图像的边缘与平滑 人” 结合分裂算子技术和Bregman迭代正则化方法,提出分 区自动选取保边较好的Tv模型与平滑较好的调和模型。但 裂Bregman迭代算法,以降低TV模型的计算复杂性。Jia等 该自适应TV模型不具备反扩散能力,对噪声非常敏感,使得 人给出了该算法详细的收敛性证明… ,并在此基础上提出了 复原后的图像边缘模糊,丢失了一些重要的纹理信息。本文将 各向异性TV模型的快速迭代求解算法 。本文在研究 冲击滤波器(shock iflter,SF)[sj和非线性各向异性扩散(aniso— Bregman迭代正则化方法的基础上,将各向异性TV模型的快 tropic diffusion,AD)滤波器 结合起来,对含噪图像进行平滑 速迭代求解算法推广至本文提出的自适应全变分模型,避免 去噪和边缘增强;然后利用边缘检测算子区分预处理图像的边 自适应全变分模型中求解偏微分方程的不足,提高含噪图像 缘与平滑区,得到控制全变分模型扩散水平的自适应参数,降 的复原速度。 低其对噪声的敏感性,使引入的去噪模型更符合图像处理的形 态学原则,同时兼顾图像的平滑去噪与边缘细节保留,并克服 1 自适应全变分图像去噪模型 分片常数效应的产生。 令 为原始清晰图像,/为被噪声污染的退化图像,则 同时,有关TV模型的求解也一直是相关学者的研究重 ,=Ⅱ+ 收稿日期:2011-05—24;修回日期:2011—07一叭 基金项目:国家自然科学基金资助项目(61070009);武汉理工大学自主创新研究基金资 助项目(2010一ZY—LX一022) 作者简介:刘文(1987一),男,湖北孝感人,硕士研究生,主要研究方向为反问题和数字图像处理(1wsunlight@gmail.corn);吴传生(1957-),男, 教授,博士,主要研究方向为偏微分方程反问题、小波分析和智能计算;许田(1987-),女,硕士研究生,主要研究方向为数字图像处理. ・4798- 计算机应用研究 “( )一ATV 第28卷 (“)( )=一 curv(ATV , ( ))( )(7) 其中:n为具有零均值、标准差为 的高斯白噪声。1992年Ru— din等人提出的全变分图像去噪模型即为如下的能量泛函: TV lⅡ(¨)=arg rainTV(“)+ }』nIf( )一Ⅱ( )f dx (2) , 根据式(6)对控制全变分模型扩散的自适应参数 的定 义,在图像的光滑区域 (doof)。--*0,自适应参数 ,一1,根 据对方法噪声式(3)和(7)的分析,此时全变分模型式(2)和自 适应全变分模型式(4)表现出类似的平滑去噪性能。但在图 像的边缘区域∑( 其中: >0为正则化参数;TV(“)为复原图像 的全变分;力 为图像空间且 ∈力。对于式(2)中的全变分项TV( ),采用 各向异性全变分,即TV(u)=I V uI+I V uI。此时全变分图 像去噪式(2)对应的方法噪声(method noise) 为 一。。,自适应参数 ,— ,对于一致较 )一TV ( )=一÷cu“(TV ( ( ) (3) 其中:curv(TV F(u))为 Ⅳ (H)所有水平集的曲率(curva- 小的正则化参数 >0,自适应全变分模型的方法噪声u( )一 ATV , (u)( )珈,而全变分模型的方法噪声u( )一TV (n)( )一一 ,此时全变分模型式(2)在消除噪声时会模糊图 ture)。为有效地消除图像l厂中的噪声,需取较小的正则化参数 肛。由于图像的细节纹理区域与平滑区区域的局部结构特征 并不相同,此时较小的正则化参数 将会导致去噪图像中的细 像的边缘,而自适应全变分模型式(4)却能有效地保留图像的 边缘区域,并随着自适应参数 的变化,恢复含噪图像中的纹 理等结构信息。 节与纹理信息得不到保留,并模糊图像边缘区域。为保证全变 分模型在消除噪声的同时能够保留图像的细节和纹理等结构 信息,引入控制全变分图像去噪模型扩散的自适应参数 ,得 到改进的全变分模型,即自适应全变分模型(adaptive total variation,ATV) ATV ,2 图像去噪模型的快速求解 利用变分原理推导全变分模型对应的Euler—Lagrange偏微 分方程(partial differentila equations,PDEs),并引入一个时间辅 助变量,将该静态非线性PDEs的求解问题转换成一个动态 PDEs的演化迭代问题,此时PDEs演化终止时的稳态解即为该 全变分模型的解。这种常用来求解全变分模型的方法存在速度 慢的问题,且受CFL条件的影响。于是本文基于Bregman迭代 (H):arg mi“ ,TV(u)+ }fnI,( )一u( )I dx(4) 为降低自适应参数 对噪声的敏感性,文章综合利用冲 击滤波器和非线性各向异性扩散滤波器对含噪图像进行预处 理,在平滑噪声的同时得到预处理后的边缘增强图像,然后利 正则化方法设计一种自适应全变分图像去噪模型的快速迭代求 解算法,以避免自适应全变分模型中求解PDEs的不足。 用图像边缘检测算子作用预处理图像计算自适应参数 。其 中,Alvarez等人 将冲击项与扩散项结合起来,提出了下列 形式的方程(anisotropic diffusion with shock iflter,ADSF): 令C= ∈R , 五=c M, 五:CV ,五= ,_广: ,五= ,自适应全变分模型式(4)可等效为 ATV , 号=一sign(G O N)sign(G o )l vft+ T (5) ( )=argminTV( )+ fnI,( )一 ( )i 2dx (8) 对含噪图像_厂进行预处理,在消除图像噪声和锐化图像边 缘的同时,能够保持并增强图像的重要特征,提高自适应参数 的计算精度。其中:“o”表示卷积运算符;G 是一个标准差 为 的高斯函数;c是一个正的常数; 和 分别为对应演化 曲线的梯度方向(Ⅳ)和切线方向( )的二阶方向导数。 假设预处理后的图像为.厂I考虑到人类视觉系统对图像轮 廓与光滑区域的敏感性,重点在于计算这两个区域所对应的自 引入两个辅助变量 、 ,则式(8)可等效地转换为如下的 有约束最优化问题: n nIv I+l I+ l} 一,Il; “ S.t. = , = (9) 为方便约束问题式(9)的求解,将其转换成无约束最优化 问题的形式,即 适应参数 ,以构建图像去噪的自适应全变分模型。如图1所 示,本文引入四个边缘检测算子d (大小为6×6)” 。当0: 厂0l 1 +等II 一,II;+争ll 一 一 1};+ ÷ 一V —b, (10) 其中:A>0;6 与b 为引入的辅助变量。本文将利用Goldstein 等人 提出的分裂Bregman迭代算法来求解各向异性全变分 自适应图像去噪模型。首先将问题式(10)分解成如下的几个 o =击・l。 M 0 I,M:[一1。一22—22 L 0l _J 0 =t o。00。0。0。:0 ]J 【,02=[:0 】J,且对应的边缘检测算子 子问题: d。由 经旋转角度0∈0(0={0。,45。,90。,135。})而得到。 -一m in等 一 ;+÷ k— —b + 叠叠 口 3一i ̄: 01,. 5 l 2 3 4 5 6 1 2 3 4 5 6 l 2 3 4 5 6 1 2 3 4 5 6 ÷ k— —b ; k (11 1) (11.2) :=rainIv I+睾 一V “ 一b ; = (a)0=0。 (b)0=-45。 (c)0=90。 (d)0=135。 图1边缘检测算子 I 争 一 , 一6 )一 k (11.3) (11.4) 根据对边缘检测算子d 的定义,全变分模型式(4)中的 自适应参数为 1 6 :=(b + 6: :=(6 + =——_==1==;=E(0,1] (6) )一 k n O (11.5) l+√ (do⑧/) 其中:O={0。,45。,90。,135。t,此时自适应全变分图像去噪模 型式(4)对应的方法噪声为 Jia在文献[12]中giiEN:当0<一A<÷时,对于给定的 B。=五+A△,子问题(11.1)等效为 ¨’:= n, 1 l JB( 一,) 一(B ( 一,), — )+ 第l2期 刘 文,等:自适应全变分图像去噪模型及其快速求解 ・4799・ 睾 一 Bz( ;+车 k—二 钏; + yT k)+A△ ’ (12) (13) 常数c=1,迭代次数为10,快速迭代算法式(15)中的参数 对式(12)两边关于五求导,经整理得到 一 )=A( 分别为A=0.005, :0.08,且上述三种求解算法的迭代过 程满足如下停机准则: I1 <5 xl0-4 又因为B =五+A△,则式(13)可转换成如下形式: ~ (16) 五( 一 )=A T k— )+Av ( :一 ) 实验是在Intel唧2.0 GHz CPU,1 GB内存,MATLAB 7.0的 其中:又由式(11.4)(11.5)得到 一 :6 一一b 与 :一 环境下进行,综合采用传统的量化指标峰值信噪比(peak sig・ :6:。。一b:,则 丘( 一 )=A ( 一b )+A :( 一b ) 即 五( 一 )= A (6 _。一6 )+A T i~一6;) (14) 令复原图像的初始条件为: =cf,b :b:=o,又因为 =Cu¨ ,五= ,则根据式(14)可得到子问题式(11.1)最终的 求解格式: “¨l:,~÷(“ T6 + ) 对于式(11 2)(11.3)两子问题的求解,可采用shrink算 子,即 =shrink(V +b ,_1) =shrink( “+6 ÷) 又因为shrink(c,÷)+c^ ut(c,÷)=c,^ 则式(11.2)~(11. 5)可简化为 6 =cut( +6 ,÷),^ 6 :cut(V, +6;,÷)^ 对于K>0,cE/R,本文定义算子cut如下: cul(c,÷):=c一音・m (Icl一÷,o) 综上所述,对于自适应全变分模型式(4)的求解,令 = f,b :b::0(k=1,2,…),有如下的快速迭代算(fastiteration algorithm,FIA): b :=cut(CV ¨ +6 ~,_1) (15.1) b :=cut(CV y¨ + ,T1) (15.2) ::,一÷( + ,T k) (15.3) Jja已证明:对于 =0,1,…,且b ,b:,” 由迭代步骤 (15.1) (15.3)得到,当0<A <1/8时,liaru = ’ 。 因五= c,C∈(0,1],则由0<A <1/8可得到0<C・A <1/8。对于任意的C∈(0,1],要使得0<C・A <1/8均成 立,则只需保证0<A <1/8便可使得FIA收敛至ATV的最 优解,即liaru =u 。 3实验结果及分析 为验证本文ATV与FIA的有效性,现以Lena、Camera— man和Testpatl图像为例进行实验,并与标准全变分模型 (standard total variation,STV)和张红英等人提出的全变分 自适应模型(Zhang total variation,ZTV)的图像去噪结果进 行比较分析。其中STV模型分别采用人工时间演化方法 (artiifcial time marching method,ATMM)、分裂Bregman迭代 算法(split Bregman iteration algorithm,SBIA)和FIA进行数 值计算,同时ZTV模型也利用ATMM进行求解。实验仿真 中,输入图像的大小均为256×256,灰度级为256,加性噪 声的标准差分别为20、3O和4O;并取扩散方程式(5)中的 nal to lqois ̄ratio,PSNtt)和最近受关注的平均结构相似度 (mean structural similarity,I ̄ISSIM)作为图像去噪质量的评价 标准。MSSIM指标认为人类视觉系统能高度自适应地识别图 像中的结构信息,用以衡量复原图像与清晰图像在亮度、对比 度和结构三方面的综合相似性,其取值为[0,1]。MSSIM为l 表明复原图像与清晰图像有着完全相同的结构信息;而 MSSIM为0则意味着两幅图像间无任何相似性。其中PSNR 和MSSIM的定义分别为 PsNR(“, ):10 log ̄0 dB (17) IJ u一 JJ 2 MssIM(“, )=÷∑.[z(“ , )] ・ [c(“ , )] ・[s(“ ,五 )] (18) 其中:“为清晰图像; 为去噪图像;M×N为图像大小; 为图 像分块数目; 与u 分别为去噪图像块与其对应的清晰图像 块。式(18)中亮度比较函数z(“ , )、对比度比较函数c( , )与结构比较函数s( ,瓦 )的详细定义请见文献[16]。本文 取MSSIM指标中调整三个比较函数所占比重的参数d、一一●一一一卢与 均为1,且最终的实验结果如图2—4和表1所示。● 一●~ 图3 去噪前后的cameramaIl图像(噪声标准差&-30) 图2、3中的图像分别为存在一定纹理结构的Lena和 Cameraman。由于sTV不完全符合图像处理的形态学原则,用 ATMM求解该模型时,含噪图像的复原演化迭代过程同时取决 于它的水平集(由7 u描述)和对应像素点的灰度值( ),使得 ■■-4800・ 计算机应用研究 第28卷 去噪图像中存在明显的分片常数效应。而Bregman迭代正则 化可被理解成“将噪声残量加回去”的技巧。基于该种正则化 方法构建的SBIA和FIA在求解STV模型时均能有效地提高 图像的去噪性能,一定程度上克服了 模型不完全符合图 像处理形态学原则的不足,避免分片常数效应的产生,且有着 相近的图像去噪效果。因ATV模型依图像的局部结构特征自 适应选取扩散系数从而更符合图像处理的形态学原则,较sTV 模型在PSNR和MSSIM方面有着更好的图像去噪效果,如表l 所示。考虑到梯度模值在描述图像局部结构特征的局限性, ZTV模型的图像含噪效果仅次于ATV模型。对于图4中只存 在边缘和平滑区域、没有其他纹理结构的Testpatl图像,边缘 更像是一种阶梯跳跃,与sTV模型产生的分片常数效应具有 一种类似的结构和性质,使得梯度模值能够很好地表征和刻画 此时的边缘跳跃特性。在表1中,STV+ATMM和ZTV十AT- MM得到的去噪图像(Test pat1)在PSNR方面优于其他三种方 法,但由于ATV模型更符合图像处理的形态学原则,此时ATV +FIA在MSSIM评价标准方面依然优于另外四种方法。 图4去噪前后的Tes ̄at1][像f噪声标准差 ̄--4o) 表1 图像去噪方法的实验结果比较 由表1和图2—4可知,含噪图像的纹理结构越复杂,ATV 较s rv和 模型表现出更好的图像复原质量;但当含噪图 像的纹理结构较简单时,五种方法在传统量化指标PSNR方面 并无优劣之分。由于ATV模型中的扩散系数是依图像的结构 信息自适应选取,更满足图像去噪的形态学原则,不管含噪图 像的纹理结构如何,ATV+FIA方法在图像平均结构相似性指 标MSSIM方面均优于其他四种方法。因A1 +FIA方法中决 定扩散强弱的系数 ,是一个动态变量,其求解速度相对于 STV+FIA稍慢,但明显比利用ATMM和SBIA求解 和 sTV模型要快,即在快速去除噪声的同时保留了图像的边缘轮 廓和纹理等结构信息,有利于图像处理的后续工作。 4结束语 为降低去噪模型中自适应参数对噪声的敏感性,在联合冲 击滤波器和非线性各向异性扩散滤波器对含噪图像做预处理 的基础上,借助于边缘检测算子区分含噪图像的边缘和平滑区 域以选取自适应参数,使构建的自适应全变分模型能同时兼顾 图像的平滑去噪与边缘保留。同时针对去噪模型求解算法普 遍存在的收敛速度慢的问题,引入一种基于Bregman迭代正则 化方法的快速迭代算法。实验结果表明,本文构建的自适应去 噪模型及其求解算法在快速去除噪声的同时保留了图像的细 节信息,得到的复原图像在峰值信噪比、平均结构相似度和主 观视觉效果方面有着明显提高。 参考文献: [1]RUDIN L 1,OSHER S,FATEM1 E.Nonlinear total vairation based noise removal algorithm[J].Physica D,1992,60(1-4):259—268. [2]GRASMAIR M.Locally adaptive total variation regularization[C]// Proc of the 2nd International Conference on Scale Space and Varia- tional Methods in Computer Vision.Berlin:Springer—Verlag,2009: 331.342. [3]SONG B.Topicla in variation PDE image se ̄nentation,inpainting and denoising[D] Los Angeles:University of Caliofrnia,2003. [4]张红英,彭启琮.全变分自适应图像去噪模型[J].光电工程, 2006,33(3):50 53. [5]OSHER S J,RUD1N L I.Feature—oriented image enhancement using shock filters[J].SIAM Journal on Numerical Analysis,1990,27 (4):919-940. [6]PERONA P,MALIK J.Scale—space and edge detection using anise- tropic diffusion[J].IEEE Trans on Pa钍em Analysis and Ma— chine Intelligence,1990,12(7):629-639. [7]AUBERT G,KORNPROBST P.Mathematical problems in image pro— cessing:partila differentila equations and the calculus of variations [M].New York:Springer,2009. [8]MARQUINA A,OSHER S.Explicit algorithms for a new time de— pendent model based on level set motion for nonlinear deblurring and noise removal[J].SIAM Journal on Scientific Computing,2000, 22(2):387 405. [9]VOGEL C,OMAN M.Iteration methods ofrtotal variation denoising[J]. SIAM Joumal on Scientific Computing,1996,17(1):227—238. [10]COLDSTE1N T,OSHER S.The split Bregman method for L1一regalari— zed problems[J].SIAM Journal on Imaging cSiences,20O9,2(2): 323—343. [11]JIA R Q,ZHAO H Q,ZHAO W.Convergence analysis ofthe Breg— mall method for the variation model of image denoising[J].Applied and Computational Harmonic Analysis,2009,27(3):367 379. [12]JIA R Q,ZHAOHQ.Afast algorithmforthetotal variationmodel of image denoising[J].Advances in Computational Mathematics, 2010,33(2):231—241. [13]BUADES A,CeLL B,MOREL J M.A non—local lagoirthm ofr image denoising[C]//Pro:of IEEE Conference Oll Computer Vision and Pat- tern Recognition.Washington DC:IEEE Computer Society,2005:60. 65. [14]ALVAREZ L,MAZORRA L.Singal and image restoration using shock iflters and anisotropie diffusion『J].SIAM Journal on Nu- merical Analysis,1994,31(2):590-605. 【15]ALMEIDA S C,ALMEIDA L B.Blind and semi・blind deblurring of naturla images[J].IEEE Trans on Image Processing,2010,19 (1):36—52. 【16]ZHOU W,ALNA C B,HAMID R S,et a1.Image quality assess— ment:from CiTer visibility to structural similarity[J].IEEE Trans on Image Processing,2004,13(4):600—612.