基于约束非负矩阵分解的高光谱图像解混 下载: 994次
1 引言
高光谱遥感是将成像技术和光谱技术相结合的多维信息获取技术,可同时探测目标的二维集合空间与一维光谱信息,可获取高光谱分辨率的连续、窄波段图像数据[1-2]。由于高光谱图像的光谱分辨率较高,高光谱图像在农业生产、地物分类、矿产勘测、天然及人造材料识别等[3-4]方面都有应用。然而,由于技术限制,成像光谱仪的空间分辨率较低;另外,由于近距离的混合效应,混合像元在高光谱图像中普遍存在,混合像元的存在是影响高光谱图像分类和目标检测精度的重要原因[5-7]。因此,解决混合像元问题是高光谱遥感研究中的一个重要内容。通过分解混合像元,可以研究其中包含的端元和各个端元参与混合的程度,其中,确定组成混合像元的基本地物的过程称为端元提取,各个基本地物在混合像元中所占比例称为丰度[8],高光谱解混合算法能够有效克服因图像分辨率不足而造成的端元光谱估计不准确的问题。
光谱解混合算法依赖于场景混合模型的建立,现有的混合模型主要有线性模型和非线性模型[9]。其中,线性光谱混合模型已成为分析高光谱图像内容的最常用工具,当前的光谱解混合算法主要可分为6类:纯像元假设下的端元提取算法、最小体积模型下的端元提取算法、统计模型下的端元提取算法、融合空间信息的端元提取算法、智能端元提取算法以及其他端元提取算法等。
1) 纯端元假设下的端元提取方法是目前研究得最多、最成熟的算法,包括纯像元指数(PPI)[10-11]、内部最大体积分析(N-FINDR)[12-13]、迭代误差法(IEA)[14]、顺序最大角凸锥法(SMACC)[15]和顶点成分分析(VCA)[16]等。
2) 最小体积模型下的端元提取算法主要包括外包单形体收缩算法(SSWA)[17]、最小体积单形体分析(MVSA)[18]、最小体积外包单形体(MVES)[19]、最小体积约束的非负矩阵分解(MVC-NMF)[20],以及结合高光谱图像分类和单形体理论的光谱解混合[21]等。
3) 基于统计模型的端元提取也有很多算法,如独立成分分析(ICA)[22]、依赖成分分析(DECA)[23]、贝叶斯分析(Bayesian approaches)[24]等。该类算法的原理主要是通过引入几何约束等,统计分析计算得到最小误差,从而得到光谱解混结果。
4) 融合空间信息的端元提取算法主要包括自动形态学端元提取法(AMEE)[25]、空间光谱端元提取法(SSEE)[26]、空间光谱预处理算法(SSPP)[27]等。与前面几类混合像元分解方法不同的是,融合空间信息法能够将高光谱图像数据的空间信息引入到求解过程目标函数的约束中,充分考虑相邻像素之间丰度变化的连续平滑特性及空间相关性,能够更合理地体现地物含量变化,避免不同像元间丰度值差异变化大的问题。
5) 智能端元提取算法主要包括蚁群优化端元提取法(ACO-EE)[28]、离散粒子群优化端元提取法(PSO)[29]、自适应布谷鸟端元提取算法(ACSEE)[30]等。
6) 其他的端元提取算法主要有凸集分离端元提取法(PCOMMEND)[31]和支持向量机端元提取法(SVM-BEE)[32]。
以上这些方法均存在一些潜在的缺陷,基于纯端元存在的假设不成立时,第一类方法难以提取出精确的端元光谱特征。在图像噪声非常高的情况下,由于基于最小体积模型的算法对噪声非常敏感,大多数没有纯像素假设的算法无法得到好的端元。随着噪声的降低,两种端元提取算法(含和不含纯像元)的性能变得非常相似和均衡。在最优条件下,没有纯像素假设的算法具有更好的性能。基于NMF的算法无需假设纯像元的存在,并且可以同时将混合像元分解为端元和丰度。因此,基于NMF的算法在高光谱解混合领域受到了越来越多的关注。然而,如果直接将NMF应用到高光谱解混合中,它将会因为存在多个候选解而导致ill-posed问题,并且其算法性能也会受到初始化的影响。为了在线性高光谱解混中找到更好的NMF解,研究者们构建了许多正则项来构造新的NMF目标函数,如Sparse NMF[33]、Minimum Volume Constraint NMF (MVC-NMF)[18]、L1/2 Sparsity Constrained NMF (CNMF)[34]和全变分空间正则化NMF[35]等。最近,丰度的低秩和稀疏先验信息也被应用到目标函数的构造中,它们能够找到更加准确的端元和丰度。
本文结合基于图论改进的SSPP算法和丰度的稀疏性约束,提出了一种新的NMF算法。一方面,本算法考虑了相邻像素之间的空间关系和光谱相似性,这样可以有效地减小算法的初始光谱搜索空间,然后,对丰度施加稀疏性约束,以探索空间域中的几何信息;另一方面,引入拉普拉斯矩阵的图正则项来保持图像的局部不变性。本文设计了一个新的矩阵分解目标函数,将图结构融入其中,兼顾了图像局部不变性和丰度的全局稀疏性。
2 基本原理
2.1 NMF的稀疏约束
NMF的基本思想可以简单描述为:对于任意给定的一个非负矩阵
式中:
然而,由于成本函数的非凸性,不可能计算得到全局最优解。为了解决这一问题,采用对丰度加以约束的策略。在光谱解混合中,稀疏性是合理的约束条件。在本文中,(2)式的数学符号所表示的意义为:
式中:第一项为保真项;
2.2 改进的空间光谱预处理技术
空间光谱预处理(SSPP)算法采用了空间和光谱两种信息对高光谱图像进行预处理[27],筛选出空间同质且光谱纯度较高的像元用于后续的端元提取过程。SSPP算法具有提高端元提取算法精度的潜力,同时不会增加端元提取过程的复杂度,有利于高计算复杂度的端元提取算法缩短运算时间。本文与原SSPP算法的区别在于,本文采用的是谱聚类而不是K-means聚类算法,因为谱聚类对数据分布的适应性更强,聚类效果良好,同时聚类的计算量小且易于实现。使用改进的SSPP算法对原高光谱数据进行预处理,然后使用VCA(Vertex Component Analysis)[16]算法和完全约束最小二乘法(FCLS)对经过预处理后保留下来的高光谱像元点进行初始化计算,得到较好的初始端元和丰度。
谱聚类是从图论中演化出来的算法,谱图理论研究[36]和流形学习理论[37]已经证明,几何结构可以通过散布的数据点的最近邻图来有效建模。用点的集合
对于整幅高光谱图像来说,利用每个点的度
利用高光谱图像中的任意两点之间的权重值
式中:
式中:
式中:
式中:1表示元素全为1的
式中:(.*)和(./)分别表示两个矩阵对应元素相乘和相除。最终得到本文的解混合算法(SSPP-CNMF)流程如
3 实验数据与评价
3.1 实验数据
为了验证本算法的性能,使用一组模拟实验数据和两组真实实验数据进行实验,如
图 2. 高光谱图像。(a) Fractal1;(b) Jasper;(c) Cuprite
Fig. 2. Hyperspectral images. (a) Fractal1; (b) Jasper; (c) Cuprite
1) Fractal1是一幅模拟的高光谱数据[38],它包含221个波段,具有100×100个像素,合成这幅模拟数据时,在USGS光谱库中选择了9种矿物光谱,即Halloysite、Nontronite、Kaolinite CM9、Sphene、Muscovite、Kaolinite KGa1、Dumortierite、Pyrophyllite和Alunite。
2) Jasper数据集[39]是一种常用的高光谱数据集,每个波段具有512×614个像素,它具有224个光谱波段,光谱分辨率高达9.46 nm,光谱范围为380~2500 nm。由于图像的高复杂性,从原图像中的第(105, 269)像素开始裁剪出由100×100个像素组成的区域。此外,为了避免大气的影响,已移除了编号为1~3、108~112、154~166和220~224的通道(保留198个通道)。这个场景中包含了4种端元:Road、Soil、Water和Tree。
3) Cuprite数据集[39]是光谱解混合领域最具挑战性的数据集之一,它覆盖了美国内华达州的赤铜矿采矿区。该图像包含224个波段,光谱范围从370~2480 nm。与其他数据集类似,本文使用了包含250×190个像素的区域,此外还删除了噪声通道(1~2通道和221~224通道)和水蒸气吸收通道(104~113通道和148~167通道),最后保留了188个通道。Cuprite数据集中共有12种矿物: Alunite、Andradite、Buddingtonite、Dumortierite、Kaolinite1、Kaolinite2、Muscovite、Montmorillonite、Nontronite、Pyrope、Sphene和Chalcedony.
3.2 评价指标
为了验证所提SSPP-CNMF解混合算法的有效性,需要比较估计的实验结果和ground truth(GT)之间的差异。本文使用以下典型评估指标。
1) 光谱角距离(SAD)
光谱角距离(SAD)可以用来衡量两个光谱之间的差异程度,当两个光谱差异越大时,所对应的光谱角距离的值就会越大。对于两个光谱向量
2) 均方根误差(RMSE)
均方根误差(RMSE)通过计算估计的丰度(
式中:
4 分析与讨论
对比本文所提方法和各种文献中所提出的几种不同方法,例如VCA [16]、CoNMF[40]、MVSA[18]、MVC-NMF[20]等,采用两个指标(SAD和RMSE)来验证算法的有效性。所提SSPP-CNMF算法中有两个参数:
图 3. Jasper的ground truth(GT)和所提方法端元提取结果。(a)树木;(b)土壤;(c)水体;(d)道路
Fig. 3. Jasper ground truths (GT) and endmember results obtained by proposed algorithm. (a) Tree; (b) soil; (c) water; (d) road
图 4. Jasper丰度图的ground truth。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 4. Ground truths of Jasper abundance. (a) Water; (b) soil; (c) road; (d) tree
图 5. VCA算法估算的Jasper丰度图。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 5. Jasper abundances estimated by VCA algorithm. (a) Water; (b) soil; (c) road; (d) tree
图 6. CoNMF算法估算的Jasper丰度图。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 6. Jasper abundances estimated by CoNMF algorithm. (a) Water; (b) soil; (c) road; (d) tree
图 7. MVC-NMF算法估算的Jasper丰度图。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 7. Jasper abundances estimated by MVC-NMF algorithm. (a) Water; (b) soil; (c) road; (d) tree
图 8. SSPP-VCA算法估算的Jasper丰度图。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 8. Jasper abundances estimated by SSPP-VCA algorithm. (a) Water; (b) soil; (c) road; (d) tree
图 9. SSPP-CNMF算法估算的Jasper丰度图。(a)水体;(b)土壤;(c)道路;(d)树木
Fig. 9. Jasper abundances estimated by SSPP-CNMF algorithm. (a) Water; (b) soil; (c) road; (d) tree
图 10. SSPP-CNMF算法估算的Fractal1的丰度图。(a) Halloysite;(b) Nontronite;(c) Kaolinite CM9; (d) Sphene;(e) Muscovite;(f) Kaolinite KGa1;(g) Dumortierite;(h) Pyrophyllite;(i) Alunite
Fig. 10. Fractal1 abundances estimated by SSPP-CNMF algorithm. (a) Halloysite; (b) Nontronite; (c) Kaolinite CM9; (d) Sphene; (e) Muscovite; (f) Kaolinite KGa1; (g) Dumortierite; (h) Pyrophyllite; (i) Alunite
图 11. Fractal1的ground truth和SSPP-CNMF算法估算的端元波谱。(a) Dumortierite;(b) Halloysite;(c) Kaolinite CM9;(d) Kaolinite KGa1;(e) Muscovite;(f) Nontronite;(g) Pyrophyllite;(h) Sphene
Fig. 11. Fractal1 ground truth and endmember spectra estimated by SSPP-CNMF algorithm. (a) Dumortierite; (b) Halloysite; (c) Kaolinite CM9; (d) Kaolinite KGa1; (e) Muscovite; (f) Nontronite; (g) Pyrophyllite; (h) Sphene
图 12. SSPP-CNMF算法估算的Cuprite的丰度图。(a)端元1;(b)端元2;(c)端元3;(d)端元4;(e)端元5;(f)端元6;(g)端元7;(h)端元8;(i)端元9;(j)端元10;(k)端元11;(l)端元12
Fig. 12. Cuprite abundances estimated by SSPP-CNMF algorithm. (a) Endmember 1;(b) Endmember 2; (c) Endmember 3; (d) Endmember 4; (e) Endmember 5; (f) Endmember 6; (g) Endmember 7; (h) Endmember 8; (i) Endmember 9; (j) Endmember 10; (k) Endmember 11; (l) Endmember 12
SAD表示各种算法的端元提取结果和ground truth端元光谱曲线之间的差异程度,SAD越小,则差异程度越小,光谱曲线就越接近。RMSE表示的是各算法估计的丰度和ground truth丰度之间的差异,RMSE越小,则表示丰度和ground truth越接近。从
表 1. 不同高光谱解混合算法之间的SAD比较
Table 1. Comparison of SAD of different hyperspectral unmixing algorithms
|
表 2. 不同高光谱解混合算法之间的RMSE比较
Table 2. Comparison of RMSE of different hyperspectral unmixing algorithms
|
5 结论
通过结合约束非负矩阵分解和改进的空间光谱预处理来进行高光谱图像混合像元分解,对所有像元的丰度采取稀疏性约束策略,并基于高光谱图像的空间信息和光谱信息结合图论来构造图结构约束项,然后用正则化的方法构造稀疏解混目标函数。模拟数据及真实高光谱数据都证明本文所提出的SSPP-CNMF方法能够取得较好的解混合结果。
[1] Goetz A F H, Rowan L C. Geologic remote sensing[J]. Science, 1981, 211(4484): 781-791.
[2] Goetz A F H, Vane G, Solomon J E, et al. . Imaging spectrometry for earth remote sensing[J]. Science, 1985, 228(4704): 1147-1153.
[3] Plaza A, Benediktsson J A, Boardman J W, et al. Recent advances in techniques for hyperspectral image processing[J]. Remote Sensing of Environment, 2009, 113(S1): S110-S122.
[4] 蔡亮红, 丁建丽. 基于高光谱多尺度分解的土壤含水量反演[J]. 激光与光电子学进展, 2018, 55(1): 013001.
[5] LillesandT, Kiefer RW, ChipmanJ. Remote sensing and image interpretation[M]. Hoboken: John Wiley & Sons, Inc., 2014.
[6] 于纯妍, 赵猛, 宋梅萍, 等. 基于目标约束与谱空迭代的高光谱图像分类方法[J]. 光学学报, 2018, 38(6): 0628003.
[7] 李非燕, 霍宏涛, 白杰, 等. 基于稀疏表示和自适应模型的高光谱目标检测[J]. 光学学报, 2018, 38(12): 1228004.
[9] PlazaA, MartínG, PlazaJ, et al. Recent developments in endmember extraction and spectral unmixing[M] ∥Prasad S, Bruce L, Chanussot J. Optical remote sensing. Augmented vision and reality. Berlin, Heidelberg: Springer, 2011, 3: 235- 267.
[10] Boardman J W. Mapping target signatures via partial unmixing of aviris data[C]∥Fifth Annual JPL Airborne Geoscience Workshop. [S.l.]: JPL Publication, 1995: 23- 26.
[11] Boardman J W. Automating spectral unmixing of AVIRIS data using convex geometry concepts[C]∥4th Annual JPL Airborne Geoscience Workshop. [S.l.]: JPL Publication, 1993: 11-14: 23- 26.
[12] Winter M E. N-FINDR:an algorithm for fast autonomous spectral end-member determination in hyperspectral data[J]. Proceedings of SPIE, 1999, 3753: 266-275.
[14] Neville RA, StaenzK, SzerediT, et al. Automatic endmember extraction from hyperspectral data for mineral exploration[C]∥Fourth International Airborne Remote Sensing Conference and Exhibition / 21st Canadian Symposium on Remote Sensing, June 21-24, 1999, Ottawa, Ontario, Canada. Canada: Earth Sciences Sector, 1999.
[15] Gruninger J H, Ratkowski A J, Hoke M L. The sequential maximum angle convex cone (SMACC) endmember model[J]. Proceedings of SPIE, 2004, 5425: 1-14.
[16] Nascimento J M P, Dias J M B. Vertex component analysis: a fast algorithm to unmix hyperspectral data[J]. IEEE Transactions on Geoscience and Remote Sensing, 2005, 43(4): 898-910.
[17] Fuhrmann D R. Simplex shrink-wrap algorithm[J]. Proceedings of SPIE, 1999, 3718: 501-511.
[18] Li J, Agathos A, Zaharie D, et al. Minimum volume simplex analysis: a fast algorithm for linear hyperspectral unmixing[J]. IEEE Transactions on Geoscience and Remote Sensing, 2015, 53(9): 5067-5082.
[19] Chan T H, Chi C Y, Huang Y M, et al. A convex analysis-based minimum-volume enclosing simplex algorithm for hyperspectral unmixing[J]. IEEE Transactions on Signal Processing, 2009, 57(11): 4418-4432.
[20] Miao L D, Qi H R. Endmember extraction from highly mixed data using minimum volume constrained nonnegative matrix factorization[J]. IEEE Transactions on Geoscience and Remote Sensing, 2007, 45(3): 765-777.
[21] 严阳, 华文深, 崔子浩, 等. 高光谱分类体积的端元提取[J]. 激光与光电子学进展, 2018, 55(9): 093004.
[22] Tu T M. Unsupervised signature extraction and separation in hyperspectral images: a noise-adjusted fast independent component analysis[J]. Optical Engineering, 2000, 39(4): 897-906.
[24] Dobigeon N, Moussaoui S, Tourneret J Y, et al. Bayesian separation of spectral sources under non-negativity and full additivity constraints[J]. Signal Processing, 2009, 89(12): 2657-2669.
[25] Plaza A, Martinez P, Pérez R, et al. Spatial/spectral endmember extraction by multidimensional morphological operations[J]. IEEE Transactions on Geoscience and Remote Sensing, 2002, 40(9): 2025-2041.
[26] Rogge D M, Rivard B, Zhang J, et al. Integration of spatial-spectral information for the improved extraction of endmembers[J]. Remote Sensing of Environment, 2007, 110(3): 287-303.
[28] Gao L R, Gao J W, Li J, et al. Multiple algorithm integration based on ant colony optimization for endmember extraction from hyperspectral imagery[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2015, 8(6): 2569-2582.
[29] Gao L R, Zhuang L N, Wu Y F, et al. A quantitative and comparative analysis of different preprocessing implementations of DPSO: a robust endmember extraction algorithm[J]. Soft Computing, 2016, 20(12): 4669-4683.
[30] Zhao H H, Jiang Y M, Wang T, et al. A method based on the adaptive cuckoo search algorithm for endmember extraction from hyperspectral remote sensing images[J]. Remote Sensing Letters, 2016, 7(3): 289-297.
[31] ZareA, GaderP. Piece-wise convex spatial-spectral unmixing of hyperspectral imagery using possibilistic and fuzzy clustering[C]∥2011 IEEE International Conference on Fuzzy Systems (FUZZ-IEEE 2011), June 27-30, 2011, Taipei, Taiwan. New York: IEEE, 2011: 741- 746.
[32] Filippi A M, Archibald R. Support vector machine-based endmember extraction[J]. IEEE Transactions on Geoscience and Remote Sensing, 2009, 47(3): 771-791.
[33] Feng X R, Li H C, Li J, et al. Hyperspectral unmixing using sparsity-constrained deep nonnegative matrix factorization with total variation[J]. IEEE Transactions on Geoscience and Remote Sensing, 2018, 56(10): 6245-6257.
[36] BelkinM, NiyogiP. Laplacian eigenmaps and spectral techniques for embedding and clustering[C]∥Proceedings of the 14th International Conference on Neural Information Processing Systems: Natural and Synthetic, December 3-8, 2001, Vancouver, British Columbia, Canada. Cambridge: MIT Press, 2002: 585- 591.
[37] Chung F RK, Graham FC. Spectral graph theory[M]. Providence, Rhode Island: American Mathematical Society, 1997.
[38] Plaza J. Hendrix E M T, García I, et al. On endmember identification in hyperspectral images without pure pixels: a comparison of algorithms[J]. Journal of Mathematical Imaging and Vision, 2012, 42(2/3): 163-175.
[40] Li J. Bioucas-Dias J M, Plaza A, et al. Robust collaborative nonnegative matrix factorization for hyperspectral unmixing[J]. IEEE Transactions on Geoscience and Remote Sensing, 2016, 54(10): 6076-6090.
方帅, 王金明, 曹风云. 基于约束非负矩阵分解的高光谱图像解混[J]. 激光与光电子学进展, 2019, 56(16): 161001. Shuai Fang, Jinming Wang, Fengyun Cao. Hyperspectral Image Unmixing Based on Constrained Nonnegative Matrix Factorization[J]. Laser & Optoelectronics Progress, 2019, 56(16): 161001.