Year: 2024
Type: article
Abstract: In forest ecosystems, the application of hyperspectral (HS) imagery offers unprecedented opportunities for refined identification and classification. The diversity and complexity of forest cover make it challenging for traditional remote-sensing techniques to capture subtle spectral differences. Hyperspectral imagery, however, can reveal the nuanced changes in different tree species, vegetation health status, and soil composition through its nearly continuous spectral information. This detailed spectral information is crucial for the monitoring, management, and conservation of forest resources. While Convolutional Neural Networks (CNNs) have demonstrated excellent local context modeling capabilities in HS image classification, their inherent network architecture limits the exploration and representation of spectral feature sequence properties. To address this issue, we have rethought HS image classification from a sequential perspective and proposed a hybrid model, the Spectral Weaver, which combines CNNs and Transformers. The Spectral Weaver replaces the traditional Multi-Head Attention mechanism with a Channel Attention mechanism (MCA) and introduces Centre-Differential Convolutional Layers (Conv2d-cd) to enhance spatial feature extraction capabilities. Additionally, we designed a cross-layer skip connection that adaptively learns to fuse “soft” residuals, transferring memory-like components from shallow to deep layers. Notably, the proposed model is a highly flexible backbone network, adaptable to both hyperspectral and multispectral image inputs. In comparison to traditional Visual Transformers (ViT), the Spectral Weaver innovates in several ways: (1) It introduces the MCA mechanism to enhance the mining of spectral feature sequence properties; (2) It employs Centre-Differential Convolutional Layers to strengthen spatial feature extraction; (3) It designs cross-layer skip connections to reduce information loss; (4) It supports both multispectral and hyperspectral inputs, increasing the model’s flexibility and applicability. By integrating global and local features, our model significantly improves the performance of HS image classification. We have conducted extensive experiments on the Gaofen dataset, multispectral data, and multiple hyperspectral datasets, validating the superiority of the Spectral Weaver model in forest hyperspectral image classification. The experimental results show that our model achieves 98.59% accuracy on multispectral data, surpassing ViT’s 96.30%. On the Jilin-1 dataset, our proposed algorithm achieved an accuracy of 98.95%, which is 2.17% higher than ViT. The model significantly outperforms classic ViT and other state-of-the-art backbone networks in classification performance. Not only does it effectively capture the spectral features of forest vegetation, but it also significantly improves the accuracy and robustness of classification, providing strong technical support for the refined management and conservation of forest resources.
Source: Forests
DOI: 10.3390/f16010021
Authors: Haotian Yu, Xuyang Li, Xinggui Xu, Hong Li, Xiangsuo Fan
Language: English
Subfield: Media Technology
Field: Engineering
Domain: Physical Sciences
SDG: Life on land
Open Access status: gold
Estimated APC paid: $2,955
Awards: G2595788366, G4164185178
OpenAlex ID: W4405817973
