计算机应用 ›› 2015, Vol. 35 ›› Issue (9): 2706-2709.DOI: 10.11772/j.issn.1001-9081.2015.09.2706

• 行业与领域应用 • 上一篇    下一篇

基于栈式自编码的水体提取方法

王知音1, 禹龙2, 田生伟1, 钱育蓉1, 丁建丽3, 杨柳1   

  1. 1. 新疆大学 软件学院, 乌鲁木齐 830008;
    2. 新疆大学 网络中心, 乌鲁木齐 830046;
    3. 新疆大学 资源与环境科学学院, 乌鲁木齐 830046
  • 收稿日期:2015-04-09 修回日期:2015-06-02 出版日期:2015-09-10 发布日期:2015-09-17
  • 通讯作者: 禹龙(1974-),女,新疆乌鲁木齐人,教授,硕士,主要研究方向:计算机智能、计算机网络,yul_xju@163.com
  • 作者简介:王知音(1988-),女,湖北襄阳人,硕士研究生,主要研究方向:大数据处理;田生伟(1973-),男,新疆乌鲁木齐人,教授,博士,主要研究方向:计算机智能、大数据处理;钱育蓉(1980-),女,山东武城人,副教授,博士,CCF会员,主要研究方向:遥感图像处理、人工智能;丁建丽(1974-),男,新疆乌鲁木齐人,教授,博士,主要研究方向:遥感技术;杨柳(1990-),女,新疆乌鲁木齐人,硕士研究生,主要研究方向:大数据处理。
  • 基金资助:
    国家自然科学基金资助项目(41261090,61363083);新疆研究生科研创新项目(XJGRI2014033)。

Water body extraction method based on stacked autoencoder

WANG Zhiyin1, YU Long2, TIAN Shengwei1, QIAN Yurong1, DING Jianli3, YANG Liu1   

  1. 1. School of Software, Xinjiang University, Urumqi Xinjiang 830008, China;
    2. Network Center, Xinjiang University, Urumqi Xinjiang 830046, China;
    3. College of Resource and Environment Sciences, Xinjiang University, Urumqi Xinjiang 830046, China
  • Received:2015-04-09 Revised:2015-06-02 Online:2015-09-10 Published:2015-09-17

摘要: 为了进一步提高利用遥感图像进行水体提取的准确率和自动化程度,提出一种基于栈式自编码(SAE)深度神经网络的水体提取方法。通过堆叠稀疏自编码器构建深度网络模型,使用逐层贪婪训练法依次训练每层网络,从像素层面无监督学习特征,避免传统神经网络等方法需进行人工特征分析与选取的问题;用学习到的特征结合相应的样本标签有监督训练softmax分类器;利用反向传播(BP)算法微调优化整个模型。采用塔里木河ETM+数据进行实验,基于SAE的水体提取方法准确率达到94.73%,比支持向量机(SVM)和反向传播(BP)神经网络方法分别高出3.28%和4.04%。实验结果表明,所提方法能有效提高水体提取的精度。

关键词: 遥感图像, 水体提取, 深度学习, 栈式自编码, softmax分类器

Abstract: To improve the accuracy and automation of extracting water body by using remote sensing image, a method was proposed for water body extraction based on Stacked AutoEncoder (SAE). A deep network model was built by stacking sparse autoencoders and each layer was trained in turn with the greedy layerwise approach. Features were learnt without supervision from the pixel level to avoid the problem that methods such as traditional neural network needed artificial feature analysis and selection. Softmax classifier was trained with supervision by using the learnt features and corresponding labels. Back Propagation (BP) algorithm was used to fine-tune and optimize the whole model. The accuracy of SAE-based method reaches 94.73% by using the Tarim River's ETM+ data to do the experiment, which is 3.28% and 4.04% higher than that of Support Vector Machine (SVM) and BP neural network separately. The experimental results show that the proposed method can effectively improve the accuracy of water body extraction.

Key words: remote sensing image, water body extraction, deep learning, stacked autoencoder, softmax classifier

中图分类号: