压缩感知的天文图像去噪算法

张 杰, 朱 奕, 史小平

(哈尔滨工业大学 控制与仿真中心, 哈尔滨 150080)

压缩感知的天文图像去噪算法

张 杰, 朱 奕, 史小平

(哈尔滨工业大学 控制与仿真中心, 哈尔滨 150080)

针对压缩感知迭代收缩阈值算法在图像处理中存在收敛速度慢和去噪性能差的缺陷,提出了一种改进的高性能迭代收缩阈值天文图像去噪重建算法.首先，使用经典最速下降法中的BB线性搜索步长算子加快迭代收缩阈值算法的收敛速度；其次，为了进一步提高重构天文图像的质量，在传统VisuShrink收缩阈值的基础上，提出一种下降VisuShrink收缩阈值对图像信息进行筛选；由于阈值去噪方法在迭代重建的过程中会导致重建的图像中出现伪吉布斯效应，最后采用循环平移的方法在每次迭代过程中对获取的重建图像进行调整. 多次的试验结果表明，与传统的压缩感知迭代收缩阈值算法相比,所提出的算法不仅能够获得较优的去噪性能和较快的收敛速度,同时可以有效地保护天文图像的特征和纹理等细节信息.此外，当选取的压缩采样比较低时,本算法也可以获得相对较高的峰值信噪比和视觉质量，进一步验证了本算法在天文图像去噪中的有效性.

收缩阈值; 天文图像; 去噪; 压缩感知; 循环平移

利用天文图像对外太空进行研究是深空探索的一个重要分支.从获得的天文图像可以直接得知某一星体上的地表结构,是否存在未知生命等重要信息.为了能够获得更多的天文信息,天文图像的采集一般都是采用高分辨率CMOS/CCD传感器,但是卫星或者其他的深空探测设备所携带的存储空间有限，存储这些高分辨率图像将会给有限的存储空间带来较大压力.为解决这一难题,天文图像在存储之前通常都要经过压缩处理.然而常用的JPEG/JPEG-2000方法[1-3]很难获得较低的图像压缩采样比.

另一方面,由于环境、拍摄条件等因素的影响,天文图像在采集和传输的过程中经常受到噪声信号的干扰,在地面接收站接收到的天文图像通常含有噪声.从接收到的图像中很难分辨出某一区域的实际地形特征.因此在接收到这些天文图像后,通常要进行去噪处理.然而目前大部分的去噪方法经常由于很难从高维数的信号观测值中获得足够有效的图像信息,导致重构信号的质量较差[4-5].为有效解决高维数信号的重建问题,学者们一直在探索如何从高维数信号中获得一种低维的信号结构,同时保证原始信号能从这种低维结构中获得精确重建.即,从信号中提取重要的信息同时舍弃非重要信息,直接使用这些重要信息重建高维数信号.近几年提出的信号稀疏性[5-6]可能是探索信号低维结构的一种比较简单方法.

基于信号的稀疏性，Donoho[7]提出了著名的压缩感知(compressed sensing,CS)理论.该理论一经提出,就引起了各领域的极大关注.它指出:如果信号在某一稀疏变换基上是稀疏的,则可以使用一个与该稀疏基不相关的低维测量矩阵对原信号进行观测,同时使用某一CS重建算法就可以从获得的少量信号观测值中精确重构原始信号.可以看出,信号在采集的同时就已经完成了压缩过程.传统的奈奎斯特/香农采样定理要求信号的采样速率必须大于或者等于两倍的信号带宽才能够精确重建原始信号,但是采样速率在低于两倍的信号带宽时,CS方法仍可以精确重建原始信号.因此,CS理论可以有效地解决高维数信号的重建问题,本文将其应用到高分辨率天文图像去噪中.

CS理论主要包含3个部分:稀疏变换、测量矩阵和重建算法.本文主要关注于如何设计一个高性能的重建算法.经过近几年的努力,学者们提出了许多的重建算法,例如线性规划类算法[8-9]、迭代收缩阈值类(iterative shrinkage thresholding,IST)算法[10-12]、梯度下降类方法[13]和贝叶斯类方法[14]等.在这些算法当中,IST算法设计简单且易于实现,更重要的是大部分的稀疏基都能较容易地应用到IST框架当中.这些优势使得IST算法经常受到学者们的青睐.然而该算法的收敛速度较慢,去噪性能也需要进一步提高.

