«上一篇
文章快速检索     高级检索
下一篇»
  应用科技  2020, Vol. 47 Issue (3): 46-50  DOI: 10.11991/yykj.201908011
0

引用本文  

房森, 焦淑红. 结合分层和ADMM的高光谱图像解混方法[J]. 应用科技, 2020, 47(3): 46-50. DOI: 10.11991/yykj.201908011.
FANG Sen, JIAO Shuhong. Hyperspectral image unmixing method combining hierarchy and alternating direction method of multipliers (ADMM)[J]. Applied Science and Technology, 2020, 47(3): 46-50. DOI: 10.11991/yykj.201908011.

通信作者

房森,E-mail:Sen_Fang2019@163.com

作者简介

房森,男,硕士研究生;
焦淑红,女,教授,博士生导师

文章历史

收稿日期:2019-08-14
网络出版日期:2020-05-15
结合分层和ADMM的高光谱图像解混方法
房森, 焦淑红    
哈尔滨工程大学 信息与通信工程学院,黑龙江 哈尔滨 150001
摘要:高光谱图像有较高的光谱分辨率,但是单个像元覆盖的面积比较大,导致单个像元中出现多于一种地物的现象,即混合像元。混合像元的存在严重影响了高光谱数据的后续利用。高光谱图像解混技术的目的就是将混合像元中存在的地物种类(端元)以及各个地物种类所对应的比例(丰度)精确地表示出来。高光谱数据覆盖的范围比较大,不可避免存在端元变异的现象。为了应对端元变异现象,利用扩展的线性混合模型对高光谱数据进行建模。在基于分层解混技术的基础上,利用乘子交替方向法对其进行优化。实验结果表明,解混效果得到提升。
关键词混合像元    端元    丰度    线性模型    乘子交替方向法    解混    光谱变异    多端元    
Hyperspectral image unmixing method combining hierarchy and alternating direction method of multipliers (ADMM)
FANG Sen, JIAO Shuhong    
College of Information and Communication Engineering, Harbin Engineering University, Harbin 150001, China
Abstract: Hyperspectral image has a higher spectral resolution, but the area covered by a single pixel is relatively large, resulting in more than one material exist in a single pixel, called a mixed pixel. The presence of mixed pixels severely affects the subsequent use of hyperspectral data. The purpose of hyperspectral image unmixing technique is to determine the materials (endmembers) present in the mixed pixels and their corresponding proportions (abundance). Because the coverage of hyperspactral data is relatively large, the phenomenon of endmember variation exists inevitably. In order to take endmember variation into consideration, an extended linear mixed model is used to describe the hyperspectral data. Based on the hierarchical unmixing technique, the alternating direction method of multipliers is used to optimize the result of unmixing. The experimental results show that the unmixing effect has been greatly improved.
Keywords: mixed pixel    endmember    abundance    linear model    alternating direction method of multipliers    unmixing    spectral variation    multiple endmembers    

高光谱遥感成像技术是在多光谱遥感成像技术的基础上发展而来的,高光谱遥感的出现可以称得上是遥感技术的一场革命。高光谱成像传感器单位照射面积比较大,因此混合像元[1-2]现象在高光谱图像中普遍存在。混合像元的存在是影响遥感图像分类精度和目标探测效果的重要原因。光谱解混是处理混合像元的主要技术,目的就是了解混合像元内参与混合的成分以及它们各自所对应的比例,是一种更精确的分类技术。由于线性混合模型(linear mixing model, LMM)[3-5]具有物理意义明确且使用简便,在多数的光谱解混方法中均使用LMM对高光谱数据进行建模。该模型潜在假设一种地物种类可以用一条单一的光谱曲线完全表示,对光谱变异(spectral variation, SV)[6-7]现象考虑不周。为了应对光谱变异,使用一种扩展的线性混合模型(extended linear mixing model, ELMM)[8]对高光谱数据进行建模。

