Binary Neural Network for Multispectral Image Classification

doi:10.1109/LGRS.2022.3161360

Binary Neural Network for Multispectral Image Classification

2022-01-01

6

PRE

AI

DOI：10.1109/LGRS.2022.3161360

原文链接

原文求助

摘要

En 中文

Compared with traditional images, multispectral images (MSIs) contain more spectral bands and higher data dimensions. The existing MSI classification model has high computational complexity and consumes a lot of computing resources. In this letter, we propose a lightweight multispectral classification method named CABNN based on binary neural networks (BNNs) to effectively have a trade-off between model performance and computational cost. First, we modify and binarize the MobileNetV1 network and add almost computation-free shortcuts to enhance the expressive capability. Secondly, since the BNN is sensitive to the distribution of activation functions, we introduce RPReLU with learnable coefficients to automatically adjust activation distribution at almost no extra cost. Lastly, considering that MSIs have multiple channels, we utilize an efficient channel attention (ECA) module to assign different weights to each channel to concentrate on crucial features and suppress insignificant features. We conduct experiments on four public MSI datasets, including NaSC-TG2, EuroSAT, GID Fine land-cover classification, and UC Merced Land Use. Extensive experiments demonstrate that the proposed CABNN has higher efficiency and better comprehensive performance than the state-of-the-art methods across the board.

Keyword:

Neural networks

Computational modeling

Image coding

Quantization (signal)

Meters

Image resolution

Deep learning

Binary neural network (BNN)

efficient channel attention (ECA)

model compression

multispectral image (MSI)

Binary Neural Network for Multispectral Image Classification

摘要

期刊

机构