Automatic Design of CNNs via Differentiable Neural Architecture Search for PolSAR Image Classification
Hongwei Dong, Bin Zou, Lamei Zhang, Siyu Zhang
Abstract
Convolutional neural networks (CNNs) have shown good performance in polarimetric synthetic aperture radar (PolSAR) image classification. Excellent hand-crafted CNN architectures incorporated the wisdom of human experts, which is an important reason for CNNs success. However, the design of architectures is a difficult problem, which needs a lot of professional knowledge as well as computational resources. Moreover, the manually designed architecture might be suboptimal, because it is only one of the thousands of unobserved but objective existed paths. Considering that the success of deep learning is largely due to its automation of the feature engineering process, how to design automatic architecture search methods to replace the hand-crafted ones is an interesting topic. In this article, the application of neural architecture search (NAS) in the PolSAR area is explored for the first time. Different from the utilization of existing methods, a PolSAR-tailored Differentiable Architecture Search (DARTS) method, called PDAS, is proposed in order to adapt NAS to the PolSAR classification. A PolSAR-tailored search space and an improved one-shot search strategy are equipped with the proposed method. By PDAS, the architecture (corresponds to the hyperparameter but not the topology) parameters can be optimized with high efficiency by a stochastic gradient descent (SGD) method. The optimized architecture parameters should be transformed into corresponding architecture and retrained to achieve classification. In addition, a complex-valued PDAS (CVPDAS) is developed to fit the data form of PolSAR images so as to improve the performance. Experiments on three benchmark data sets show that the architectures obtained by searching have better classification performance than hand-crafted ones.