《测绘学报》
构建与学术的桥梁 拉近与权威的距离
遥感影像要素提取的可变结构卷积神经网络方法
王华斌1,2
, 韩旻1,2
, 王光辉2, 李玉1
1. 辽宁工程技术大学测绘与地理科学学院, 辽宁 阜新 123000;
2. 自然资源部国土卫星遥感应用中心, 北京 100048
收稿日期:2018-03-20;修回日期:2018-08-27
基金项目:国家重点研发计划(2016YFB0501403)
第一作者简介:王华斌(1980-), 男, 副研究员, 研究方向为卫星遥感应用。E-mail:whb@sasmac.cn
通信作者:韩旻 E-mail:mzlapqowjf321@qq.com
摘要:针对利用经典卷积神经网络提取遥感影像地物要素的方法中,模型容量受既定网络固有结构的限制而难以得到良好提取效果的问题,提出利用可变结构卷积神经网络的遥感影像要素提取方法。该方法将结构搜索与权重求解过程统一,在定义卷积神经网络架构的基础上将其中的关键结构作为变量,并以要素提取精度指标作为目标函数,利用遗传算法求解网络结构,最后以该网络为模型提取遥感影像中的目标要素。相关试验表明,可变结构卷积神经网络具备灵活的模型容量,对遥感影像中目标要素的提取效果良好。
关键词:地物要素提取 卷积神经网络 可变结构 遗传算法
Surface features extraction in remote sensing images based on architecture-variant CNN
WANG Huabin1,2
, HAN Min1,2
, WANG Guanghui2, LI Yu1
1. School of Geomatics, Liaoning Technical University, Fuxin 123000, China;
2. Land Satellite Remote Sensing Application Center, MNR, Beijing 100048, China
Foundation support: The National Key Research and Development Program of China (No. 2016YFB0501403)
First author: WANG Huabin(1980—), male, associate research fellow, majors in application of satellite remote sensing.E-mail:whb@sasmac.cn.
Corresponding author: HAN Min E-mail: mzlapqowjf321@qq.com.
Abstract: To exceed limited capacity of established convolutional neural network(CNN) with fixed architecture in traditional surface feature extraction in remote sensing images, we propose a new feature extraction method based on architecture-variant convolutional neural network (AVCNN). In AVCNN, key units are variables and the performance of the unknown model become object function. That means architecture search is added before traditional weights solving. Genetic algorithm is introduced to search proper architecture and classical algorithm is used to solve unknown weights in the candidate CNN. The CNN with final architecture is used to extract the surface feature in remote sensing images. The experiment result shows that AVCNN has flexible capacity and performances well in surface features extraction in remote sensing images.
Key words: surface feature extraction convolution neural network variant architecture genetic algorithm
近年来,以卷积神经网络模型为典型代表的深度学习技术带动了遥感影像要素提取研究的迅速发展[1-2]。相较于传统的人工定义指数和浅层机器学习方法[3-6],卷积神经网络模型通过对海量图像的深层特征建模实现了图像信息的有效提取[7],在非样本数据集上的泛化能力显著增强。因此,基于卷积神经网络的遥感影像要素提取成为研究热点。
基于卷积神经网络模型的遥感影像要素提取方法通常利用包含卷积操作的神经网络模型描述影像与要素的类属关系,进而利用随机梯度下降算法求解模型中的未知参数[8]。按照要素提取的粒度,利用卷积神经网络模型的要素提取研究主要分为3个层次[9]:图像层的场景分类、像素层的影像分割和目标层的实例分割。
文献[10]提出的LeNet,将卷积操作引入神经网络模型,也是现代卷积神经网络模型的原型。该模型利用2次重复的卷积、下采样和3次全连接对图像的深层特征进行建模,在文本数字图像识别领域得到了有效应用。但由于卷积操作的计算代价过于高昂且缺乏明确的理论基础,该算法在提出后并未引起学界的关注。文献[1]提出的AlexNet以绝对优势荣获2012年ImageNet竞赛的冠军,加之低价的图形处理单元(GPU)提供了充足的计算资源,引起研究人员对于卷积神经网络模型的广泛关注。文献[11]利用AlexNet提取多尺度遥感影像的深层卷积特征,并对其进行特征编码以获取全局表征,结合支持向量机(support vector machine,SVM)分类影像所属的场景类别,是卷积神经网络模型在遥感领域的探索。文献[12]提出了一种结合显著性检测与稀疏自编码器的显著要素稀疏特征非监督提取算法,结合SVM准确地区分出了遥感影像的场景类别。然而,以场景分类为目标的要素提取结果受影像边界的制约,无法顾及影像中非主要类别的要素。
文献[13]提出的FCN(fully convolutional networks)是一种支持逐像素分类的卷积神经网络结构,将分割粒度由图像层细化到像素层。在该网络结构中,原本用于分类的卷积神经网络中的全连接层被替换为卷积层和转置卷积层,并加入了多个跨越连接。由于网络中间层全部为(转置)卷积层,训练所需的数据量较大,训练时间长,分割边缘不够清晰。针对这一问题,文献[14]引入数据增强技术并提出了U-Net。该网络先通过多个卷积、池化层提取高度抽象的深层特征,再通过转置卷积和卷积操作还原背景信息,同时利用带有裁切的跨越连接细化分割边缘。数据增强技术显著减少了模型训练时所需的样本数量,然而裁切和卷积操作降低了输出标记图像的空间分辨率,模型训练缓慢的问题依然存在。文献[15]将端到端的影像分割网络抽象分解为编码器、解码器和像素分类层,并且引入了利用最大池化指数的上采样层和批标准化层[16]定义SegNet。该网络结构通过边缘补零的卷积操作统一了输出标记图像与输入图像的空间分辨率,通过批标准化层缩短了训练所需的时间。但是该网络利用Softmax层作为像素多类别分类器,其类别间的竞争影响了分类结果,平均像素分类精度仍然有待提高。
文献[17]提出的R-CNN率先将候选区生成过程、卷积神经网络与支持向量机(SVM)结合用于目标检测问题。在该研究中,卷积神经网络模型主要负责候选区的特征提取,与其他过程耦合较为松散。其续作Fast R-CNN通过多任务损失(multi-task loss)建立了分类与包围盒修正网络[18]。在此基础上,Faster R-CNN将候选区生成过程替换为候选区生成网络[19]。至此,卷积神经网络模型完整地覆盖了目标检测任务的各个阶段。受Faster R-CNN启发,文献[20]提出的Mask R-CNN在原有架构的基础上增添了目标掩膜预测分支,从而将目标的检测、分类、分割纳入到统一的卷积神经网络模型中。该网络由多个部分构成,第1部分是残差连接网络(ResNet50或101)与特征金字塔网络(feature pyramid network)组成的特征提取网络,第2部分是候选区生成网络(region proposal network),第3部分是感兴趣区分类、位置修正及分割网络。特征网络首先提取图像多个尺度的特征,并由候选区生成网络直接在卷积特征层直接生成感兴趣区域,最后由分类及分割网络对感兴趣区域中的目标进行分类和分割。相关试验表明该模型能够十分准确地标注并分割出图像中的地物。然而,目标层的要素提取适用于独立特殊地物[21](如房屋、机场等)的提取,而对成片的大面积要素(如耕地)的提取支持较弱。
在卷积神经网络模型中,网络的结构定义是关键。上述几种方法虽然提取粒度不同,但普遍采用固定结构的卷积神经网络构造要素提取模型,网络的表达能力受到固定结构的制约。另一方面,固定结构网络的设计完全由专家依据专业知识手动定义,自动化水平低。针对以上问题,本文以像素层为要素提取粒度,将卷积神经网络的结构纳入要素提取问题的数学模型中,提出了基于可变结构卷积神经网络的遥感影像要素提取方法。本文方法将卷积神经网络中的关键结构作为变量,将形式化模型的要素提取指标作为目标函数。目标函数的最优解对应为关键结构最优的卷积神经网络模型。由于该目标函数对于网络中关键结构的导函数不可求取,采用遗传算法搜索最佳关键结构。对于遗传算法列举出的每个网络模型,基于传统交叉熵目标函数和梯度下降算法求解其中的未知参数,并以网络的要素提取性能作为相对应个体的适应性度量。最后,通过试验证明了本文方法的有效性。
1 方法描述1.1 模型变量
1.1.1 数据变量
约定影像张量X=[xn, h, w, c]N×H×W×C,其中xn, h, w, c为灰度值,其下标代表4个维度,n为影像索引,h为像素高度位置索引,w为像素宽度位置索引,c为像素光谱通道索引,Y=[yn, h, w, 0]N×H×W×1是像素类属目标要素的布尔标记。则对于某一类别的要素(如道路),已知数据集可表示为
(1)
式中,UInt16={0, 1, …, 216-1};Boolean={0, 1}。
1.1.2 网络结构变量
考虑要素提取粒度为像素层,必须对卷积神经网络模型的架构进行约束。本文选定编码器解码器网络作为基本架构。典型的编码器、解码器网络包括U-Net和SegNet。SegNet可以描述为带有跨越连接的编码器和解码器操作序列,其中编、解码器中的不可变部分是网络骨架,可变部分为关键结构,符合本文方法的应用要求。同时,为了便于拓展,本文设计了类SegNet作为可变结构卷积神经网络的基础架构。一个简单的具备两个编码单元和解码单元的类SegNet如图 1所示。
图 1 具有两个编码单元和解码单元的类SegNetFig. 1 A SegNet with 2-units encoder and 2-units decoder
约定F代表类SegNet引入的全部操作序列,fij是其中的一个操作,下标i为操作的顺序索引,下标j为操作的种类索引,θij为fij引入的变量,ξij为fij引入的常量。可以将该网络模型表达为
(2)
式中,Z(I)=[zn, h, w, 1(I)]N×H×W×1为像素类属目标要素的概率张量,即满足zn, h, w, 1(I)∈[0, 1]。引入F′⊂F表示类SegNet中编码单元与解码单元中可变部分的操作序列。
1.2 目标函数构建
基于上述模型变量,形式化地定义要素分类指标平均Jaccard指数为目标函数
(3)
式中
(4)
式中,σ为概率阈值,一般按照经验设定为0.5。本文方法按照目标函数最大化准则求取最优F′,从而获得关键结构设计最优的卷积神经网络模型。
1.3 基于遗传算法的组合优化
由于F′代表类SegNet中可变部分全部操作的一组序列,在已知数据D的条件下最大化目标函数问题可以转化为对F′的组合优化问题。遗传算法(genetic algorithm, GA)[22]是一种模仿生物种群进化过程的随机优化算法。对数据规模较大的组合优化问题,GA在求解非线性的函数优化问题等方面有良好的适用性[23]。因此,本文基于GA框架求解对F′的组合优化问题。利用GA求解编码单元和解码单元组合优化问题的关键步骤在于可行解的编码、种群初始化设定、个体适应度的定义和遗传算子的设计。
1.3.1 可行解的编码
本文利用一个个体P代表一组可行解,由结构染色体CA和超参数染色体CP组成
(5)
结构染色体由一组结构基因AUu组成,超参数染色体由一组超参数基因PUu组成,下标u为基因位置索引, U为基因长度,取偶数
(6)
结构基因是一组二进制数,可通过映射字典对应编码单元或解码单元中的操作类型,超参数基因是设置操作的超参数,例如卷积核数量、卷积核大小和激活函数名称等。为了对可行解的编码过程进行更加清晰的阐述,举例如下。
表 1列出基因-结构单元的映射字典。根据表 1给出数据,当取U=6,类SegNet编码过程可得如图 2所示网络结构。
表 1 基因-结构单元映射字典Tab. 1 The map dictionary of genetics and units
AUPU
结构单元
12
12
00卷积核激活函数
卷积激活
01激活函数卷积核
激活卷积
10卷积核批标准化轴
卷积批标准化
11批标准化轴卷积核
批标准化卷积
图 2 类SegNet的个体编码过程Fig. 2 The code of individual and corresponding SegNet
在图 2右侧所示的类SegNet中,第1个编码单元的定义是先卷积层后激活层,其中卷积操作的卷积核数为32,卷积核大小为(3, 3),激活操作的激活函数为elu(文献[24]),对应到结构染色体中第1个结构基因为00,超参数染色体中第1个超参数基因为((32, (3, 3)), (elu)),其余单元可依次类推,最终形成左侧的个体。
1.3.2 种群初始化设定
设Gv表示第v世代所有个体的集合
(7)
式中,o为个体索引; O为个体总数; v为世代索引; Po(v)为第v世代编号为o的个体。引入卷积核数量集合filters_ud,卷积核数量与单元网络深度比例因子集合filters_scale,卷积窗口宽高集合kernel_size,激活函数字典集合activation_dic,批标准化动量集合momentum_ud。记random为随机从集合抽取一个元素的函数,对于o=1, …, O, u=1, …, U,初始化的种群个体表示为
(8)
式中,AUu(0, o)=(random({0, 1}), random({0, 1}));PUu(0, o)=Φ(AUu(0, o))。Φ是根据结构基因初始化对应超参数基因的函数,可依据基因-结构单元映射字典和AUu定义。利用表 1给出的基因-结构单元映射字典,Φ示例为
(9)
式中
(10)
1.3.3 个体适应度定义
考虑到GA中的个体数量较多,为降低估计个体适应度所需的计算量,本文采用数据抽样和不充分训练的策略计算个体适应度。
数据抽样策略假定已知数据集D是均匀的,对其抽样N′张影像和标注,形成抽样数据集D′⊂D,
(11)
约定操作序列F′对应的个体为P′,则将D′和F′(代入式(3)定义为个体P′的适应度。由此,为了计算适应度函数,需要计算F′中操作引入的未知参数θ′ij。本文借助传统卷积神经网络求解方式,定义经验损失函数为
(12)
在经验损失函数最小化准则下,利用自适应低阶矩的随机一阶梯度下降算法[25]。在固定数据遍历次数和附加提前停止的条件下,求解模型中未知参数的数值解。由此,可实现以较小的计算成本初步估计网络的表达能力。
1.3.4 遗传算子的设计
(1) 随机联赛选择算子。对当前世代G,随机选取K个个体作为参加联赛的个体集合GSt,以t表示联赛次数索引
(13)
将每次联赛中适应度最优的个体保留,组成抚育下一种群的候选GP
(14)
(2) 单点交叉算子。将选择出个体按奇偶顺序两两配对,分别作为父代和母代
(15)
式中,a=1, 3, …, T-1;b=2, 4, …, T。以一定概率αco交换父母代染色体中的部分基因,即
(16)
式中,P′a与P′b为交换后的个体;CA′a、CP′a、CA′b、CP′b为交换后的染色体;β为交换位置起点,δ为交换部分基因长度,β与δ均在可行域内取随机数。得到交换后的种群GE
(17)
(3) 基本位变异算子。染色体交叉后种群中的每个个体以一定概率αmt发生基本位的变异。假设Pmt为概率判定要发生基本位变异的个体
(18)
对其结构染色体CA(mt)中每个基因按一定概率αfp重新初始化。假设AUγ(mt)为概率判定要发生变异的结构基因
(19)
则重新初始化后结构基因
(20)
由于AUγ(mt)发生变化,超参数染色体CP(mt)相同位置的超参数基因PUγ(mt)也必须相应改变。将AUγ(mt′)代入式(9),将PUγ(mt)重新初始化为PUγ(mt′)。由此得到CA(mt′)和CP(mt′),合并后表示变异后个体Pmt′
(21)
1.3.5 方法流程
综合以上概念,本文方法分解为3个部分,第1部分为架构定义,第2部分为结构优化,第3部分为固定结构CNN的要素提取。流程如图 3所示,描述如下:
图 3 本文方法流程Fig. 3 The flowchart of the proposed method
步骤1:定义参数。包括网络层范围参数、基因结构字典设计、单元数量和遗传参数等。
步骤2:初始化种群。设定随机初始化种群个体的染色体。
步骤3:估计每个个体的适应度。将种群中每一个个体映射为类SegNet卷积神经网络,采用数据抽样和不充分训练的策略求解模型未知参数,计算目标函数值作为个体适应度。
步骤4:进化终止条件判定。判断最佳个体适应度是否达到预期值或达到限制的最大进化世代数。若满足停止条件则转向步骤9,否则进入步骤5。
步骤5:选择运算。将随机联赛选择算子作用于种群。
步骤6:交叉运算。将单点交叉算子作用于种群。
步骤7:变异运算。将基本位变异算子作用于种群,由此得到下一代种群。
步骤8:返回步骤3。
步骤9:报告最优个体。最优个体映射的类SegNet卷积神经网络结构即为求解得到的决策函数族,在取消数据抽样和不充分训练策略的条件下求解模型未知参数。
步骤10:利用求取的模型对遥感影像中的要素进行提取。本文这里采用方法在独立测试集上的指标进行效果评估。
2 试验和分析2.1 试验设定
为了验证本文方法的有效性,研究基于DSTL竞赛数据集中的遥感影像和10种要素标记,利用本地交叉验证[26]的策略评估本文方法的可靠性。
DSTL数据集是以World-View3卫星所拍摄的遥感影像为基础制成的要素标记数据集,提供了57个区域的遥感影像,其中25个区域包含居民地、沥青道路、林木与河流等10种要素标记。每个区域提供空间分辨率为31 cm的融合后RGB遥感影像、空间分辨率为31 cm的全色遥感影像、空间分辨率为1.24 m的多光谱8波段遥感影像及空间分辨率为7.5 m的短波8波段遥感影像。
数据准备过程如下:①将25张影像及其对应的地物要素类别图划分为17张模型训练集与8张独立测试集;②在模型训练集中,对每种地物要素生成与8波段多光谱影像长宽相同的目标要素标记二值图像;③将模型训练集中8波段多光谱影像与目标要素标记二值图像裁切成大小相同的图块,影像图块与其对应的二值标记图块称为一组;④对所有的组抽样32%作为抽样集,其余作为余样。表 2为各组数据集中的组数量。
表 2 数据准备分组Tab. 2 The group of data
集合模型训练集独立测试集
影像/张178
要素类别要素训练集/组
抽样集余样集总计
建筑物102215317
混杂附属物98208306
沥青道路98206304
非沥青道路96205301
林木114241355
耕地121257378
河流100212312
湖泊98206304
大型车辆100212312
小型车辆98207305
试验使用表 3所示的基因-结构单元映射表,不充分训练的具体设定是固定数据遍历次数为70,批张量大小设定为2。为了避免计算时间过长,在试验中设定最大进化数V为200代。卷积核数量候选集filters_ud为{3, …, 16},卷积核窗口大小kernel_size为{1, 2, 3},卷积核数量随深度变化系数filters_scale为{2, 3, 4, 5},激活函数候选集{'elu', 'selu', 'relu', 'tanh', 'softplus', 'softsign', 'sigmoid', 'hard_sigmoid'},批标准化动量集{0.8, 0.81, …, 0.99}, 失活率集{0.5, 0.51,…, 0.99},种群规模O为20,联赛规模K为3,染色体交叉概率αco为0.5,基因变异概率αmt为0.02。卷积神经网络模型中权重求解的收敛条件为损失小于0.001或超过2000次数据遍历。
表 3 基因-结构单元映射字典Tab. 3 The map dictionary of genetics and units
AUPU
结构单元
123
123
000卷积核动量系数激活函数
卷积批标准化激活
001卷积核激活函数动量系数卷积激活批标准化
010激活函数动量系数卷积核激活批标准化卷积
011激活函数卷积核动量系数激活卷积批标准化
100动量系数激活函数卷积核批标准化激活卷积
101动量系数卷积核激活函数批标准化卷积激活
110卷积核激活函数失活率卷积激活失活
111卷积核动量系数失活率卷积批标准化失活
试验的硬件环境为Intel(R) Core(TM) i7-7700 HQ 2.80 GHz,NVIDIA Quadro M1200和Intel(R) Xeon(R) E5-2630U3 2.40 GHz,4块NVIDIA Tesla M40,软件环境为Python 3.5.2,主要依赖库为Tensorflow1.4.0,Keras2.0.8。
2.2 试验结果
2.2.1 遗传过程
利用可变结构卷积神经网络模型对10种要素提取过程建模并进行试验,图 4反映了结构优化过程中种群适应度与进化世代数之间的关系。其中,种群适应度分布由小提琴宽度表示,种群适应度上下限由小提琴上下顶点表示。对进化过程中种群适应度绝对值的观察显示,长度为6的类SegNet架构对不同要素的提取表现不同,体现为耕地要素提取网络的绝对提取效果最高,非沥青公路要素提取网络次之,之后依次是河流、沥青公路、林木、湖泊、建筑物,小型车辆、大型车辆。
图 4 10种要素网络模型结构优化遗传过程小提琴图Fig. 4 The violin figures of evolution of CNN architectures for ten features
从种群进化过程前后个体适应度的相对变化来看,多数要素提取网络的表达能力在进化过程后都得到了一定程度的优化(表 4)。分析表 4数据可以看出,个体适应度优化的幅度与要素种类相关,例如提取河流、沥青公路、非沥青公路的指标优化幅度在0.15至0.35之间,而提取建筑物、混杂附属物、林木、耕地、湖泊的指标优化幅度均在0.10以内,大型车辆、小型车辆的提取指标在优化前后并未有所提升。由此得出结论:在长度为6的类SegNet架构约束下,网络结构优化过程对不同要素的适用性不同,对非沥青公路、沥青公路、河流的要素提取网络的结构优化较为有效,而对建筑物、混杂附属物、林木、耕地的网络优化效果有限。对于大型车辆和小型车辆而言,提取模型的绝对指标几乎为0,并且优化结构也不能获得任何提升,因此可以判断网络结构并非提取该类特殊要素的制约条件。
表 4 结构优化前后适应度变化Tab. 4 The fitness changes caused by architecture optimization
要素初代最优个体末代最优个体优化幅度
建筑物0.5750.6380.063
混杂附属物0.0570.1320.075
沥青公路0.4210.6530.232
非沥青公路0.5220.8800.358
林木0.5750.6510.077
耕地0.8250.9250.099
河流0.6580.8050.147
湖泊0.6010.6480.048
大型车辆0.0000.0020.002
小型车辆0.0010.0070.006
2.2.2 最终网络
取结构优化过程中末代网络的最优个体,并映射为所对应的最终网络,表 5为10种要素的可变结构卷积神经网络的最终结构。B、C、A分别表示批标准化层、卷积层、激活层,后面附加的是该层所需的超参数。(C-卷积核数量,卷积窗口大小,B-动量系数,A-激活函数)。
表 5 10种要素提取网络模型结构Tab. 5 The architecture table for CNNs extracting ten features
结构最终结构
建筑物混杂附属物沥青公路非沥青公路林木耕地河流湖泊大型车辆小型车辆
EU1B-0.83B-0.99C-32, 3B-0.94C-28, 2C-16, 3C-26, 3B-0.91B-0.85A-tanh
C-32, 3A-tanhA-hard sigmoidC-16, 3A-hard sigmoidA-eluB-0.83C-30, 2A-eluC-22, 2
A-softsignC-14, 3B-0.96A-tanhB-0.85B-0.82D-0.56A-eluC-22, 5B-0.91
EU2A-reluB-0.82B-0.89A-hard sigmoidB-0.85B-0.87A-seluA-tanhC-56, 4B-0.97
C-28, 2A-eluA-reluC-64, 3A-softplusA-reluC-44, 3C-56, 2A-seluC-40, 3
B-0.81C-40, 3C-44, 3B-0.92C-40, 3C-32, 5B-0.99B-0.87D-0.82A-sigmoid
EU3A-reluC-56, 3A-reluC-96, 5B-0.84A-reluB-0.97A-seluB-0.88C-24, 3
C-72, 3B-0.94B-0.84A-softplusA-softplusB-0.95A-tanhB-0.9A-eluA-tanh
B-0.83A-reluC-96, 5B-0.87C-128, 3C-112, 4C-112, 5C-40, 5C-112, 2D-0.79
DU1B-0.82C-24, 5A-hard sigmoidC-64, 5B-0.98C-32, 4A-sigmoidB-0.87B-0.93C-32, 5
A-tanhA-seluC-104, 2A-reluC-56, 5A-tanhC-112, 2A-reluA-softplusA-hard sigmoid
C-88, 3B-0.8B-0.89B-0.85A-reluB-0.86B-0.92C-120, 5C-40, 4B-0.96
DU2B-0.95B-0.87B-0.82B-0.91C-48, 2A-softsignB-0.98B-0.9C-40, 5C-36, 5
A-reluA-reluA-reluA-reluA-reluC-56, 5C-16, 2A-tanhA-reluA-sigmoid
C-52, 5C-56, 3C-36, 3C-40, 5B-0.91A-reluA-softplusC-24, 3B-0.9B-0.93
DU3C-32, 4C-24, 4C-20, 4C-10, 3C-20, 5C-14, 4C-32, 4B-0.85C-6, 3A-tanh
B-0.9A-seluB-0.96B-0.82B-0.8B-0.91A-sigmoidA-hard sigmoidA-reluB-0.83
A-reluD-0.73A-hard sigmoidA-sigmoidA-reluA-reluB-0.96C-12, 5D-0.74C-32, 4
适应度0.640.130.650.880.650.920.810.650.0020.006
图 5(a)为对多种要素提取网络模型中激活操作的函数统计频数。分析发现relu函数的出现频数显著高于其他函数,tanh次之,最少出现的是softsign。说明激活函数的选择上relu函数非线性激活的处理效率高于其他函数,通用性强于其他函数。
图 5 参数分析统计Fig. 5 The statistics of parameters analysis
建筑物、沥青公路、非沥青公路、林木、河流等要素的提取网络中,卷积核数量服从随深度递增的设定,图 5(b)的折线图呈现倒U形;而耕地、湖泊、混杂附属物、小型车辆、大型车辆要素提取网络中,卷积核数量并未服从随深度递增的设定。说明卷积核数量的设定受要素种类影响,并非完全服从随深度递增的规律。对卷积核窗口大小的统计显示宽度为3的卷积窗口频数较高,为22次;宽度为5的窗口出现了18次;宽度为4的出现了10次。说明宽度为3的卷积窗口较宽度为4的卷积窗口通用性更强。图 5(c)中批标准化操作中动量系数按要素种类不同呈现3种分布:混杂附属物和沥青公路的要素提取网络中,动量系数呈现W形,非沥青公路、耕地、河流、湖泊的要素提取网络中,动量系数呈现M形,其他要素提取网络中,动量系数在编码器阶段变化幅度不大,而在解码阶段剧烈变化,呈现倒扣的勺状。Dropout处理出现在混杂附属物、河流、大型车辆、小型车辆4种要素提取网络中。说明模型优化过程对多光谱特征较为复杂的要素会利用Dropout的模型平均作用来尝试提高其提取效果。
2.2.3 提取效果
图 6为独立测试集上一张影像按本文方法提取要素的效果。其中要素真值与预测值的真正(true positive)部分以绿色标记,假负(false negative)部分以红色标记,假正(false positive)部分以黄色标记,真负(true negative)部分以白色标记。
图 6 目标要素提取结果Fig. 6 The extraction results of target surface features
通过视觉观察,对于测试影像中的林木、非沥青公路、耕地、建筑物、沥青公路、湖泊要素,模型提取出了大部分目标区域,说明模型具备一定的分类泛化能力。由于河流要素未出现在示例图像中,模型的预测结果均为假正,说明对负样本的处理是模型实用化的关键。另一方面,模型对于混杂附属物、大型车辆、小型车辆要素的提取效果并不理想。究其原因是混杂附属物包含了多种建筑物附属结构,其像元光谱十分复杂导致模型难以区分。另外,多光谱影像的分辨率为1.24 m,这3类要素在影像上的图斑尺寸过小,客观上也增加了提取的难度。
将本文AVCNN方法与Logistic、SVM、SegNet模型进行对比,评价指标选取模型在独立测试集合(8张影像)上的Jaccard指数。必须说明的是,方法对要素负样本的正确提取将使得Jaccard指数无意义,在统计时被忽略,但对要素负样本的错误提取仍会被统计。这种统计方式会极大地降低无法对负样本做完全正确预测的模型的评分,本文将其称为Jaccard指数的负样本抑制统计效应。图 7为评价指标分布箱形图,表 6为各提取方法在测试集合上的平均指标。
图 7 4种方法在独立测试集Jaccard指数箱形图Fig. 7 The box plots of Jaccard indexes of four methods on test set
表 6 不同方法提取目标要素的Jaccard指数Tab. 6 The Jaccard index of different methods for target features
要素名称提取方法
LogisticSVM[27]SegNet[15]AVCNN
建筑物0.050.000.120.22
混杂附属物0.000.000.000.04
沥青公路0.030.010.000.19
非沥青公路0.020.000.010.49
林木0.200.000.050.52
耕地0.100.120.100.26
河流0.180.060.000.02
湖泊0.010.000.020.09
大型车辆0.000.000.000.00
小型车辆0.000.000.000.00
图 7反映出AVCNN对多数要素的提取效果优于其他3种方法。在独立测试集中,仅有1张影像包含河流要素,Logistic模型通过负样本上的正确预测规避了平均Jaccard指数在统计上的负样本抑制效应,而本文方法虽然在正样本上的表现与Logistic模型相当(图 7(d)中河流要素的孤立点),但在负样本上产生了少量的错误预测,导致在统计的平均Jaccard指数上显著低于Logistic模型(表 6)。
2.3 分析
图 8为遗传过程中多个网络模型对非沥青道路要素提取的对比。可以看出,第50、100、150代以及最终输出网络经充分训练在测试集的提取结果中,假负部分(红色部分)逐渐减少,真正部分(绿色部分)逐渐增多,提取效果逐渐提升,说明利用不充分训练估计模型表达力的策略是有效的。
图 8 遗传过程中类SegNet的性能改善Fig. 8 The performance improvement of AVCNN in evolution
将本文方法得出各要素最终提取网络结构在要素间交叉地执行训练和提取,得到图 9所示的适配性矩阵(纵坐标代表网络权重训练所基于的要素索引,横坐标代表最终提取网络结构所基于的要素索引,坐标值为模型在独立测试集上的平均Jaccard指数)。由图中可以看出,遗传算法得到的网络结构与要素间的适配性与要素类型有关:对非沥青公路、林木两种要素,遗传算法得到的网络结构对其要素本身的提取是最优的,其中非沥青公路要素的网络与其他网络差异较为明显;对于建筑物、混杂附属结构、沥青公路,耕地、河流、湖泊、小型车辆等7种要素,遗传算法得到的网络结构对其要素本身的提取效果并非最优,但也具备相当的提取能力,说明本文方法表现出相当大的随机性,但能够作为求解具有一定容量的网络结构的方法;对于大型车辆,所有要素提取网络都未获得有效的提取效果,说明本文方法的适用性受限于数据条件和要素特点。综上,遗传算法对一种要素求解得到的网络结构对其他要素也具有一定的适应性,当目标地物发生变化时,重新利用遗传算法得到的新的网络可以获得具备一定容量的模型结构,但仍受要素特点与数据条件的限制。
图 9 各要素与解网络间的适配性矩阵Fig. 9 The adaptive array of AVCNN of different features
以非沥青道路为例,调整输入模型的影像块大小并对网络结构进行相应重构,得到影像块大小对最终网络结构提取效果的影响(图 10)。从图中可以看出,输入影像块大小会影响网络模型的预测效果。具体表现为随着影像块大小的增加,模型在独立测试集要素提取指标呈现先增后减的趋势。
图 10 影像块大小对模型性能影响Fig. 10 The performance of models with different input size
遥感影像的要素提取中,一般认为像素的类别具有唯一性。然而,在DSTL数据集中,部分要素的矢量数据在空间上存在重叠,例如林木与耕地,林木与非沥青公路等。这一部分重叠要素从影像中像元的光谱特征上难以分辨,但在逻辑上存在合理性。基于Softmax分类器的SegNet虽然架构简单,训练方便,但其所分像素具备类别上的排他性。因此本文认为针对每种要素建立像元的二值分类模型更加符合人类的主观认知和数据集特性。
AVCNN的核心步骤是利用进化算法对卷积神经网络模型进行结构及参数的调优。进化算法的收敛效率低,卷积神经网络的训练时间长,都使得本次研究中AVCNN的时间成本异常高昂,在使用4块24 GB显存GPU的条件下,各步骤的平均时间消耗如图 11所示。事实上,现有关于网络结构自动化搜索的其他研究[28-31]都依赖于强大的硬件支撑。本文方法继承了进化算法的可并行特性、卷积神经网络模型的可分布式特性。因此可以结合分布式计算技术缩短方法的执行时间,提高方法的实用性。
图 11 本文方法各步骤的平均时间消耗/hFig. 11 The average time consuming of our method/h
3 结论
针对现有卷积神经网络模型多为人工固定定义,在提取遥感影像中的目标要素时表达能力受到固定结构制约且自动化水平低的问题,提出了基于可变结构卷积神经网络要素提取方法。本文面向遥感影像要素提取场景定义类SegNet架构,将其中的关键结构作为变量,将模型的要素提取精度作为目标函数,设计了适用于该优化问题的遗传算法以求解网络结构,最后依据求解后得到的网络结构提取遥感影像中的目标要素。得出以下结论:①以“定义架构而非定义结构,求解结构而非固定结构”为核心思想的可变结构卷积神经网络模型设计思想是可行的;②本文设计的GA能够作为求解具备一定容量的可变卷积神经网络中关键结构的一种实用方法;③相比传统固定结构卷积神经网络,可变结构卷积神经网络具备更加灵活的模型容量和更强的模型表达能力。在仅使用多光谱数据的条件下,基于可变结构卷积神经网络模型提取DSTL数据集中各要素的Jaccard指数均优于基于Logistic、SVM和SegNet模型的方法。更重要的是,可变结构卷积神经网络将结构定义提升到了架构定义,减少了网络设计过程对专家知识的依赖,是深度学习自动化方面的重要探索。然而,必须注意的是模型自优化算法普遍需要高密度的计算资源,需要与分布式计算集群配合才能满足一般的业务的时间需求。另外,本文方法仍然受限于卷积神经网络的基础架构和网络深度,要素分类的绝对精度仍然有限。如何将整体架构和网络深度也纳入网络结构优化选择的过程中是下一步研究的方向。
【引文格式】王华斌, 韩旻, 王光辉, 等. 遥感影像要素提取的可变结构卷积神经网络方法. 测绘学报,2019,48(5):583-596. DOI: 10.11947/j.AGCS.2019.20180122