«上一篇
文章快速检索     高级检索
下一篇»
  应用科技  2021, Vol. 48 Issue (3): 7-11  DOI: 10.11991/yykj.202011020
0

引用本文  

崔颖, 王铃秀, 李文山. 二次样本筛选的高光谱图像分类研究[J]. 应用科技, 2021, 48(3): 7-11. DOI: 10.11991/yykj.202011020.
CUI Ying, WANG Lingxiu, LI Wenshan. College of Information and Communication Engineering   Harbin Engineering University[J]. Applied Science and Technology, 2021, 48(3): 7-11. DOI: 10.11991/yykj.202011020.

基金项目

中央高校基金科研业务费专项资金项目(3072021CF0805);黑龙江省自然科学基金项目(LH2020F021);国家自然科学基金项目(62071084)

通信作者

崔颖,E-mail:cuiying@hrbeu.edu.cn

作者简介

崔颖,女,副教授

文章历史

收稿日期:2020-11-02
二次样本筛选的高光谱图像分类研究
崔颖1,2, 王铃秀1,2, 李文山1,2    
1. 哈尔滨工程大学 信息与通信工程学院,黑龙江 哈尔滨 150001;
2. 先进船舶通信与信息技术工业和信息化部重点实验室,黑龙江 哈尔滨 150001
摘要:主动学习能够在有标记样本较少的分类任务中得到较好的分类结果,其中熵值装袋算法最为常用,其利用熵值来衡量样本的不确定性,但熵值并不能完全地代表样本的不确定度。针对这一问题,本文提出二次样本筛选的分类算法,通过超像素分割进行边缘区域样本筛选,选择出不确定度较高的样本。利用熵值装袋算法对区域筛选样本进行二次筛选,选择信息量较高的样本。实验表明,该方法可以得到更理想的分类效果。
关键词高光谱图像    图像分类    超像素分割    主动学习    区域筛选    信息熵筛选    样本选择    熵值装袋    
College of Information and Communication Engineering   Harbin Engineering University
CUI Ying1,2, WANG Lingxiu1,2, LI Wenshan1,2    
1. School of Information and Communication Engineering Harbin Engineering University, Harbin 150001, China;
2. Key Laboratory of Advanced Marine Communication and Information Technology, Ministry of Industry and Information Technology, Harbin 150001, China
Abstract: Active learning can get better classification results in the classification task with fewer labeled samples, and entropy bagging algorithm is the most commonly used. It uses entropy to measure the uncertainty of samples, but entropy can not completely represent the uncertainty of samples. In order to solve this problem, this paper proposes a classification algorithm of secondary sample selection, which selects the samples with high uncertainty by super-pixel segmentation. The entropy bagging algorithm is used to screen the samples selected in the previous step for the second time, and the samples with higher information content are selected. The experiments show that this method can obtain better classification effect.
Keywords: hyperspectral images    image classification    super-pixel segmentation    active learning    regional screening    information entropy screening    sample selection    entropy bagging    

对于高光谱分类来说,获取标记样本是非常困难的,主动学习的提出缓解了这一问题。近年来,主动学习在声音事件检测[1]、视觉情感分析[2]、医学图像标注[3]和故障诊断[4]等领域得到广泛应用。1992年,基于投票机制的委员会方法(Query-by-Committee,QBC)一经提出就得到了人们极大关注[5],该算法根据成员的投票结果选择样本。之后,基于委员会的查询方法被不断改进。熵作为信息量的衡量标准走进了人们的视野。文献[6]将信息熵用于硬件木马检测。Li等 [7]提出了一种基于信息熵和博弈论的混合遗传算法,该算法克服传统遗传算法容易陷入局部最优的缺点,获得更好的性能。Liu等 [8]提出了新的指数熵函数作为图像纹理平滑的指标。李衡等[9]将信息熵引入桩基信号特征提取与识别算法中,得到了较好的识别效果。在主动学习方面人们利用熵来度量样本的不确定性。熵值装袋算法(entropy query-by-bagging algorithm, EQB) [10]的优势在于其结果与分类器的种类无关,可以使用任何种类的分类器。Copa等[11]指出熵值存在多值偏置问题,即选择样本时会倾向于选择同一种类的样本,这会使得分类器不能泛化,导致分类结果的不理想。针对这一问题,李宠等[12]提出了平均熵值装袋查询算法(aEQB),该算法利用待分类数据的种类数来惩罚熵值,使其可以得到较好的分类效果。Copa等[11]提出了归一化熵值装袋查询算法(nEQB),通过预测类别数的对数项对熵值进行归一化处理,改善了熵值的多值偏置问题。但是对于多分类问题,基于熵值的不确定度查询算法的效果可能并不理想。陈荣等[13]研究发现,样本的不确定性不能由熵值完全决定,会出现熵值小的样本比熵值大的样本的不确定度还要高的情况。

通过对上述方法的分析,本文提出二次样本筛选的高光谱图像分类算法。首先,采用超像素分割选择区域边缘的不确定性较高的样本。然后,利用信息熵在不确定度较高的样本中进行二次选择,挑选出更有价值的样本。

1 二次样本筛选的分类算法

Demir等 [14]通过实验表明,在选取样本的过程中,选取样本越多,熵值装袋算法的性能越好。这表明基于熵值装袋算法只能确定样本信息量的大致分割范围,而不能按照信息量大小对样本进行准确的排列。为了使得熵值装袋算法能够更好地选取不确定度高的样本,本文提出在未标记样本集中进行一轮样本筛选,选取出未标记样本集中不确定度较高的样本,再利用主动学习策略在不确定度较高的样本中进行二次样本筛选。本算法在分类模型的训练过程中仅利用了较少的标记样本和未标记样本中不确定度较高的部分样本。相较于传统的主动学习算法使用样本少。图1为样本二次筛选算法的结构框图。

Download:
图 1 样本二次筛选算法的框图
1.1 区域样本筛选

本文通过改进的SLIC算法[15]进行区域样本选取。

每个超像素内的像素视作同类地物,这样位于超像素边缘位置的像素也就是位于多类之间的像素。这样像素的特征并不鲜明,在分类时容易误判,具有较高的不确定度,利用价值很高。改进的SLIC算法在初始化网格时采用六边形网格而不是传统的正方形网格。这样做的好处是:可以更大程度地利用周围的空间信息,从而更好地减少边界像素的距离失真。同时该算法利用光谱信息和空间信息计算像素到聚类中心的距离,最大限度地利用图像信息。六角形网格如图2所示。

Download:
图 2 初始化超像素模型

图2中,ω为六边形的宽度,h为六边形的高度,相邻六边形的水平距离用horiz表示(公式中用whoriz表示),垂直距离为vert(公式中用hvert表示)。在搜索过程中,在2ω×2ω大小的窗中搜索与中心像素相似的像素,六边形中心的计算可以用矩阵乘法进行简单的表示

$ \left[ {\begin{array}{*{20}{c}} {{m^i}}\\ {{n^i}} \end{array}} \right] = \left[ {\begin{array}{*{20}{c}} {{h_{{\rm{vert }}}}}&0\\ 0&{w{_{{\rm{horiz }}}}} \end{array}} \right]\left[ {\begin{array}{*{20}{c}} {2{i_{{\rm{row }}}} - 1}\\ {{i_{{\rm{column }}}}} \end{array}} \right] $ (1)

式中:超像素的中心像素的坐标为(mi,ni);irow为超像素的行索引;icolumn为列索引。

为了充分利用图像中的空间信息,本算法将光谱距离和空间距离进行加权求和,用得到的距离作为像素到聚类中心的距离。本算法中,用欧氏距离来衡量每一像素到聚类中心的空间距离,计算公式如下

