An End-to-End Neural Network for Road Extraction From Remote Sensing Imagery by Multiple Feature Pyramid Network

被引:81
作者
Gao, Xun [1 ,2 ]
Sun, Xian [1 ]
Zhang, Yi [1 ]
Yan, Menglong [1 ]
Xu, Guangluan [1 ]
Sun, Hao [1 ]
Jiao, Jiao [1 ,2 ]
Fu, Kun [1 ,2 ,3 ]
机构
[1] Chinese Acad Sci, Inst Elect, Key Lab Technol Geospatial Informat Proc & Applic, Beijing 100190, Peoples R China
[2] Univ Chinese Acad Sci, Sch Elect Elect & Commun Engn, Beijing 100049, Peoples R China
[3] Chinese Acad Sci, Inst Elect, Suzhou 215123, Peoples R China
来源
IEEE ACCESS | 2018年 / 6卷
基金
中国国家自然科学基金;
关键词
Multiple feature pyramid network (MFPN); feature pyramid; pyramid pooling; weighted balance loss;
D O I
10.1109/ACCESS.2018.2856088
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Unlike single geospatial objects extraction from high-resolution remote sensing images, the task of road extraction faces more challenges, including its narrowness, sparsity, diversity, multiscale characteristics, and class imbalance. Focusing on these challenges, this paper proposes an end-to-end framework called the multiple feature pyramid network (MFPN). In MFPN, we design an effective feature pyramid and a tailored pyramid pooling module, taking advantage of multilevel semantic features of high-resolution remote sensing images. In the optimization stage, a weighted balance loss function is presented to solve the class imbalance problem caused by the sparseness of roads. The proposed novel loss function is more sensitive to the misclassified and the sparse real labeled pixels and helps to focus on the spare set of hard pixels in the training stage. Compared with the cross-entropy loss function, the weighted balance loss can reduce training time dramatically for the same precision. Experiments on two challenging datasets of high-resolution remote sensing images which illustrate the performance of the proposed algorithm have achieved significant improvements, especially for narrow rural roads.
引用
收藏
页码:39401 / 39414
页数:14
相关论文
共 47 条
[1]   SLIC Superpixels Compared to State-of-the-Art Superpixel Methods [J].
Achanta, Radhakrishna ;
Shaji, Appu ;
Smith, Kevin ;
Lucchi, Aurelien ;
Fua, Pascal ;
Suesstrunk, Sabine .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2012, 34 (11) :2274-2281
[2]   Simultaneous extraction of roads and buildings in remote sensing imagery with convolutional neural networks [J].
Alshehhi, Rasha ;
Marpu, Prashanth Reddy ;
Woon, Wei Lee ;
Dalla Mura, Mauro .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2017, 130 :139-149
[3]  
Anil P. N., 2010, Proceedings of the 2nd International Conference on Machine Learning and Computing (ICMLC 2010), P263, DOI 10.1109/ICMLC.2010.36
[4]  
[Anonymous], P IEEE INT C COMP VI
[5]  
[Anonymous], J ELECT INF TECHNOL
[6]  
[Anonymous], 2015, ARXIV PREPRINT ARXIV
[7]  
[Anonymous], 2014, Computer Science
[8]  
[Anonymous], P 31 AAAI C ART INT
[9]  
[Anonymous], 2016, PROC CVPR IEEE, DOI DOI 10.1109/CVPR.2016.348
[10]  
[Anonymous], 2016, P 2016 IEEE INT S, DOI DOI 10.1109/iNIS.2016.60