为了提高梯度下降法的收敛速度,Barzilai等[15]提出了著名的BB线性搜索步长,并取得较快的收敛效果.本文将其应用到IST算法中调节其收敛速度.在图像去噪中,通常采用阈值去噪方法对图像信息进行筛选,本文提出一种下降VisuShrink收缩阈值筛选天文图像信息.阈值去噪虽然设计简单且能获得较好的去噪效果,但在奇异点(如边缘或者纹理)附近会出现较大的幅值振荡,最终导致重构的图像出现伪吉布斯现象[16].循环平移方法[17-18]可有效地减小或者消除这种幅值振荡,提高重构图像质量.本文将其应用到IST算法中,在迭代过程中对重构的天文图像进行调整.基于上述技术,本文提出了高性能的IST改进算法.实验结果表明,该算法可以以较快的收敛速度重构一幅清晰的天文图像.当压缩采样比较低时,该算法也具有较好的重建性能.

1 压缩感知去噪模型

假设某一N×1信号x是K-稀疏的[6-7],则可以用一个低维非自适应矩阵(又称为测量矩阵)Ф∈RM×N(M<

y=Φx+e.

(1)

从式(1)可以看出,由于M<

y=Φx+e=ΦΨΨ-1x+e=Θs+e,

(2)

式中s=Ψ-1x为稀疏系数.这里Θ=ΦΨ可以作为CS测量矩阵直接对s进行观测.测量矩阵Ф要求与稀疏基Ψ不相关,两者越不相关,需要的测量次数就越少,则可以获得更低的信号压缩采样比.

具有稀疏限制的l1范数最小化方法经常用来求解CS问题(2),即

或者转化为对稀疏系数s的求解,可表示为

式中第1项代表惩罚项,用来估算计算值与观测值之间的偏差；第2项为正则化项,表示原始信号的先验知识.

2 高性能IST算法

经典的IST算法迭代过程可描述为

xk+1=xk+HS(μkΦT(y-Φxk)).

式中：μk为线性搜索步长,为了计算方便通常设定为1,虽然简化了计算,但影响了算法的收敛速度；HS(·)表示阈值算子,通常考虑为硬阈值算子.其中

式中：q为迭代索引；Q为最大迭代次数.当q=0时,T为通用阈值.

2.1 BB线性搜索步长

最速下降法[20]的迭代过程可表示为

xk+1=xk-μkfk.

(3)

式中：fk=f(xk)为任一目标函数Γ在位置xk处的梯度向量；μk>0为线性搜索步长,它要求满足以下条件：

).

fk=ΦT(Φxk-f)，

和经典的SD迭代步长算子为

为了获得高性能的步长算子,文献[15]用前一次的迭代信息设定当前迭代所使用的步长算子,同时修改了式(3)的迭代过程,即

xk+1=xk-Dkfk,

式中Dk=μkI.其中I为单位向量.为了保证它具有某种拟牛顿特性,需要满足以下任一限制条件：

文献[20]证明了BB步长算子比SD步长算子更能有效地提升最速下降法的收敛速度.本文将BB步长算子应用到IST算法中调节其收敛速度.

2.2 循环平移方法

在图像去噪的过程中,当一幅图像包含有较多奇异点时,将会遇到如下问题:对于某一个奇异点具有较好去噪效果的平移量可能对另一个奇异点去噪效果较差.因此,对于包含较多纹理和边缘特征的天文图像,就很难获得针对所有奇异点都具有较好去噪效果的最佳平移量.循环平移方法可有效解决这一难题,其循环平移过程可描述为

式中K1、K2分别为沿行和列方向的最大平移量.在小波变换中，如果测试一幅N×N图像x且N=2K,则认为K1=K2=K为最大平移量.Ci,j(x)为定义的循环平移算子，对于二维图像x(m,n),可定义为

Ci,j(x)=x[mod(m+i)/N,mod(n+j)/N)].

C-i,-j(x)为Ci,j(x)的逆过程,可表示为

C-i,-j(x)=[Ci,j(x)]-1.

此外，S(x)为小波变换过程,S-1(x)为小波逆变换.θ(x)为阈值函数,本文将其考虑为硬阈值函数HS(x),并且阈值T为下降VisuShrink收缩阈值.

2.3 本文算法

综上所述,本文算法的设计步骤概括如下：

10 月29 日上午，由山东省旅游饭店协会和山东新业态旅游住宿业分会主办的首届山东省文化主题饭店发展论坛在泰安铭座三泰宾馆开幕。会场内“大腕”云集，在一天的时间里，业内实践专家围绕有关文化、生活、酒店三个议题展开了讨论，共谋文化主题酒店的未来发展路径。

1)初始化过程.初始化迭代索引q=0,最大迭代次数Q=30和重构图像xq=0.

2)计算下降VisuShrink收缩阈值和BB步长算子μq.

3)使用下式更新估计值.

xq+1=xq+HS(μqΦT(y-Φxq)).

4)对重构的图像进行循环平移,可表示为

).

6)q=q+1.如果q=Q,输出重构图像;否则,进入步骤2).

3 实验及分析

从图1可以看出,与其他算法相比,本文算法能获得较好的视觉质量和较高的峰值信噪比(PSNR),但花费的重构时间(reconstruction time,RT)较长.同时可以看出,IST-TV、IST-CS和本文算法都能有效地抑制伪吉布斯效应.

设定噪声标准差和压缩采样比不变,如图3所示不同算法随迭代次数增长获得的PSNR值.可以看出,本文算法具有比其他算法更快的收敛速度.

图1 不同算法重构得到的结果

图2 不同算法重构得到变化曲线

图3 不同迭代次数下获得的PSNR

图4 不同算法获得的重构结果

表1 压缩采样比变化时获得的PSNR

4 结 论

1)本文将压缩感知理论应用到天文图像去噪,并在迭代收缩阈值算法的基础上,提出了一种高性能改进算法.该算法首先使用BB步长算子调节收敛速度;其次使用提出的下降VisuShrink阈值对重构图像进行筛选;最后对重建图像进行循环平移以消除伪吉布斯效应.

2)本文算法与IST、IST-TV和IST-CS算法的对比分析结果表明,本文算法可以以较快的收敛速度重构一幅清晰的天文图像,并且能有效地保护天文图像的细节特征.在压缩采样比较低时,本文算法也可获得较优的重构效果.

3)虽然本文算法的收敛速度和去噪重建性能得到很大的提高,但是花费的重建时间较长,因此如何提高本文算法的重建速度是以后改进的方向.

图5 两种算法重构结果对比

Fig.5 Comparison of the reconstructed result between two algorithms

[1] SKODRAS A, CHRISTOPOLOS C, EBRAHIMI T. The JPEG 2000 still image compression standard[J]. IEEE Signal Processing Magazine, 2001, 18(5): 36-58.DOI: 10. 1109/79.952804.

[2] WALLANCEG K. The JPEG still picture compression standard[J]. IEEE Transactions on Consumer, 2002, 38(1): 18-34. DOI: 10.1109/30.125072.

[3] SHI Xiaoping, ZHANG Jie. Reconstruction and transmission of astronomical image based on compressed sensing[J]. Journal of Systems Engineering and Electronics, 2016, 27(3):680-690.DOI: 10.1109/JSEE.2016.00071.

[4] ELDARY C, KUTYNIOK G. Compressed sensing: theory and applications[M]. New York: Cambridge University Press, 2012:1-515.

[5] BENEDETTO J J. Compressed sensing and its applications[M]. USA: Birkhauser, 2013:97-143.

[6] CANDES E, ROMBERG J. Sparsity and incoherence in compressive sampling[J]. Inverse Problems, 2007, 23(3): 969-985.DOI:10.1088/0266-5611/23/3/008.

[7] DONOHO D L. Compressed sensing[J]. IEEE Transactions on Information Theory, 2006, 52(4): 1289-1306.DOI: 10.1109/TIT.2006.871582.

[8] CANDES E J, TAO T. Decoding by linear programming[J]. IEEE Transactions on Information Theory, 2009, 51(12): 4203-4215.DOI:10.1109/TIT.2005.858979.

[9] FIGUEIREDOM A T, NOWAK R D, WRIGHT S J. Gradient Projection for sparse reconstruction: application to compressed sensing and other inverse problems[J]. IEEE Journal of Selected Topics in Signal Processing, 2007,1(4):586-597.DOI:10. 1109/JSTSP.2007.910281.

[10]BLUMENSATH T, DAVIES M E. Iterative hard thresholding for compressed sensing[J]. Applied and Computational Harmonic Analysis, 2009, 27(3):265-274. DOI:10.1016/j. acha.2009.04.002.

[11]CAI Jianfeng, OSHER S, SHEN Zuowei. Linearized bregman iterations for compressed sensing[J]. Mathematics of Computation, 2009, 78(267): 1515-1536. DOI:10.1090/S0025-5718-08- 02189-3.

