A hybrid neural architecture search for hyperspectral image classification

Autor: Aili Wang, Yingluo Song, Haibin Wu, Chengyang Liu, Yuji Iwahori
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Frontiers in Physics, Vol 11 (2023)
Druh dokumentu: article
ISSN: 2296-424X
DOI: 10.3389/fphy.2023.1159266
Popis: Convolution neural network (CNN)is widely used in hyperspectral image (HSI) classification. However, the network architecture of CNNs is usually designed manually, which requires careful fine-tuning. Recently, many technologies for neural architecture search (NAS) have been proposed to automatically design networks, further improving the accuracy of HSI classification to a new level. This paper proposes a circular kernel convolution-β-decay regulation NAS-confident learning rate (CK-βNAS-CLR) framework to automatically design the neural network structure for HSI classification. First, this paper constructs a hybrid search space with 12 kinds of operation, which considers the difference between enhanced circular kernel convolution and square kernel convolution in feature acquisition, so as to improve the sensitivity of the network to hyperspectral information features. Then, the β-decay regulation scheme is introduced to enhance the robustness of differential architecture search (DARTS) and reduce the discretization differences in architecture search. Finally, we combined the confidence learning rate strategy to alleviate the problem of performance collapse. The experimental results on public HSI datasets (Indian Pines, Pavia University) show that the proposed NAS method achieves impressive classification performance and effectively improves classification accuracy.
Databáze: Directory of Open Access Journals