霍雷剛 馮象初
?
基于主成分分析和字典學(xué)習(xí)的高光譜遙感圖像去噪方法
霍雷剛*馮象初
(西安電子科技大學(xué)數(shù)學(xué)與統(tǒng)計學(xué)院 西安 710126)
高光譜圖像變換域各波段圖像噪聲強度不同,并具有獨特的結(jié)構(gòu)。針對這些特點,該文提出一種基于主成分分析(Principal Component Analysis, PCA)和字典學(xué)習(xí)的高光譜遙感圖像去噪新方法。首先,對高光譜數(shù)據(jù)進行PCA變換得到一組主成分圖像;然后,對信息量較小的主成分圖像分別采用基于自適應(yīng)字典的稀疏表示方法和對偶樹復(fù)小波變換方法去除空間維和光譜維的噪聲;最后,通過PCA逆變換得出去噪后的數(shù)據(jù)。結(jié)合主成分分析和字典學(xué)習(xí)的優(yōu)勢,該文方法相對于傳統(tǒng)方法對高光譜圖像具有更好的自適應(yīng)性,在細節(jié)得到保留的同時有效地抑制了斑塊效應(yīng)。對模擬和實際高光譜遙感圖像的實驗結(jié)果驗證了該文方法的有效性。
圖像處理;高光譜圖像;去噪;主成分分析;稀疏表示;字典學(xué)習(xí)
近年來,高光譜遙感圖像受到廣泛關(guān)注,并成功應(yīng)用到地質(zhì)勘探、考古發(fā)現(xiàn)、森林火災(zāi)檢測、軍事作戰(zhàn)等多個領(lǐng)域。然而,由于在獲取、傳輸過程中受到各種因素的影響,高光譜遙感圖像的噪聲比較嚴重。這些噪聲對后續(xù)的解混[1]、分類[2]和識別[3]等造成嚴重影響,制約了高光譜圖像的廣泛應(yīng)用。因此,圖像去噪技術(shù)是高光譜遙感應(yīng)用中非常重要的關(guān)鍵技術(shù)之一,也是遙感科學(xué)和信息科學(xué)的研究熱點。
與全色圖像、多光譜圖像相比,高光譜圖像的噪聲更加復(fù)雜。一方面,高光譜圖像的波段數(shù)較多,各波段圖像受噪聲影響的程度不同;另一方面,由于光譜分辨率較高(單個波段的光譜頻率范圍較窄),各波段之間的相關(guān)性較強。上述特點使得傳統(tǒng)的全色圖像、彩色圖像和多光譜圖像去噪方法不能很好地用于去除高光譜圖像中的噪聲。因此,有必要針對高光譜圖像的特點,設(shè)計更加合理的去噪方法。根據(jù)是否對高光譜數(shù)據(jù)進行光譜維變換,傳統(tǒng)的高光譜圖像去噪方法基本上可以分為兩大類:(1)直接對獲取的高光譜數(shù)據(jù)在空間維和光譜維建立去噪模型。例如:文獻[4]把高光譜圖像看作超立方體,研究了基于3維各向異性擴散的高光譜圖像增強方法;文獻[5]提出光譜-空間維自適應(yīng)的總變差模型;文獻[6]在最大后驗估計框架下提出空間-光譜維混合先驗?zāi)P偷龋?2)先將高光譜數(shù)據(jù)進行光譜維變換,在變換后的空間維和光譜維進行去噪,然后進行逆變換得到去噪后的數(shù)據(jù)。例如:文獻[7]提出在光譜微分域采用小波閾值收縮方法去噪;文獻[8]提出在光譜微分域采用BayesShrink算法和Savitzky- Golay濾波進行去噪;文獻[9]提出一種基于主成分分析和小波閾值收縮的高光譜圖像去噪方法等。
以上方法考慮了高光譜圖像的特點,但沒有充分利用高光譜數(shù)據(jù)本身的特性。例如,文獻[9]對主成分圖像采用對偶樹復(fù)小波變換,這些事先定義的、固定的解析字典沒有充分考慮數(shù)據(jù)本身的屬性,使得去噪方法對每個主成分圖像去噪的自適應(yīng)性受到限制。
針對上述缺陷,本文提出一種基于主成分分析和字典學(xué)習(xí)的高光譜圖像去噪新方法。一方面,新方法充分利用PCA變換的優(yōu)點,前幾個主成分圖像攜帶的主要信息得以有效保留;另一方面,新方法根據(jù)當(dāng)前數(shù)據(jù)自適應(yīng)地學(xué)習(xí)字典,結(jié)合稀疏表示方法對每個噪聲成分圖像進行去噪。由于字典是自適應(yīng)學(xué)習(xí)的,更加全面地反映了數(shù)據(jù)的特殊性。
本文的結(jié)構(gòu)安排為:第2節(jié)詳細敘述了本文提出的方法;第3節(jié)是對實驗結(jié)果的分析;結(jié)論在第4節(jié)給出。
如前所述,本文方法的基本原理是綜合利用光譜維變換和字典學(xué)習(xí)的優(yōu)點來解決高光譜圖像去噪的問題。不失一般性,本文結(jié)合經(jīng)典的主成分分析[10]和基于K-奇異值分解(K-Singular Value Decomposition, K-SVD)算法[11]的圖像去噪模型[12]給出一種新的高光譜圖像去噪方法。
本文的方法主要包括如下4個步驟:
主成分分析[10]是一種最基本的數(shù)據(jù)降維方法,主要是通過對數(shù)據(jù)的協(xié)方差矩陣進行特征值分解,特征值分解后的特征向量對應(yīng)數(shù)據(jù)的主成分,特征值對應(yīng)數(shù)據(jù)在各個主成分上的權(quán)重。因此,PCA變換是一種線性變換,變換后各主成分分量彼此不相關(guān),且隨著主成分編號的增加該分量包含的信息量減小。同其他基于光譜維變換的高光譜圖像去噪方法一樣,本文的方法主要是利用PCA變換去除高光譜圖像波段間的相關(guān)性,去除冗余信息。對高光譜圖像進行PCA變換后,大部分信息集中在前幾個主成分分量中,其他的主成分分量以噪聲為主,本文稱之為噪聲成分分量。通??梢灾蝗∏皫讉€信息量大的主成分分量而丟掉信息量小的噪聲成分分量。值得注意的是,噪聲成分分量仍包含一些有用信息,如果丟棄噪聲成分分量,重構(gòu)的高光譜圖像將會有信息損失。針對以上情況,文獻[9]提出保留信息量大的前幾個主成分分量,對噪聲成分分量進行去噪,然后取處理后的所有主成分分量進行PCA逆變換。本文也采用這種方式。
小波變換由于具有良好的時頻特性,在信號和圖像處理領(lǐng)域應(yīng)用廣泛。傳統(tǒng)的基于閾值收縮的小波去噪方法考慮了小波系數(shù)的大小但忽略了小波系數(shù)的結(jié)構(gòu)信息,很多學(xué)者對其進行改進,如采用雙變量收縮的方法[15],結(jié)合稀疏表示的方法[16]等。根據(jù)信號的先驗知識,可以設(shè)計出合適的字典。進一步,考慮信號樣本,可以優(yōu)化這些字典[17]。近年來,稀疏表示理論和字典學(xué)習(xí)算法的研究取得了很大的進展。字典學(xué)習(xí)的方法因其對不同圖像的自適應(yīng)性獲得了比采用固定字典方法更好的效果,已被成功地用于圖像去噪[12,18]、Bayer格式圖像彩色恢復(fù)[19]、高光譜遙感圖像分類[20]等。其中,K-SVD算法[11]是目前最具代表性的字典學(xué)習(xí)方法之一。
經(jīng)過適當(dāng)?shù)爻跏蓟鲜鰞?yōu)化問題中的待求變量可以采用交替迭代的方法按如下步驟求解[12]:
步驟1和步驟2交替迭代進行,直到滿足停止條件(達到指定最大迭代次數(shù)或誤差)。
由于噪聲強度隨波長變化,為進一步比較不同方法在每個波段上的去噪性能,圖2顯示了第(2)種情況和第(3)種情況下(即噪聲強度隨波段變化的情況)每隔5個波段含噪聲數(shù)據(jù)和不同去噪方法去噪后數(shù)據(jù)在各個波段上的信噪比之差(峰值信噪比之差的比較與此類似),其他未顯示的波段的比較結(jié)果與此類似。從中可以看出,BandBishr2D方法去噪的性能偏低,BandK-SVD2D 方法較BandBishr2D方法效果有所提升,部分波段達到甚至超過了4種方法在該波段達到的最高信噪比(例如第1個波段)。PCABishr2D 方法和本文方法在所有波段都取得了很好的去噪效果,整體比較穩(wěn)定且大多數(shù)波段高于前兩種方法。其中,本文方法去噪效果較PCABishr2D 方法又有所提升,并且對大部分波段提升效果明顯。
表1模擬數(shù)據(jù)上4種方法去噪性能比較(所有波段的噪聲強度都相同的情況)
評價指標(biāo)噪聲圖像BandBishr2DBandK-SVD2DPCABishr2DPCAK-SVD2D (a)(b)(a)(b)(a)(b)(a)(b) 5SNR386.57971.67972.192409.042413.554311.918632.828430.0110177.39 PSNR(dB)34.1738.1738.1842.1242.1344.6547.6647.5648.38 10SNR98.44334.59334.89773.49789.301171.622496.782818.732913.52 PSNR(dB)28.2333.5433.5537.1837.2738.9942.2742.8042.94 15SNR44.90185.93186.13399.88399.38531.841163.641373.881852.71 PSNR(dB)24.8230.9931.0034.3234.3135.5638.9639.6840.98 20SNR25.94123.55123.73229.05206.17302.45655.73744.96997.47 PSNR(dB)22.4429.2229.2231.9031.4433.1136.4737.0238.29 25SNR17.1090.5990.74150.01163.33194.90418.28401.88546.36 PSNR(dB)20.6327.8727.8830.0630.4331.2034.5134.3435.67
注:(a)列為光譜維不進行處理時的結(jié)果;(b)列為光譜維去噪后的結(jié)果。
表2模擬數(shù)據(jù)上4種方法的去噪性能比較(噪聲強度隨波段變化的情況)
評價指標(biāo)噪聲圖像BandBishr2DBandK-SVD2DPCABishr2DPCAK-SVD2D (a)(b)(a)(b)(a)(b)(a)(b) SNR217.18643.52643.881576.891584.152474.084563.185912.816938.06 PSNR(dB)31.6436.3836.3940.2840.3042.2344.8946.0246.71 SNR361.69874.83875.482235.002202.483870.566720.268979.139776.35 PSNR(dB)33.8837.7237.7241.7941.7344.1846.5747.8348.20
圖2 模擬數(shù)據(jù)上不同去噪方法去噪后數(shù)據(jù)和含噪聲數(shù)據(jù)的信噪比差值比較
圖3 真實數(shù)據(jù)上不同算法的去噪結(jié)果比較
表3真實數(shù)據(jù)和不同方法去噪后數(shù)據(jù)的MAQI指標(biāo)
方法真實數(shù)據(jù)BandBishr2DBandK-SVD2DPCABishr2DPCAK-SVD2D MAQI0.01640.01670.01680.01660.0177
圖4采用不同方法去噪后數(shù)據(jù)和真實數(shù)據(jù)各波段的AQI指標(biāo)差值比較
高光譜圖像變換域各波段噪聲強度不同,為利用各波段圖像的獨特結(jié)構(gòu)并提高去噪性能,本文提出了一種基于主成分分析和字典學(xué)習(xí)的高光譜遙感圖像去噪新方法。該方法根據(jù)當(dāng)前數(shù)據(jù)學(xué)習(xí)自適應(yīng)字典,利用稀疏表示方法對噪聲成分圖像去噪。由于字典反應(yīng)了圖像本身的結(jié)構(gòu),本文方法在保留細節(jié)的同時有效抑制了斑塊效應(yīng)。該方法可以推廣到采用其他光譜維變換和字典學(xué)習(xí)的方法,也可以將本文方法加以改進,針對非高斯噪聲的情況設(shè)計合理的去噪模型或算法。
[1] Eches O, Benediktsson J A, Dobigeon N,Adaptive markov random fields for joint unmixing and segmentation of hyperspectral images[J]., 2013, 22(1): 5-16.
[2] Bai Jun, Xiang Shi-ming, and Pan Chun-hong. A graph-based classification method for hyperspectral images[J]., 2013, 51(2): 803-817.
[3] Li Qing-ting, Lu Lin-lin, Zhang Bing,The recognition of altered rock based on spectral modeling and matching using hyperspectral data[C]. Proceedings of 2nd IEEE International Conference on Remote Sensing, Environment and Transportation Engineering, Nanjing, 2012: 1-4.
[4] Marti?n-Herrero J. Anisotropic diffusion in the hypercube[J]., 2007, 45(5): 1386-1398.
[5] Yuan Qiang-qiang, Zhang Liang-pei, and Shen Huan-feng. Hyperspectral image denoising employing a spectral-spatial adaptive total variation model[J]., 2012, 50(10): 3660-3677.
[6] Chen Shao-lin, Hu Xi-yuan, and Peng Si-long. Hyperspectral imagery denoising using a spatial-spectral domain mixing prior[J]., 2012, 27(4): 851-861.
[7] Othman H and Qian Shen-en. Noise reduction of hyperspectral imagery using hybrid spatial-spectral derivative domain wavelet shrinkage[J]., 2006, 44(2): 397-408.
[8] 孫蕾, 羅建書. 高光譜遙感圖像微分域三維混合去噪方法[J]. 光譜學(xué)與光譜分析, 2009, 29(10): 2717-2720.
Sun Lei and Luo Jian-shu. Three-dimensional hybrid denoising algorithm in derivative domain for hyperspectral remote sensing imagery[J]., 2009, 29(10): 2717-2720.
[9] Chen Guang-yi and Qian Shen-en. Denoising of hyperspectral imagery using principal component analysis and wavelet shrinkage[J]., 2011, 49(3): 973-980.
[10] Jolliffe I T. Principal Component Analysis[M]. Second Edition, New York: Springer, 2002: 1-6.
[11] Aharon M, Elad M, and Bruckstein A. K-SVD: an algorithm for designing overcomplete dictionaries for sparse representation[J]., 2006, 54(11): 4311-4322.
[12] Elad M and Aharon M. Image denoising via sparse and redundant representations over learned dictionaries[J]., 2006, 15(12): 3736-3745.
[13] Donoho D L and Johnstone I M. Ideal spatial adaptation by wavelet shrinkage[J]., 1994, 81(3): 425-455.
[14] Chen Guang-yi and Zhu Wei-ping. Signal denoising using neighbouring dual-tree complex wavelet coefficients[C]. Proceedings of 22nd Canadian Conference on Electrical and Computer Engineering, St Johns, NL, 2009: 565-568.
[15] Sendur L and Selesnick I W. Bivariate shrinkage with local variance estimation[J]., 2002, 9(12): 438-441.
[16] Zhao Rui-zhen, Liu Xiao-yu, Li Ching-chung,Wavelet denoising via sparse representation[J].:, 2009, 52(8): 1371-1377.
[17] Mallat S. A Wavelet Tour of Signal Processing: The Sparse Way[M]. Third Edition, San Diego: Academic Press, 2008: 693-695.
[18] Hao Yan, Feng Xiang-chu, and Xu Jian-lou. Multiplicative noise removal via sparse and redundant representations over learned dictionaries and total variation[J]., 2012, 92(6): 1536-1549.
[19] 朱波, 汶德勝, 王飛, 等. 應(yīng)用字典學(xué)習(xí)算法改善Bayer格式圖像彩色恢復(fù)效果[J]. 電子與信息學(xué)報, 2013, 35(4): 812-819.
Zhu Bo, Wen De-sheng, Wang Fei,Improvement of Bayer-pattern demosaicking with dictionary learning algorithm[J].&, 2013, 35(4): 812-819.
[20] 宋相法, 焦李成. 基于稀疏表示及光譜信息的高光譜遙感圖像分類[J]. 電子與信息學(xué)報, 2013, 34(2): 268-272.
Song Xiang-fa and Jiao Li-cheng. Classification of hyperspectral remote sensing image based on sparse representation and spectral information[J].&2013, 34(2): 268-272.
[21] Pati Y C, Rezaiifar R, and Krishnaprasad P S. Orthogonal matching pursuit: recursive function approximation with applications to wavelet decomposition[C]. Proceedings of 27th Asilomar Conference on Signals, Systems and Computers, Pacific Grove, 1993: 40-44.
[22] Hyperspectral Remote Sensing Scenes[OL].http://www.ehu. es/ccwintco/index.php/Hyperspectral_Remote_Sensing_Scenes, 2013. 10.
[23] Landgrebe D and Biehl L. MultiSpec[OL]. https:// engineering.purdue.edu/~biehl/MultiSpec/, 2013.10.
[24] Clark R N, Swayze G A, Wise R,USGS digital spectral library splib06a, 2007[OL]. http://speclab.cr.usgs.gov/ spectral.lib06/, 2013. 10.
[25] Qian Yun-tao and Ye Min-chao. Hyperspectral imagery restoration using nonlocal spectral-spatial structured sparse representation with noise estimation[J]., 2013, 6(2): 499-515.
[26] Gabarba S and Cristobal G. Blind image quality assessment through anisotropy[J].&, 2007, 24(12): 42-51.
霍雷剛: 男,1986年生,博士生,研究方向為高光譜圖像去噪模型和算法.
馮象初: 男,1962年生,教授,研究方向為小波理論及應(yīng)用、數(shù)值分析、圖像處理等.
Denoising of Hyperspectral Remote Sensing Image Based on Principal Component Analysis and Dictionary Learning
Huo Lei-gang Feng Xiang-chu
(,,’710126,)
To reflect different intensities of noises among the different bands in the transform domain and the intrinsic structures of the transformed data, a new approach for denoising the hyperspectral images is proposed based on Principal Component Analysis (PCA) and dictionary learning. At first, a group of the principle component images are achieved by using the PCA transform. Then, these noises which exist in the spatial- and the spectral- domain of the components with low energy are denoised by an adaptively learned dictionary based sparse representation method and the dual-tree complex wavelet transform, respectively. Finally, the denoised data is obtained using the inverse PCA transform. By taking advantages of principal component analysis and dictionary learning, the proposed approach is superior to the traditional ones in preserving the details and alleviating the blocking artifacts. The experiment results on the synthetic and real hyperspectral remote sensing images demonstrate the effectiveness of the proposed approach.
Image processing; Hyperspectral image; Denoising; Principal Component Analysis (PCA); Sparse representation; Dictionary learning
TP751
A
1009-5896(2014)11-2723-07
10.3724/SP.J.1146.2013.01840
霍雷剛 leiganghuo@163.com
2013-11-22收到,2014-03-13改回
國家自然科學(xué)基金(61271294, 60872138)資助課題