欢迎访问林业科学,今天是

林业科学 ›› 2020, Vol. 56 ›› Issue (11): 97-107.doi: 10.11707/j.1001-7488.20201110

• 论文与研究报告 • 上一篇    下一篇

面向机载高光谱数据的3D-CNN亚热带森林树种分类

赵霖,张晓丽*,吴艳双,张斌   

  1. 北京林业大学森林培育与保护教育部重点实验室 精准林业北京市重点实验室 北京 100083
  • 收稿日期:2019-02-11 出版日期:2020-11-25 发布日期:2020-12-30
  • 通讯作者: 张晓丽
  • 基金资助:
    国家重点研发计划项目(2017YFD06009)

Subtropical Forest Tree Species Classification Based on 3D-CNN for Airborne Hyperspectral Data

Lin Zhao,Xiaoli Zhang*,Yanshuang Wu,Bin Zhang   

  1. Key Laboratory for Silviculture and Conservation of Ministry of Education Precision Forestry Key Laboratory of Beijing, Beijing Forestry University Beijing 100083
  • Received:2019-02-11 Online:2020-11-25 Published:2020-12-30
  • Contact: Xiaoli Zhang

摘要:

目的: 探讨深度卷积神经网络在机载高光谱数据分类中的应用,以提高亚热带地区森林树种分类精度。方法: 以广西南宁高峰林场为试验区,基于中国林业科学研究院LiCHy系统获取的机载高光谱数据,以三维卷积层为基础,提出一种高效的卷积神经网络(CNN)结构。CNN模型以端到端方式处理高光谱影像分析问题,将原始数据作为输入,不需要降维或特征筛选,可省去传统分类方法在不同程度上人工筛选特征的工作;网络中3D卷积层可同时提取光谱特征和空间特征,学习特征立方体空间和光谱维度的局部信号变化,利用重要的识别特征进行分类,以提高对高光谱影像的判别能力。针对机载高光谱数据维度高、训练样本相对较少的问题,对模型进行优化,以避免过拟合。结果: 相较传统的特征筛选与面向对象分割结合的方法,本研究提出的3D-CNN结构森林树种总体分类精度达98.38%,Kappa系数为0.98,与随机森林特征选择结合支持向量机分类相比,总体精度提高8.82%,Kappa系数提高0.11;小样本训练情况下(减少75%训练样本),总体精度仍可达95.89%,Kappa系数为0.94。结论: 三维卷积神经网络在处理机载高光谱影像特征提取和分类问题时能够充分利用影像中的丰富信息,实现高精度区分亚热带森林树种;合理的网络结构以及训练策略(加入Dropout)能够极大提高网络训练速度,并在小样本训练时仍能得到很好的结果,可实现高效、准确的森林树种分类。

关键词: 高光谱遥感, 卷积神经网络, 树种分类, 小样本

Abstract:

Objective: This study was implemented to explore the potential of deep convolutional neural networks in airborne hyperspectral data classification,so as to improve the classification accuracy of forest tree species in subtropical regions. Method: The aeronautical hyperspectral data of Nanning Gaofeng forest farm in Guangxi Zhuang Autonomous Region obtained by the LiCHy system of Chinese Academy of Forestry was used. The CNN model used in this paper aimed to deal with hyperspectral image analysis problems in an end-to-end manner. It could take raw data as input,without dimension reduction or feature screening,eliminating the need for traditional classification methods to manually feature selection in different degrees. The 3D convolutional layers in the network could extract spectral features and spatial features simultaneously,learn the local signal changes in the spatial and spectral dimensions of the feature cube,and classify them with important recognition features to improve the discriminating ability of hyperspectral images. For the problem of high dimensionality of airborne hyperspectral data and relatively few training samples,the CNN model was optimized to avoid over-fitting. Result: Compared with the traditional feature selection and object-oriented segmentation method,CNN could obtain a higher classification accuracy,the overall accuracy reached 98.38%,Kappa coefficient was 0.98. Compared with support vector machine combined with random forest (RF) feature selection classification,the overall accuracy was improved by 8.82%,and the Kappa coefficient was increased by 0.11. In the case of small sample training (75% reduction in training samples size),the overall accuracy still reached 95.89%,and the Kappa coefficient was 0.94. Conclusion: The three-dimensional convolutional neural network could fully utilize the rich information in the image processing of the feature extraction and classification of airborne hyperspectral imagery,which could achieve high-precision discrimination of subtropical forest tree species; in addition,reasonable network structure and training strategy (adding the Dropout layer) could greatly improve the network training speed and still get good results in small sample training,and could achieve efficient and accurate classification of forest species.

Key words: hyperspectral remote sensing, convolutional neural network, tree species classification, small sample

中图分类号: