A Self-Learning-Update CNN Model for Semantic Segmentation of Remote Sensing Images
Zheng Chen, Hu Chen, Yuncheng Chen, Jingying Li
Abstract
Convolutional neural network (CNN) has been widely used in semantic segmentation for remote sensing images, and it has achieved great success. Due to the diversity of the spatial distribution of terrestrial objects in remote sensing images, it is difficult to effectively learn general geographical laws and apply them to a specific image. To introduce geographical knowledge into the CNN model more effectively, a self-learning-update CNN model (SLU-CNN) is proposed in this letter. It learns the representation of specific spatial dependence among different objects according to the CNN result, and then incorporates it with the CNN result to make semantic inference available. The proposed method mainly involves two modules. First, geographical objects generated from the CNN result are used as inference units. Second, the spatial dependence between inference units is learned to build a specific adaptive geographical relationship. And then, it is embedded as an adaptive penalty term into an object-based Markov random field model to achieve the collaboration between the CNN result and the semantic inference. Our method provides a general data-knowledge dual-driven framework for the deep neural network. Experiments of the GID and Sentinel-2 datasets validate the effectiveness of the proposed method by comparing it with different state-of-the-art CNN methods.