Acta Optica Sinica, Volume. 42, Issue 24, 2428005(2022)

Scene Classification of Remote Sensing Images Based on Wavelet-Spatial High-Order Feature Aggregation Network

Kang Ni1,2、*, Mingliang Zhai3, and Peng Wang4
Author Affiliations
  • 1School of Computer Science, Nanjing University of Posts and Telecommunications, Nanjing 210023, Jiangsu , China
  • 2Jiangsu Key Laboratory of Big Data Security & Intelligent Processing, Nanjing 210023, Jiangsu , China
  • 3College of Automation & College of Artificial Intelligence, Nanjing University of Posts and Telecommunications, Nanjing 210023, Jiangsu , China
  • 4College of Electronic and Information Engineering, Nanjing University of Aeronautics and Astronautics, Nanjing 211106, Jiangsu , China
  • show less

    This paper proposes a wavelet-spatial high-order feature aggregation network (WHFA-Net) which can be divided into two branches: wavelet domain feature extraction and spatial domain feature extraction. Firstly, a Harr wavelet transform is embedded into convolutional neural networks (CNNs), and low-frequency components of the depth-wise convolutional features are retained as wavelet depth features. Secondly, depth feature learning is performed by the max pooling, and then the output is used as spatial depth features. In addition, the wavelet domain and spatial domain depth features are vectored, and their auto-correlation and cross-correlation high-order depth feature vectors are obtained. Feature regularization, feature aggregation, and feature normalization are then performed in sequence. Finally, a cross-entropy loss function is utilized for end-to-end network training. The experimental results on NWPU45 (NWPU-RESISC45 Dataset) and AID (Aerial Image Dataset) show that compared with that of the benchmark network (VGG-16), the accuracy of the proposed WHFA-Net in scene classification is improved by 5.13%-12.12%. Furthermore, compared with DCCNN, APDC-Net, GBNet, LCNN-BFF, MSCP, and Wavelet CNN, the accuracy of WHFA-Net in scene classification is higher. Additionally, the effectiveness and performance differences of each module and branch are verified through the ablation experiments. Therefore, WHFA-Net can effectively and stably extract the high-order aggregated features of different feature domains in remote sensing scene images and accurately perform scene classification.

    Tools

    Get Citation

    Copy Citation Text

    Kang Ni, Mingliang Zhai, Peng Wang. Scene Classification of Remote Sensing Images Based on Wavelet-Spatial High-Order Feature Aggregation Network[J]. Acta Optica Sinica, 2022, 42(24): 2428005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Remote Sensing and Sensors

    Received: Apr. 26, 2022

    Accepted: Jun. 16, 2022

    Published Online: Dec. 14, 2022

    The Author Email: Ni Kang (tznikang@163.com)

    DOI:10.3788/AOS202242.2428005

    Topics