$ {d_x} = \arccos \left( {\frac{{x_j^{\rm{T}}{x^i}}}{{{{\left\| {{x_j}} \right\|}_2}{{\left\| {{x^i}} \right\|}_2}}}} \right) $ (2)
$ {d_{mn}} = \sqrt {{{\left( {{m^i} - {m_j}} \right)}^2} + {{\left( {{n^i} - {n_j}} \right)}^2}} $ (3)
$ {D_j} = \sqrt {d_x^2 + {{\left( {\frac{{{d_{mn}}}}{\omega }} \right)}^2}\omega _S^2} $ (4)

式中:光谱距离dx为光谱角距离(spectral angle distance,SAD);空间距离dmn为欧氏距离;Dj为每一像素xj与聚类中心Ci的距离;因为空间距离dmn的变化较为明显,所以用六边形的宽度ω对其进行归一化处理;ωS为衡量空间相似度和光谱相似度之间的权重,ωS越大,空间相似度就越重要,超像素越紧密,其大小、形状也更加规范,衡量相似度时,光谱相似度比空间相似度更为重要,因此在实验中将ωS设置为0.3。参数ω用来控制超像素的平均面积进而调整其数量P,只要可以得到高光谱图像的尺寸及超像素的尺寸ωP就可以通过将超像素的最大行索引和最大列索引相乘得到。

图3给出了改进的SLIC算法的分割结果,其中图3(a)图代表Pavia大学数据集的分割结果,图3(b)图代表肯尼迪太空中心(Kennedy Space Center,KSC)数据集的分割结果。

Download:
图 3 改进的SLIC算法的分割结果

图3可知,在平滑的中心区域超像素的形状呈现较为规则的六边形,在边界处超像素的形状规则性较差,由于ω设定为7,所以超像素的面积较小,个数较多,每个超像素的面积也更为平均,分割效果更好。

在区域划分之后,将整幅较大的高光谱图像分割为许多较小的超像素,并且为每1个超像素中的像素贴上伪标签(假设同属于1个超像素内的样本为一类,它们的伪标签相同)。本算法利用滑窗机制来实现第1轮样本筛选。在滑动窗口的过程中,若窗口内的像素分属几个类别(即类别数大于1),则将窗口内的全部样本选择出来,在本文中这样的样本定义为边界样本。如图4所示。

Download:
图 4 滑窗示意

图4中的数字表示每1个样本的伪标签,滑动窗口的大小为2×2。图中虚线框中的样本即为边界样本(图4中仅画出窗口在第1行和第2行滑动时的效果图)。在窗口滑动的过程中,若判定窗口内的样本为边界样本,则将整个窗口内的样本的伪标签全部置为0。在滑窗结束后,所有伪标签为0的样本都被视为边界样本。

1.2 信息熵二次筛选

信息是一个很抽象的东西,最初人们并不能用一个具体的指标去衡量信息的多少。直到1948年,信息熵的提出使得人们可以具象的描述信息量。之后信息熵这一概念被广泛地应用于各个领域。信息熵作为衡量样本不确定度的指标被引入主动学习,产生了熵值装袋算法。

本文通过熵值装袋算法来进行第2次的样本筛选。接下来介绍3种熵值装袋算法。

1) EQB算法

通常情况下,样本的不确定性与熵值成正比。因此EQB算法将熵作为衡量不确定度的指标。式(5)中xi表示未标记样本,H( xi)是xi的信息熵

$ H\left( {{x_i}} \right) = \sum\limits_{\omega = 1}^{{N_i}} - p\left( {y_i^* = \omega \mid {x_i}} \right)\log \left[ {p\left( {y_i^* = \omega \mid {x_i}} \right)} \right] $ (5)

式中: $y_i^*$ xi作为输入时分类器输出的类别; $p\left( {y_i^* = \omega \mid {x_i}} \right)$ 为样本xi属于地物类别ω的概率;NiM个分类器的输出的类别的总数( $1 \leqslant {N_i} $ $ \leqslant N$ N是输出类别总数)。

在多分类问题中,分类器将样本分为更少的类,表明样本的信息量较少。分类器将样本分为更多的类,表明样本的信息量更丰富。EQB由式(6)定义