文献[9]提出了一种基于分层的高光谱解混技术,通过分层的思想,把复杂的问题分解成若干层计算,以此降低总的计算量。乘子交替方向法(alternating direction method of multipliers, ADMM)[10]是一种求解优化问题的计算框架,适用于求解分布式凸优化问题。ADMM通过分解协调过程,将大的全局问题分解为多个较小、较容易求解的局部子问题、并通过协调子问题的解而得到大的全局问题的解。本文提出了一种将分层解混算法和ADMM优化算法相结合的高光谱解混算法,该算法同时利用了分层算法和ADMM的特点,并利用模拟数据和真实数据对该算法进行实验。

1 扩展的线性混合模型

ELMM主要用于克服由光照条件以及地形变化引起的SV,而且同时保持了LMM的特点,并证明了该模型的有效性。传统的线性混合模型表达式为

${{{x}}_k} = \sum\limits_{p = 1}^P {{a_{pk}}{{{s}}_p}} + {{{e}}_k}$ (1)

式中: ${{{x}}_k}$ 表示第k个像元,是一个向量,维度和光谱图像的波段数相等;P表示高光谱图像中总的端元种类; ${{{s}}_p}$ 表示第p个端元,维度和像元的相同, ${a_{pk}}$ 表示第p个端元在第k个像元中所占的比例; ${{{e}}_k}$ 表示噪声,维度也和像元相同。将式(1)运用到整幅图像则可以将其写成矩阵形式:

${{X}} = {{SA}} + {{E}}$

考虑到实际场景中存在SV,可以将一种映射关系作用到端元向量上,那么式(1)可以重新写为

${{{x}}_k} = \sum\limits_{p = 1}^P {a_{pk}{f_{pk}}\left( {{{{s}}_p}} \right)} + {{{e}}_k}$ (2)

这种映射关系表示成 ${f_{pk}}$ ,作用到对应的端元上就表示端元产生了变异,传统的LMM在描述高光谱数据时具有的特殊优点,同时考虑SV和LMM时,令 ${f_{pk}} = {\psi _k}{{{s}}_p}$ ,则式(2)可以重新写为

$\begin{array}{l} {{{x}}_k} = \displaystyle\sum\limits_{p = 1}^P {{a_{pk}}{\psi _k}{{{s}}_p} + {{{e}}_k}} = {\psi _k}\displaystyle\sum\limits_{p = 1}^P {{a_{pk}}{{{s}}_p} + {{{e}}_k}} = \\ \;\;\;\;\;\;{\psi _k}{{S}}{a_k} + {{{e}}_k} = {{S}}{\psi _k}{a_k} + {{{e}}_k} \\ \end{array} $ (3)

式(3)只是通过一系列不同的非负缩放因子 ${\psi _k}$ 在像元层面上进行缩放以表示因光照条件或地形因素引起的SV。在实际情况下,通常需要考虑到每种物质所产生的变异。因此可以将式(3)写为

${{{x}}_k} = \sum\limits_{p = 1}^P {{a_{pk}}{\psi _{pk}}{{{s}}_p} + {{{e}}_k}} = {{S}}{{{\varphi}} _k}{\partial _k} + {{{e}}_k}$ (4)

式中: ${\psi _{pk}}$ 是取决于具体地物种类和像元的缩放因子; ${{{\varphi}} _k} \in {{\bf{R}}^{P \times P}}$ 是一个对角矩阵,对角线上的元素是P个端元的缩放因子,其中缩放因子也可以用 ${{\varPsi}} \in {{\bf{R}}^{P \times N}}$ 表示,该矩阵与丰度矩阵大小相同。对于整幅图像来说,式(4)可以写成:

${{X}} = {{S}}\left( {{{\varPsi}} \odot {{A}}} \right) + {{E}}$

式中 $ \odot $ 表示2个相同大小的矩阵之间逐项进行相乘。这就是ELMM的矩阵表示,当所有的缩放因子 ${\psi _{pk}}$ 等于1时,该模型将退化为传统的LMM。

2 分层解混算法和ADMM算法

Roberts D等[11-13]提出的多端元高光谱解混算法不是针对每个像元采用固定的端元集,而是为每个像元迭代生成与之最为匹配的端元集,该算法解混精度较高,但是需要穷尽列举光谱库中端元的所有组合,所需的计算量也非常大。基于分层的高光谱解混算法,是将复杂的问题化解成若干个层面进行计算,以此来降低计算的复杂度。ADMM是一种求解优化问题的计算框架,适用于求解分布式凸优化问题。

