Efficient Convolutional Neural Architecture Search for LiDAR DSM Classification

Abstract

Light detection and ranging (LiDAR) data provides rich elevation information, so it plays an irreplaceable role in ground objects classification. Recently, convolutional neural networks (CNNs) have shown excellent performance in LiDAR Digital Surface Models (DSM) classification. However, the architecture of CNN model relies heavily on manual design, so it has great limitations. In addition, different sensors capture LiDAR datasets with different properties, so the model should be designed to suit for different datasets, which further increases the workload of architecture design. Therefore, this paper proposes a method of automatic design of LiDAR DSM classification model. First, attention mechanism is introduced into search space to improve the feature extraction capability of the model. Then a gradient-based search strategy is used to obtain the optimal architecture from this search space. Second, a learning rate adjustment strategy is proposed to reduce the time spent in the search stage and evaluation stage to improve the classification accuracy of the model. Finally, a regularization scheme is introduced to enhance the robustness of the model and avoid overfitting. Experimental results on three public LiDAR datasets (Bayview Park, Recology and Houston) obtained from different sensors show that the proposed neural architecture search method achieves the impressive classification performance compared to several state-of-the-art classification methods and improves the classification accuracy under the condition of limited training samples.