$ {x^{{\rm{EQB}}}} = \arg \mathop {\max }\limits_{{x_i} \in U} H\left( {{x_i}} \right) $ (6)

式中U为未标记样本集。

2) nEQB算法

nEQB算法可以有效地改善多值偏置问题,该方法使用1个包含预测类别数的对数项来惩罚多值问题,nEQB由式(7)定义

$ {x^{{\rm{nEQB}}}} = \arg \mathop {\max }\limits_{{x_i} \in U} \left\{ {\frac{{H\left( {{x_i}} \right)}}{{\log \left( {{N_i}} \right)}}} \right\} $ (7)

3) aEQB算法

aEQB算法通过加入1个待分类数据集中的种类的数量来惩罚多值属性,aEQB由式(8)定义

$ {x^{{\rm{aEQB}}}} = \arg \mathop {\max }\limits_{{x_i} \in U} \left\{ {\frac{{H\left( {{x_i}} \right)}}{{{N_i}}}} \right\} $ (8)
2 实验结果与分析 2.1 高光谱实验数据集

本文使用PaviaU、PaviaC和Salinas数据集进行实验。PaviaU数据集共有9种地物,全部参与实验,其中训练集占23%,共有270个标记样本;在PaviaC数据集中选择除去第1类地物和第8类地物以外的样本进行实验,其中训练集占58%,在训练集中每类选择5个样本作为有标记样本,共计35个。在Salinas数据集中选择样本数量较多的8类地物参与实验,其中训练集占57%,训练集中共有240个有标记样本。每次实验会迭代100次,共重复10次,取平均值得出实验结果。

2.2 数据分析

表1图5是PaviaU数据集的实验结果。

表 1 PaviaU数据集实验结束时的各种精度对比
Download:
图 5 PaviaU数据集的仿真结果

在OA方面,改进算法与EQB算法相比提高最多为1.45%;与nEQB算法相比提升最少为0.04%;与aEQB算法相比提升0.57%。在所有的6种算法中样本筛选分类方法(aEQB)的OA最高为93.61%。从Kappa系数来看,EQB算法的Kappa系数最低为0.893 9,样本筛选分类方法(aEQB)的Kappa系数最高为0.915 2。

表2图6是PaviaC数据集上的仿真结果。对于OA来说,所有算法在实验结束时的OA都达到91%以上,改进算法与EQB算法相比提高1.81%;与aEQB算法相比提高0.09%;与nEQB算法相比没有提升。在AA方面,EQB算法的AA最低为91.32%,nEQB算法和改进算法的AA相同且最高为94.76%。

表 2 PaviaC数据集实验结束时的各种精度对比
Download:
图 6 PaviaC数据集的仿真结果

表3图7是Salinas数据集上的仿真结果。在OA方面,改进算法与EQB算法相比提高最多为0.16%;与nEQB算法相比提升最少为0.02%。在所有的6种算法中样本筛选分类方法(EQB)的OA最高为91.06%。从AA来看,aEQB算法的AA最低为94.10%,样本筛选分类方法(nEQB)的Kappa系数最高为0.893 6。

表 3 Salinas数据集实验结束时的各种精度对比
Download:
图 7 Salinas数据集的仿真结果
3 结论

本文对于熵值装袋算法中熵值不能完全决定样本的不确定度的问题,提出样本的二次筛选策略。第1次筛选,利用SLIC算法将高光谱图像进行区域分割,提取不确定度较高的样本;第2次筛选,利用熵值装袋算法对不确定度较高的无标签样本进一步筛选,选择信息量丰富的样本进行人工标注。在训练分类器的过程中,本算法仅利用了有标签样本和未标记样本中位于区域边缘的不确定度较高的部分样本。相对于传统的主动学习策略,本算法使用的样本数量相比之下少很多。实验表明,在PaviaU数据集上改进算法相比于基础算法在各种精度上均有提高。在PaviaC数据集上改进算法相比于EQB算法、aEQB算法在各种精度上均有提高,与nEQB算法相比并无提高。在未来的工作中,在一次样本筛选时,可以采用其他的图像分割算法进行实验,对于不确定度高的样本采取不同策略结合,讨论图像分割算法和不同熵值计算对分类结果的影响。