[12]DAUBECHIES I, DEFRISE M, MOL C D. An iterative thresholding algorithm for linear inverse problems with a sparisity constraint[J]. Communications on Pure and Applied Mathematics, 2004, 57(11):1413-1457.DOI:10.1002/ cpa.20042.

[13]GARG R, KHANDEKAR R. Gradient descent with sparsification: an iterative algorithm for sparse recovery with restricted isometry property[C]//Proceedings of the 26th Annual International Conference on Machine Learning. New York, NY: ACM, 2009: 337-344. DOI:10.1145/1553374.1553417.

[14]JI Shihao, XUE Ya, CARIN L. Bayesian compressive sensing[J]. IEEE Transactions on Signal Processing, 2008, 56(6): 2346-2356. DOI: 10.1109/TSP.2007.914345.

[15]BARZILAI J, BORWEIN J M. Two point step size gradient methods[J]. IMA Journal of Numerical Analysis, 1988, 8(1):141-148.DOI: 10.1093/imanum/8.1.141.

[16]王蓓，张根耀，李智，等. 基于新阈值函数的小波阈值去噪算法[J].计算机应用,2014, 34(5): 1499-1502.DOI:10. 11772/j.issn.1001-9081.2004.05.1499.

WANG Pei, ZHANG Genyao, LI Zhi, et al. Wavelet threshold denoising algorithm based on new threshold function[J]. Journal of Computer Applications, 2014, 34(5): 1499-1502.DOI: 10.11772/j.issn.1001-9081.2004.05.1499.

[17]BINHN T, KHARE A. Multilevel threshold based image denoising in curvelet domain[J]. Journal of Computer Science, 2010, 25(3): 632-640. DOI:10.1007/s11390- 010-9352-y.

[18]郭海涛，赵红叶，徐雷，等. 基于循环平移和DTCWT的声呐图像滤波方法[J]. 仪器仪表学报，2015，36(6): 1351-1356.DOI: 10.3969/j.issn.0254-3087.2015.06.020.

GUO Haitao, ZHAO Hongye, XU Lei, et al. Sonar image filtering method based on cycle shift and DTCWT[J]. Chinese Journal of Scientific Instrument, 2015,36(6): 1351-1356.DOI: 10.3969/j.issn.0254-3087.2015.06.020.

[19]CANDES E J, ROMBERG J, TAO T. Robust uncertainty principles: exact signal reconstruction from highly incomplete information[J]. IEEE Transactions on Information Theory, 2006, 52(2): 489-509. DOI: 10.1109/ TIT.2005.862083.

[20]YUAN Yaxiang. A new stepsize for the steepest descent method[J]. Journal of Computational Mathematics, 2006, 24(2):149-156.DOI: 10.1063/1.4882499.

Compressedsensingdenoisingalgorithmforastronomicalimage

ZHANG Jie, ZHU Yi, SHI Xiaoping

(Control and Simulation Center, Harbin Institute of Technology, Harbin 150080, China)

In the deep space exploration, astronomical image acquisition, transmission and processing have always been the focus of research. To solve problems of slow convergence speed, poor denoising performance in compressed sensing iterative shrinkage-thresholding algorithm for image processing, an improved iterative shrinkage-thresholding astronomical image denoising and reconstruction algorithm with high performance is proposed. Firstly, the BB linear search stepsize of the classical steepest descent algorithm is used to accelerate the convergence speed of iterative shrinkage-thresholding algorithm; secondly, to further improve the reconstructed astronomical image quality, based on the classical VisuShrink shrinkage-threshold, a decreasing VisuShrink shrinkage-threshold is proposed to select the image information; since the pseudo-gibbs effect caused by threshold denoising method will appear in the process of image reconstruction, the cycle spinning method is finally employed to adjust the reconstructed image in each iteration. Multiple experimental results show that, compared with the traditional compressed sensing iterative shrinkage-thresholding algorithm, the algorithm proposed can not only obtain better denoising performance and faster convergence speed, but also effectively protect the astronomical image detail information, such as feature and texture. In addition, when compression sampling ratio is lower, the algorithm proposed also can obtain relatively higher peak signal to noise ratio and visual quality, proving the effectiveness of the algorithm proposed for astronomical image denoising.

shrinkage-threshold; astronomical image; denoising; compressed sensing; cycle spinning

10.11918/j.issn.0367-6234.201609002

TN911.73

A

0367-6234(2017)10-0078-05

2016-09-01

国家自然科学基金(61074127)

张 杰(1986—),男,博士研究生；

史小平(1965—),男,教 授,博士生导师

史小平,sxp@hit.edu.cn

(编辑张 红)