2.1 基于分层的高光谱解混算法

分层的解混思想[1]是在第一层的计算中可以根据得到的丰度系数,确定该像元中所包含的地物种类以及所对应的地物种类中最佳的类内端元,并选择出丰度系数最大的那类地物所对应的端元。在接下来的计算中以该端元为基准进行多端元组合。

假设高光谱图像对应的地物种类为3种,且每种地物所包含的类内端元分别为3种、2种、2种,记为 ${M_{1,1}}$ ${M_{1,2}}$ ${M_{1,3}}$ ${M_{2,1}}$ ${M_{2,2}}$ ${M_{3,1}}$ ${M_{3,2}}$ ,则对应的端元集表示为 $E = \left\{ {{M_{1,1}},{M_{1,2}},{M_{1,3}},{M_{2,1}},{M_{2,2}},{M_{3,1}},{M_{3,2}}} \right\}$

1)在第一层利用端元集E对像元 $y_{{\rm{pixel}}}^k$ 进行解混,选出每个地物种类所对应的最大非零系数,假设确定的地物种类以及类内光谱为 $\left\{ {{M_{1,1}},{M_{2,1}},} \right.$ $\left. {{M_{3,1}}} \right\}$ ,然后分别计算 ${M_{1,1}}$ ${M_{2,1}}$ ${M_{3,1}}$ $y_{{\rm{pixel}}}^k$ 之间的光谱角距离,假设光谱角距离最小的端元为 ${M_{1,1}}$

2)在第一层确定的端元集合的基础上分别与 ${M_{1,1}}$ 进行两端元组合,分别记为 $\left\{ {{M_{1,1}},{M_{2,1}}} \right\}$ $\left\{ {{M_{1,1}},} \right.$ $\left. {{M_{3,1}}} \right\}$ ,然后分别计算与 ${{y}}_{{\rm{pixel}}}^k$ 之间的光谱角距离,选择出光谱角距离的最小值所对应的端元组合,在此处假设为 $\left\{ {{M_{1,1}},{M_{2,1}}} \right\}$

3)在 $\left\{ {{M_{1,1}},{M_{2,1}}} \right\}$ 的基础上与不包含以上2种地物的其他端元进行组合,构造三端元组合,记为 $\left\{ {{M_{1,1}},{M_{2,1}},{M_{3,1}}} \right\}$ ,然后计算其与 ${{y}}_{{\rm{pixel}}}^k$ 之间的光谱角距离;

4)对步骤1)~3)得到的光谱角距离进行排序,将光谱角距离最小所对应的端元组合作为该像元中实际的地物混合,以此端元组合对 ${{y}}_{{\rm{pixel}}}^k$ 进行解混计算得到最终的丰度。

2.2 ADMM寻优算法

ADMM[4]的核心思想是将一个复杂问题分解成一系列简单的子问题进行求解。ADMM经常用于求解如下形式的非约束问题:

$\mathop {\min }\limits_{x \in {{\bf{R}}^n}} {f_1}\left( x \right) + {f_2}\left( {Gx} \right)$ (5)

式中: ${f_1}$ ${{\bf{R}}^n} \to \overline {\bf{R}} $ ${f_2}$ ${{\bf{R}}^p} \to \overline {\bf{R}} $ ${{G}} \in {{\bf{R}}^{p \times n}}$

ADMM方法解决式(5)的步骤如下所示:

1) 令 $k = 0$ ,取 $\mu > 0$ ${u_0}$ ${d_0}$

2) 执行步骤3)~6),直到满足迭代终止条件;

3) ${x_{k + 1}} \in {\rm{arg }}\mathop {\min }\limits_x {f_1}\left( x \right) + \dfrac{\mu }{2}\left\| {Gx - {u_k} - {d_k}} \right\|_2^2$

4) ${u_{k + 1}} \in {\rm{arg }}\mathop {\min }\limits_u {f_2}\left( u \right) + \dfrac{\mu }{2}\left\| {G{x_{k + 1}} - u - {d_k}} \right\|_2^2$

5) ${d_{k + 1}} \leftarrow {d_k} - \left( {G{x_{k + 1}} - {u_{k + 1}}} \right)$

6) $k \leftarrow k + 1$

3 合成数据实验

为了验证本文所提算法的有效性,分别对合成数据和真实数据进行实验。在合成数据实验中,从USGS矿物光谱库中选择Actinolite、Antigorite和Chlorite的光谱,考虑到光谱变异,分别从这3种矿物所对应的端元光谱曲线中选取1条、2条、2条,分别记为Actinolite、Antigorite(1)、Antigorite(2)、Chlorite(1)、Chlorite(2),具体的光谱曲线如图1所示。

Download:
图 1 各种地物对应的端元曲线

利用这5条光谱曲线合成一幅100×100×224的高光谱图像,并加入40 dB的高斯噪声,合成的高光谱图像如图2所示。

Download:
图 2 合成高光谱图像

为了使实验结果具有更强的解释性,对该高光谱数据做特殊安排,前1~8列为Actinolite、9~16列为Antigorite(1)、17~24列为Antigorite(2)、25~32列为Chlorite(1)、33~40列为Chlorite(2)、40~45列为Actinolite和Antigorite(1)的随机组合、46~50列为Actinolite和Antigorite(2)的随机组合、51~55列为Actinolite和Chlorite(1)的随机组合、56~60列为Actinolite和Chlorite(2)的随机组合、61~65列为Antigorite(1)和Chlorite(1)的随机组合、66~70列为Antigorite(1)和Chlorite(2)的随机组合、71~75列为Antigorite(2)和Chlorite(1)的随机组合、76~80列为Antigorite(2)和Chlorite(2)的随机组合、81~85列为Actinolite、Antigorite(1)和Chlorite(1)的随机组合、86~90列为Actinolite、Antigorite(1)和Chlorite(2)的随机组合、91-95列为Actinolite、Antigorite(2)和Chlorite(1)的随机组合、96~100列为Actinolite、Antigorite(2)和Chlorite(2)的随机组合。将本次实验与部分约束最小二乘法(constrained least squares unmixing,CLSU)[14]、部分约束最小二乘法的的缩放版本(scaled version of CLSU,S-CLSU)、全约束最小二乘法(fully constrained least squares unmixng,FCLSU)[15]、以及基于分层的高光谱解混算法结果的比较,本文提出的方法表示为H-ADMM。具体的实验结果如图3所示,颜色深代表丰度值比较大,颜色浅代表丰度值较小。表1为各种算法的均方根误差。

Download:
图 3 5种算法解混丰度

实验结果从丰度图来看,本文所提算法H-ADMM在解混精度上最优,FCLSU次之,CLSU和SCLSU最差。从表1中的均方根误差来看,H-ADMM的结果最好,比分层解混算法要改善许多。由此可知本文提出算法的有效性。

表 1 均方根误差
4 真实数据实验

为了更好地验证本文算法的性能,真实数据实验部分采用的是美国印第安纳州实验农田数据。该数据共有16种地物,图像的空间尺寸大小为144×144,波段总数为100。由于真实数据地物种类繁杂,在本次实验中主要考虑图像中玉米、草和大豆并将其他地物全部作为背景。

考虑到端元变异,玉米、草和大豆中每种地物包含3条光谱曲线,分别记为玉米1、玉米2、玉米3、草1、草2、草3、大豆1、大豆2、大豆3,背景中包含7条光谱曲线,分别记为背景1、背景2、背景3、背景4、背景5、背景6和背景7。鉴于实验篇幅较大,本文只给出实验结果如图45所示。

Download:
图 4 分层解混丰度结果
Download:
图 5 H-ADMM解混丰度图

为了更好地评价这2种算法性能,还从数据层面对结果进行分析。采用整幅图像的均方根误差作为评价标准。所有像元的均方根误差直方图如图6所示,整幅图像的均方根误差如表2所示。

Download:
图 6 2种算法的均方根误差直方图
表 2 2种算法的均方根误差

