AUTOMATED AND LIGHTWEIGHT NETWORK DESIGN VIA RANDOM SEARCH FOR REMOTE SENSING IMAGE SCENE CLASSIFICATION

摘要

With the development of deep learning, remote sensing image scene classification technology has been greatly improved. However, current deep networks used for scene classification usually introduce ingenious extra modules to fit the characteristics of remote sensing images. It causes a high labor cost and brings more parameters, which makes the network more complicated and poses new intractable problems. In this paper, we rethink this popular “add module” pattern and propose a more lightweight model, called ProbDenseNet (PDN). PDN is obtained via a random search strategy in Neural Architecture Search (NAS) which is an automated network design manner. In our method, all topological connections are assigned importance degrees which subject to a uniform distribution. And we set a regulator to adjust the sparsity of the network. By this way, the design procedure is more automated and the network structure becomes more lightweight. Experimental results on AID benchmark demonstrate that the proposed PDN model can achieve competitive performance even with much fewer parameters. And we also find that excessive connections do not always improve the network’s performance while they can drag down the network’s behavior as well. Furthermore, we conduct experiments on Vaihingen dataset with classical Fully Convolutional Network (FCN) framework. Quantitative and qualitative results both indicate that the features learned by PDN can also transfer in semantic segmentation task.