基于结构化稀疏模型的压缩感知重构改进算法
来源:化拓教育网
Computer Engineering andApplications' ̄J"算机工程与应用 2013,49(14) 203 @信号处理@ 基于结构化稀疏模型的压缩感知重构改进算法 杨爱萍,栗 改,侯正信,庞 茜 YANG Aiping,LI Gai,HOU Zhengxin,PANG Qian 天津大学电子信息工程学院,天津300072 School of Electronic Information Engineering,Tianjin University,Tianjin 300072,China YANG Aiping,LI Gai,HOU Zhengxin,et a1.New recovery algorithm for compressed sensing based on structured sparse mode1.Computer Engineering and Applications,2013,49(14):203—206. Abstract:Recently,normal recovery algorithms for CS only use signal and image sparse priors under wavelet,make no use of the tree structure priors.In order to reconstruct the original signal quickly and accurately,this paper brings the tree structure sparse model into SP algorithm,CoSaMP-algorithm and gets the improved recovery algorithm for compressed sensing.Combin・ ing with structured sparse model and dual-tree complex wavelet transform,a new recovery algorithm for CS is proposed.The simulated results show that the algorithm can achieve higher reconstructed image performance. Key words:Compressed Sensing(CS);stuctrured sparse model;dual—tree complex wavelet transform 摘要:目前,标准的CS重构算法仅利用信号和图像在小波变换下的稀疏先验信息,而并没有利用变换系数具有的结构 化特性。为了能够快速精确地重建原始信号,将结构化稀疏模型与sP算法、CoSaMP算法相结合,提出了压缩感知重构的 改进算法。另外,将基于双树复小波变换的系数结构模型融入上述算法,进一步提高重构性能。实验结果表明,所提出的 算法可获得更高的图像重建质量。 关键词:压缩感知;结构化稀疏模型;双树复小波变换 文献标志码:A 中图分类号:TP391 doi:10.3778/j.issn.1002.8331.1202.0550 1 引言 重构算法作为压缩感知理论 中的关键环节,一直备 受关注。近几年来,提出了多种重构算法:基追踪(BP)算 法 ,匹配追踪(MP)算法 】,正交匹配追踪(OMP)算法 ],子 了压缩感知重构新算法。实验结果表明,本文提出的新算 法重建精度更高,运算时间更短。 2 CS模型 CS理论 指出,如果一个信号在某变换下是稀疏或可 空间(sp)算法 和压缩采样匹配追踪(CoSaMP)算法 等。 这些标准CS算法仅利用信号和图像在某变换下系数的稀 疏性,而并未利用变换系数的结构分布特点。针对图像小 波变换后的树结构,Baraniuk等人提出了基于小波树结构 的模型化方法 ,证明了将描述信号小波系数分布的结构 模型引入到cs重构理论的正确性及有效性。文献[8]将信 号小波系数呈树状分布用于MP和OMP算法中。文献[9] 提出了小波系数块分布模型并用于OMP算法。本文将小 压缩的,那么它可以通过在一个与变换矩阵不相关的空间 基上的少量投影恢复出来。cs模型的矩阵表示形式为: J,= (1) 其中Y∈R 表示观测向量, 是 ×Ⅳ(M《N)的测量 矩阵,且 是固定的、不依赖于信号 ,其列向量称为原 子。 eR№。是一个Nx 1的离散实值信号向量且在基 = 1,l;f, ,…, .v]下是稀疏的,即 Ⅳ =波系数结构模型与重构质量较好的sP算法相结合提出相 应的改进算法;在此基础上,针对小波变换的缺陷,将基于 双树复小波变换 ” 的系数结构模型融入上述算法,提出 基金项目:国家自然科学基金(No.61002027)。 E 瓤= 。 (2) 其中 是Nx 1的系数矩阵, 是内积S :<x, >= 作者简介:杨爱萍,女,博士,副教授,研究领域:图像及视频处理、压缩感知理论及应用;栗改,女,硕士生,研究领域:压缩感知理论及应 用;候正信,教授,博导,研究领域:图像及视频处理;庞茜,硕士,研究领域:压缩感知理论及应用。E-mail:ligai825@163.corn 收稿日期:2012—03-19 修回日期:2012—05.07 文章编号:1002.833I(2013)14.0203.04 CNKI出版日期:2012—08—02 http://www.cnki.net/kcms/detail/11.2127.TP.20120802.0900.001.html 204 2013,49(14) Computer Engineering andApplications计算机工程与应用 算树中以每个节点根的每个子树小波系数平均值的绝对 值,把绝对值中的最大值作为该节点的能量(称此能量最 大的节点为超节点),并保留超节点对应的子树的全部系 数,最优子树集就由这些系数组成,从而实现树结构最优 的思想。 。向量X和 是对同一个信号的等价表示,只是表达 方式不同,X是时间域或空间域的,s是 域的。将式(2) 代入式(1)得: Y=CI ̄X= =Os (3) 其中O= ,O∈R ,称为CS信息算子。由于观测维 数 《N,所以从J,的 个观测值中解出信号X是很困 将基于小波树结构的模型化方法和sP算法相结合,形 成Mb—SP算法。Mb.SP算法步骤: 难的。由于式(3)中 仅有 个非零系数,这就为求解X 提供了可能。通过求解式(3)的逆问题解得 ,然后再将s 代入式(2)即可求得 。因此CS理论重构问题的实质就 (1)初始化:初始残差rn=J,,候选向量0 =0∈R ,迭 代次数为t ,迭代变量f=1,结构化稀疏估计为 。 是在已知测量向量Y和测量矩阵 的条件下,如何快速、 准确和稳定地重构原始信号 。 3基于小波树结构的模型化方法 经研究发现,在相同稀疏度和测量数目条件下,贪婪 算法中的子空间(sp)算法和压缩采样匹配追踪(CoSaMP) 算法在整体性能上优于其他算法。本章基于小波树结构 模型提出Mb.SP算法和Mb.CoSaMP算法。 3.1 SP算法 sP算法步骤如下: (1)初始化:初始残差 =J,,候选向量0 =0∈R ,迭 代次数为t ,迭代变量t=1。 (2)更新索引:计算 = 一 ,寻找 中K个最大元 素的位置A =sup( 『】),新的索引J=i,= U sup(0f_1)。 (3)更新候选向量:计算 = J,,新的候选向量0 = l。 (4)更新余差:r = —q}0,。 (5)迭代次数加1,验证迭代停止的条件。若t<t ,则 返回步骤(2);否则执行步骤(6)。 (6)输出重建信号: ,= 。其中 ∈R价 为测量 矩阵, ∈R 为基矩阵。Y∈R 为观测向量,信号x∈R 的稀疏度为 。 3.2 Mb.SP算法和Mb.CoSaMP算法 图像经小波变换后自然形成小波系数的四叉树结构, 边缘和纹理对应的大系数和光滑部分的小系数通常聚集 在树的分支上。具体地,如果某一个尺度下的系数很小, 那么此系数的子系数也通常较小,这说明了小波系数具有 树结构稀疏性。将稀疏度为 拘树结构稀疏信号定义为 : r ,一1 2i ] FK=X=roy+ ̄2Wi,j ̄Lti,j:WIgf=0,1Q1=K} (4) L i 0J 1 J 由式(4)可知Q中的小波系数形成了相连的子树集,而不 属于 中的小波系数近似为零,即, 为由 中小波系数 子树集组成的树结构稀疏信号,且满足稀疏度为 。实现 树结构稀疏信号 重构,需用最佳 项树结构稀疏逼近 取代普通的 项稀疏逼近,即求解以下优化问题: S(x,K)=argminllx一 , (5) 采用压缩分类选择算法(CSSA) 求解式(5)。首先计 (2)更新索引:计算 = r ,寻找 中K个最大元 素的位置A =sup( 1..),新的索引 =/1,u sup(0 )。 (3)更新候选向量:计算 = Y,新的候选向量0 = f 。 (4)更新余差:,,=J,一 ,。 (5)更新估计信号: 。 (6)迭代次数加1,验证迭代停止的条件。若f<t 返回 步骤(2);否则执行步骤(7)。 (7)输出重建信号X 。 类似地,把CoSaMP算法和小波树结构的模型化方法 相结合得到Mb.CoSaMP算法。 4基于双树复小波变换的新算法 4.1 双树复小波变换(DT_CwT) 针对传统离散小波不具有平移不变性和方向选择性 差的缺点,Kingshury等人提出在同一数据上用两个的 小波变换平行作用来实现复数小波变换 。二维双树复小 波变换如图1所示。它包含两个平行的小波树,其中树a的 滤波器组表示复数小波变换的实部,树b的滤波器组表示 复数小波变换的虚部。易知,其由四个平行的2维离散小 行 列 低频 }高频 ( ,Y) 图1 二维双树复小波变换 206 2013,49(14) Computer Engineering andApplications计算机工程与应用 法和Mb—CoSaMP算法相结合,提出DTC.Mb.SP算法和 还是主观质量提出的新算法都得到了显著改善。 窍 Z∞ 为了进一步验证算法性能,比较它们在不同压缩率下 DTC・Mb-CoSaMP算法,重建效果得到进一步改善。 的峰值信噪比。图3给出了在压缩率分别为M/N:0.1, 0.2,…,0.8时各算法的PSNR值。 参考文献: [1]Donoho D L.Compressed sensing[J].IEEE Transactions on Information Theory,2006,52(4):1289—1306. 【2]Chen S,Donoho D L,Saunders M A.Atomic decomposition by basis pursuit[J].SIAM J Sci Comp,1999,20(1):33—61. 【3]Mallat S G,Zhang Z F.Matching pursuits with time—requency fdictionaries[J].IEEE Transactions on Signal Processing,1 993, 41(12). [4]Davenport M A,Wakin M B.Analysis of o ̄hogonal match— ing pursuit using the restricted isometry property[J].IEEE Transactions on Information Theory,20 1 0,56(9). [5】Dai W,Olgica M.Subspace pursuit for compressive sensing: 压缩率 closing the gap between performance and complexity[J]. IEEE Transactions on Information Theory,2009,55(5): 2230-2249. (a)几种算法峰值信噪比的比较1 [6]Needell D,Tropp J A.COSAMP:iterative signal recovery from incomplete and inaccurate samples[J].Appl Comput Harmon Anal,2008:301-321. [7]Baraniuk R G,Cevher V,Dunce M F,et a1.Model—based compressive sensing[J].IEEE Transactions oil Information Theory,2010,56(4):1982—2001. [8]Dua ̄e M F.Fast reconstruction from random incoherent pro— jections[R].Houston:Rice Universiy,t2005. [9]Yonina C E,Helmut B.Block—sparsity:coherence and effi— 压缩率 cient recovery[C]//IEEE International Conference on Acous— tics,Speech and Signal Processing,2009:2885—2888. (b)几种算法峰值信噪比的比较2 图3几种算法比较 [10]Kingsbury N G.The dual—tree complex wavelet transform: a new eficifent tool for image restoration and enhancement[C]# 由图3可以看出,在各个压缩率下,DTC.Mb.SP算法 Proc European Signal Processing Conf,Rhodes,1998. 和DTC.Mb.CoSaMP算法表现出更加优异的重建性能。 [11】Selesnick I W,Baraniuk R G,Kingsbury N C.The dual—rtee complex wavelet transform[J].IEEE Signal Processing Maga— 6结论 基于小波树结构的模型化方法Mb.SP和Mb—CoSaMP 算法充分利用了变换系数的结构特点,在相同的迭代次数 zinc,2005:123—151. [1 2】Hadeel N A.Image denoising based on dual—tree complex wavelet transform[J].Journal of Engineering and Applied Sciences.2008.31(3):587—590. 下可以在较短时间内完成高精度的信号重建。双树复小 波具有很好的平移不变性和方向选择性,将其与Mb.SP算 (上接177页) [7]Hoppe H,DeRose T,Duchamp T,et a1.Mesh optimization[C]// Proc of the SIGGRAPH’93.Anaheim:ACM Press.1993: l9.26. [13】杨威,高协平.基于双树复小波变换的微钙化诊断方法[J].光 子学报,2010,39(6):1040—1046. the VRST’96.Hong Kong:ACM Press,1996:11—19. [12]周晓云,何大曾,刘慎权.一种层次化模型构造方法[J].中国图 象图形学报,1997,2(8/9):555.561 [13】郝娟儿,唐莉萍,曾培峰.基于曲率和面积的二次误差测度网 格简化算法【J].东华大学学报:自然科学版,2012,38(3): 3l8-322. [8]Hamann B.A data reduction scheme for triangulated surface[J]. Computer Aided Geometric Design,1994,11(3):197—214. [9]Lounsbery M,DeRose T.Multiresolution analysis for surfaces [14]Antonio A,Miguel A.A flexible similarity measure for 3D shapes recognition[J].IEEE Transactions on Pattern Analy— sis and Machine Intelligence,2004,26(11):1507—1520. of arbitrary topological type[R].Washington:University of Washington,1 994. [1 0]Hoppe H.Progressive meshes[C]//Proc of the SIGGRAPH’96. New York:ACM Press,1996:99—108. [15】周昆,潘志庚,石教英.基于三角形折叠的网格简化算法【J].计 算机学报,1998,21(6):506.513. [16]薛峰,袁成风.保持外形特征的网格简化算法[J].计算机应用, 2010,30(9):243l一2433. [1 1]Isler V,Lau R W H,Green M.Real—time multi—resolution modeling for complex virtual environments[C]//Proc of