参考文献
[1] ZHAO Shuyang, HEITTOLA T, VIRTANEN T. Active learning for sound event detection[J]. IEEE/ACM transactions on audio, speech, and language processing, 2020, 28: 2895-2905. DOI:10.1109/TASLP.2020.3029652 (0)
[2] CHEN Jie, MAO Qirong, XUE Luoyang. Visual Sentiment Analysis with Active Learning[J]. IEEE Access, 2020, 8: 185899-185908. DOI:10.1109/ACCESS.2020.3024948 (0)
[3] ZHOU Jiancun, CAO Rui, KANG Jian, et al. An Efficient High-Quality Medical Lesion Image Data Labeling Method Based on Active Learning[J]. IEEE Access, 2020, 8: 144331-144342. DOI:10.1109/ACCESS.2020.3014355 (0)
[4] CHEN Meng, ZHU Kun, WANG Ran, et al. Active Learning-Based Fault Diagnosis in Self-Organizing Cellular Networks[J]. IEEE Communications Letters, 2020, 24(8): 1734-1737. DOI:10.1109/LCOMM.2020.2991449 (0)
[5] SEUNG H S, OPPER M, SOMPOLINSKY H. Query by Committee[C]//Proceedings of the 5th Workshop on Computational Learning Theory. Pittsburgh, USA, 1992, 287–294. (0)
[6] LU Renjie, SHEN Haihua, FENG Zhihua, et al. HTDet: A Clustering Method Using Information Entropy for Hardware Trojan Detection[J]. Tsinghua Science and Technology, 2021, 26(1): 48-61. DOI:10.26599/TST.2019.9010047 (0)
[7] LIU Chong, YANG Cui, WEI Mingqiang, et al. A Hybrid Genetic Algorithm Based on Information Entropy and Game Theory[J]. IEEE Access, 2020, 8: 836602-36611. (0)
[8] LIU C, YANG C, WEI M, et al. Texture Smoothing Quality Assessment via Information Entropy[J]. IEEE Access, 2020, 8: 88410-88421. DOI:10.1109/ACCESS.2020.2993146 (0)
[9] 李衡, 康维新. 基于EMD与模糊聚类的桩缺陷特征提取与识别[J]. 应用科技, 2019, 46(2): 88-93. (0)
[10] TUIA D, RATLE F, PACIFICI F, et al. Active Learning Methods for Remote Sensing Image Classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2009, 47(7): 2218-2232. DOI:10.1109/TGRS.2008.2010404 (0)
[11] COPA L, TUIA D, VOLPI M, et al. Unbiased query-by-bagging active learning for VHR image classification[C]//Proceedings of SPIE 7830, Image and Signal Processing for Remote Sensing XVI. Toulouse, France, 2010: 78300K. (0)
[12] 李宠, 谷琼, 蔡之华, 等. 改进的主动学习算法及在高光谱分类中的应用[J]. 华中科技大学学报(自然科学版), 2013, 41(S2): 274-278. (0)
[13] 陈荣, 曹永锋, 孙洪. 基于主动学习和半监督学习的多类图像分类[J]. 自动化学报, 2011, 37(8): 954-962. (0)
[14] DEMIR B, PERSELLO C, BRUZZONE L. Batch-Mode Active-Learning Methods for the Interactive Classification of Remote Sensing Images[J]. IEEE Transactions on Geoence & Remote Sensing, 2011, 49(3): 1014-1031. (0)
[15] WANG Xinyu, ZHONG Yanfei, ZHANG Liangpei, et al. Spatial Group Sparsity Regularized Nonnegative Matrix Factorization for Hyperspectral Unmixing[J]. IEEE Transactions on Geoscience and Remote Sensing, 2017, 55(11): 6287-6304. DOI:10.1109/TGRS.2017.2724944 (0)