Hyperspectral Image Classification Based on Hybrid Depth-Wise Separable Convolution and Dual-Branch Feature Fusion Network
Source abstract
Abstract
Recently, advancements in convolutional neural networks (CNNs) have significantly contributed to the advancement of hyperspectral image (HSI) classification. However, the problem of limited training samples is the primary obstacle to obtaining further improvements in HSI classification. The traditional methods relying solely on 2D-CNN for feature extraction underutilize the inter-band correlations of HSI, while the methods based on 3D-CNN alone for feature extraction lead to an increase in training parameters. To solve the above problems, we propose an HSI classification network based on hybrid depth-wise separable convolution and dual-branch feature fusion (HDCDF). The dual-branch structure is designed in HDCDF to extract simultaneously integrated spectral–spatial features and obtain complementary features via feature fusion. The proposed modules of 2D depth-wise separable convolution attention (2D-DCAttention) block and hybrid residual blocks are applied to the dual branch, respectively, further extracting more representative and comprehensive features. Instead of full 3D convolutions, HDCDF uses hybrid 2D–3D depth-wise separable convolutions, offering computational efficiency. Experiments are conducted on three benchmark HSI datasets: Indian Pines, University of Pavia, and Salinas Valley. The experimental results show that the proposed method showcases superior performance when the training samples are extremely limited, outpacing the state-of-the-art method by an average of 2.03% in the overall accuracy of three datasets, which shows that HDCDF has a certain potential in HSI classification.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Remote Sensing and Land Use · Remote-Sensing Image Classification
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Computer and Information Engineering, Tianjin Chengjian University, Tianjin 300384, China
Funding
Tianjin Science and Technology Planning Project
References (43)
Show source-provided references
1. Hyperspectral Image Classification—Traditional to Deep Models: A Survey for Future Prospects10.1109/JSTARS.2021.3133021 ↗
2. Feature extraction for hyperspectral image classification: A review10.1080/01431161.2020.1736732 ↗
3. Feature disentanglement based domain adaptation network for cross-scene coastal wetland hyperspectral image classification
4. A systematic review of hyperspectral imaging in precision agriculture: Analysis of its current state and future prospects10.1016/j.compag.2024.109037 ↗
5. Soil carbon content prediction using multi-source data feature fusion of deep learning based on spectral and hyperspectral images10.1016/j.chemosphere.2023.139161 ↗
6. Hyperspectral image classification using deep learning models: A review10.1088/1742-6596/1950/1/012087 ↗
7. Deep support vector machine for hyperspectral image classification10.1016/j.patcog.2020.107298 ↗
8. CSKNN: Cost-sensitive K-Nearest Neighbor using hyperspectral imaging for identification of wheat varieties10.1016/j.compeleceng.2023.108896 ↗
9. Spectral–spatial and cascaded multilayer random forests for tree species classification in airborne hyperspectral images10.1109/TGRS.2022.3177935 ↗
10. Spectral–spatial hyperspectral image segmentation using subspace multinomial logistic regression and Markov random fields10.1109/TGRS.2011.2162649 ↗
11. Semisupervised hyperspectral image segmentation using multinomial logistic regression with active learning
12. A multiscale dual-branch feature fusion and attention network for hyperspectral images classification10.1109/JSTARS.2021.3103176 ↗
13. Recursive autoencoders-based unsupervised feature learning for hyperspectral image classification10.1109/LGRS.2017.2737823 ↗
14. Hyperspectral image band selection based on CNN embedded GA (CNNeGA)10.1109/JSTARS.2023.3242310 ↗
15. Hyperspectral Target Detection-Based 2D-3D Parallel Convolutional Neural Networks for Hyperspectral Image Classification10.1109/JSTARS.2024.3394704 ↗
16. A multi-scale multi-channel CNN introducing a channel-spatial attention mechanism hyperspectral remote sensing image classification method10.1080/22797254.2024.2353290 ↗
17. Tree-shaped multiobjective evolutionary CNN for hyperspectral image classification10.1016/j.asoc.2023.111176 ↗
18. Deep recurrent neural networks for hyperspectral image classification10.1109/TGRS.2016.2636241 ↗
19. Degradation estimation recurrent neural network with local and non-local priors for compressive spectral imaging10.1109/TGRS.2024.3410272 ↗
20. Robust hyperspectral image classification using generative adversarial networks10.1016/j.ins.2024.120452 ↗
21. HybridSN: Exploring 3-D–2-D CNN feature hierarchy for hyperspectral image classification10.1109/LGRS.2019.2918719 ↗
22. Principal component analysis for hyperspectral image classification
23. Pyramid hierarchical spatial-spectral transformer for hyperspectral image classification10.1109/JSTARS.2024.3461851 ↗
24. Ma, W., Yang, Q., Wu, Y., Zhao, W., and Zhang, X. (2019). Double-branch multi-attention mechanism network for hyperspectral image classification. Remote Sens., 11.10.3390/rs11111307 ↗
25. Feature extraction for hyperspectral imagery: The evolution from shallow to deep: Overview and toolbox10.1109/MGRS.2020.2979764 ↗
26. A novel spatial-spectral pyramid network for hyperspectral image classification
27. Hyperspectral Image Classification Using Groupwise Separable Convolutional Vision Transformer Network
28. Alexey, D. (2020). An image is worth 16x16 words: Transformers for image recognition at scale. arXiv.
29. SpectralFormer: Rethinking hyperspectral image classification with transformers
30. Extended vision transformer (ExViT) for land use and land cover classification: A multimodal deep learning framework10.1109/TGRS.2023.3284671 ↗
31. Hyperspectral image transformer classification networks
32. Deep convolutional neural networks for hyperspectral image classification10.1155/2015/258619 ↗
33. Hyperspectral image classification method based on 2D–3D CNN and multibranch feature fusion10.1109/JSTARS.2020.3024841 ↗
34. A simplified 2D-3D CNN architecture for hyperspectral image classification based on spatial–spectral fusion10.1109/JSTARS.2020.2983224 ↗
35. Fang, B., Li, Y., Zhang, H., and Chan, J.C.-W. (2019). Hyperspectral images classification based on dense convolutional networks with spectral-wise attention mechanism. Remote Sens., 11.10.3390/rs11020159 ↗
36. Kanthi, M., Sarma, T.H., and Bindu, C.S. (2020, January 1–4). A 3D-deep CNN based feature extraction and hyperspectral image classification. Proceedings of the 2020 IEEE India Geoscience and Remote Sensing Symposium (InGARSS), Ahmedabad, India.10.1109/InGARSS48198.2020.9358920 ↗
37. Zhou, J., Zeng, S., Xiao, Z., Zhou, J., Li, H., and Kang, Z. (2022). An enhanced spectral fusion 3D CNN model for hyperspectral image classification. Remote Sens., 14.10.3390/rs14215334 ↗
38. Zhang, Z., Gao, D., Liu, D., and Shi, G. (2024). Spectral-spatial domain attention network for hyperspectral image few-shot classification. Remote Sens., 16.10.3390/rs16030592 ↗
39. A Deep Spectral-Spatial Residual Attention Network for Hyperspectral Image Classification10.1109/JSTARS.2024.3355071 ↗
40. A depthwise separable fully convolutional ResNet with ConvCRF for semisupervised hyperspectral image classification10.1109/JSTARS.2021.3073661 ↗
41. An overview on spectral and spatial information fusion for hyperspectral image classification: Current trends and challenges10.1016/j.inffus.2020.01.007 ↗
42. Woo, S., Park, J., Lee, J.-Y., and Kweon, I.S. (2018, January 8–14). Cbam: Convolutional block attention module. Proceedings of the European Conference on Computer Vision (ECCV), Munich, Germany.10.1007/978-3-030-01234-2_1 ↗
43. LSSMA: Lightweight Spectral-Spatial Neural Architecture with Multi-Attention Feature Extraction for Hyperspectral Image Classification10.1109/JSTARS.2024.3371536 ↗