图6可以看出,通过H-ADMM算法的解混结果所得到的所有像元的均方根误差分布较为集中,且误差范围分布较小。从2种算法的均方根误差来看,H-ADMM的均方根误差要远小于基于分层的高光谱解混算法,减小了约58.8%。

5 结论

无论从合成数据实验还是从真实数据实验中均可以看出,本文所提算法H-ADMM均优于基于分层的高光谱解混算法。

1)在合成数据实验中,H-ADMM在解混丰度图上明显好于其他几种经典的高光谱解混算法,解混后的丰度图边界清晰,背景噪声较小;

2)在真实数据实验部分,从解混丰度图上也可以看出H-ADMM算法对于分层解混算法的优势,从数据结果上可以明显看出H-ADMM要优于分层解混算法。

因此本文所提算法具有一定的优势。但在以后的研究中,应对H-ADMM算法进行多种的真实数据实验,探索H-ADMM算法中各个参数对解混结果的影响大小,使该算法适应于更多的真实场景。

参考文献
[1] 张兵, 孙旭. 高光谱图像混合像元分解[M]. 北京: 科学出版社, 2015. (0)
[2] 王立国, 赵春晖. 高光谱图像处理技术[M]. 北京: 国防工业出版社, 2013. (0)
[3] 李宏俏. 高光谱图像的光谱解混模型与算法研究[D]. 成都: 电子科技大学, 2017. (0)
[4] 邹丽. 高光谱图像混合像元解混技术研究[D]. 锦州: 辽宁工业大学, 2018. (0)
[5] 袁静, 章毓晋, 高方平. 线性高光谱解混模型综述[J]. 红外与毫米波学报, 2018, 37(5): 553-571. DOI:10.11972/j.issn.1001-9014.2018.05.008 (0)
[6] ZARE A, HO K C. Endmember variability in hyperspectral analysis: addressing spectral variability during spectral unmixing[J]. IEEE signal processing magazine, 2014, 31(1): 95-104. DOI:10.1109/MSP.2013.2279177 (0)
[7] SOMERS B, ASNER G P, TITS L, et al. Endmember variability in spectral mixture analysis: a review[J]. Remote sensing of environment, 2011, 115(7): 1603-1616. DOI:10.1016/j.rse.2011.03.003 (0)
[8] DRUMETZ L, VEGANZONES M A, HENROT S, et al. Blind hyperspectral unmixing using an extended linear mixing model to address spectral variability[J]. IEEE transactions on image processing, 2016, 25(8): 3890-3905. DOI:10.1109/TIP.2016.2579259 (0)
[9] 崔士玲. 多端元高光谱图像解混算法研究[D]. 哈尔滨: 哈尔滨工程大学, 2015. (0)
[10] BOYD S, PARIKH N, CHU E, et al. Distributed optimization and statistical learning via the alternating direction method of multipliers[J]. Foundations and trends in machine learning, 2010, 3(1): 1-122. DOI:10.1561/2200000016 (0)
[11] ROBERTS D A, GARDNER M, CHURCH R, et al. Mapping chaparral in the Santa Monica mountains using multiple endmember spectral mixture models[J]. Remote sensing of environment, 1998, 65(3): 267-279. DOI:10.1016/S0034-4257(98)00037-6 (0)
[12] 王可. 多端元光谱混合分析算法研究[D]. 成都: 电子科技大学, 2015. (0)
[13] 崔媛. 高光谱图像混合像元分解技术研究[D]. 沈阳: 沈阳航空航天大学, 2016. (0)
[14] BIOUCAS-DIAS J M, PLAZA A. an overview on hyperspectral unmixing: geometrical, statistical, and sparse regression based approaches[C]// Proceedings of 2011 IEEE International Geoscience and Remote Sensing Symposium. Vancouver, BC, Canada, 2011: 1135–1138. (0)
[15] AMMANOUIL R, FERRARI A, RICHARD C, et al. Blind and fully constrained unmixing of hyperspectral images[J]. IEEE transactions on image processing, 2014, 23(12): 5510-5518. DOI:10.1109/TIP.2014.2362056 (0)