DMCCT: Dual-Branch Multi-Granularity Convolutional Cross-Substitution Transformer for Hyperspectral Image Classification
Source abstract
Abstract
In the field of hyperspectral image classification, deep learning technology, especially convolutional neural networks, has achieved remarkable progress. However, convolutional neural network models encounter challenges in hyperspectral image classification due to limitations in their receptive fields. Conversely, the global modeling capability of Transformers has garnered attention in hyperspectral image classification. Nevertheless, the high computational cost and inadequate local feature extraction hinder its widespread application. In this study, we propose a novel fusion model of convolutional neural networks and Transformers to enhance performance in hyperspectral image classification, namely the dual-branch multi-granularity convolutional cross-substitution Transformer (DMCCT). The proposed model adopts a dual-branch structure to separately extract spatial and spectral features, thereby mitigating mutual interference and information loss between spectral and spatial data during feature extraction. Moreover, a multi-granularity embedding module is introduced to facilitate multi-scale and multi-level local feature extraction for spatial and spectral information. In particular, the improved convolutional cross-substitution Transformer module effectively integrates convolution and Transformer, reducing the complexity of attention operations and enhancing the accuracy of hyperspectral image classification tasks. Subsequently, the proposed method is evaluated against existing approaches using three classical datasets, namely Pavia University, Kennedy Space Center, and Indian Pines. Experimental results demonstrate the efficacy of the proposed method, achieving significant classification results on these datasets with overall classification accuracies of 98.57%, 97.96%, and 96.59%, respectively. These results establish the superiority of the proposed method in the context of hyperspectral image classification under similar experimental conditions.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Remote-Sensing Image Classification
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Surveying and Geoinformation Engineering, East China University of Technology, Nanchang 330013, China
School of Geophysics and Measurement-Control Technology, East China University of Technology, Nanchang 330013, China
References (55)
Show source-provided references
1. Gao, L., Wang, Z., Zhuang, L., Yu, H., Zhang, B., and Chanussot, J. (2021). Using low-rank representation of abundance maps and nonnegative tensor factorization for hyperspectral nonlinear unmixing. IEEE Trans. Geosci. Remote Sens., 60.10.1109/TGRS.2021.3065990 ↗
2. HSI-BERT: Hyperspectral image classification using the bidirectional encoder representation from transformers10.1109/TGRS.2019.2934760 ↗
3. Hyperspectral imagery applications for precision agriculture-a systemic survey10.1007/s11042-021-11729-8 ↗
4. Ram, B.G., Oduor, P., Igathinathane, C., Howatt, K., and Sun, X. (2024). A systematic review of hyperspectral imaging in precision agriculture: Analysis of its current state and future prospects. Comput. Electron. Agric., 222.10.1016/j.compag.2024.109037 ↗
5. Delalieux, S., Somers, B., Haest, B., Kooistra, L., Mücher, C.A., and Borre, J.V. (2010, January 14–16). Monitoring heathland habitat status using hyperspectral image classification and unmixing. Proceedings of the 2010 2nd Workshop on Hyperspectral Image and Signal Processing: Evolution in Remote Sensing, Reykjavik, Iceland.10.1109/WHISPERS.2010.5594895 ↗
6. Unsupervised ore/waste classification on open-cut mine faces using close-range hyperspectral data10.1016/j.gsf.2023.101562 ↗
7. Fine-grained classification of urban functional zones and landscape pattern analysis using hyperspectral satellite imagery: A case study of wuhan10.1109/JSTARS.2022.3174412 ↗
8. Spectral–spatial morphological attention transformer for hyperspectral image classification10.1109/TGRS.2023.3242346 ↗
9. Song, W., Li, S., Kang, X., and Huang, K. (2016, January 10–15). Hyperspectral image classification based on KNN sparse representation. Proceedings of the 2016 IEEE international geoscience and remote sensing symposium (IGARSS), Beijing, China.10.1109/IGARSS.2016.7729622 ↗
10. Classification of hyperspectral remote sensing images with support vector machines10.1109/TGRS.2004.831865 ↗
11. Random forest in remote sensing: A review of applications and future directions10.1016/j.isprsjprs.2016.01.011 ↗
12. Guo, Y., Yin, X., Zhao, X., Yang, D., and Bai, Y. (2019). Hyperspectral image classification with SVM and guided filter. EURASIP J. Wirel. Commun. Netw., 2019.10.1186/s13638-019-1346-z ↗
13. A fast learning algorithm for deep belief nets10.1162/neco.2006.18.7.1527 ↗
14. Deng, F., Pu, S., Chen, X., Shi, Y., Yuan, T., and Pu, S. (2018). Hyperspectral image classification with capsule network using limited training samples. Sensors, 18.10.3390/s18093153 ↗
15. Deep feature extraction and classification of hyperspectral images based on convolutional neural networks10.1109/TGRS.2016.2584107 ↗
16. Convolutional neural networks for hyperspectral image classification10.1016/j.neucom.2016.09.010 ↗
17. Li, Y., Zhang, H., and Shen, Q. (2017). Spectral–spatial classification of hyperspectral imagery with 3D convolutional neural network. Remote Sens., 9.10.3390/rs9010067 ↗
18. Zhong, Z., Li, J., Ma, L., Jiang, H., and Zhao, H. (2017, January 23–28). Deep residual networks for hyperspectral image classification. Proceedings of the 2017 IEEE international geoscience and remote sensing symposium (IGARSS), Fort Worth, TX, USA.10.1109/IGARSS.2017.8127330 ↗
19. Hyperspectral image classification using deep pixel-pair features10.1109/TGRS.2016.2616355 ↗
20. Spectral–spatial residual network for hyperspectral image classification: A 3-D deep learning framework10.1109/TGRS.2017.2755542 ↗
21. Li, R., Zheng, S., Duan, C., Yang, Y., and Wang, X. (2020). Classification of hyperspectral image based on double-branch dual-attention mechanism network. Remote Sens., 12.10.20944/preprints201912.0059.v2 ↗
22. Multiscale Spatial-Spectral Feature Extraction Network for Hyperspectral Image Classification10.1109/JSTARS.2022.3179446 ↗
23. Fu, L., Chen, X., Pirasteh, S., and Xu, Y. (2023). The Classification of Hyperspectral Images: A Double-Branch Multi-Scale Residual Network. Remote Sens., 15.10.3390/rs15184471 ↗
24. He, X., Chen, Y., and Lin, Z. (2021). Spatial-spectral transformer for hyperspectral image classification. Remote Sens., 13.10.3390/rs13030498 ↗
25. Yuan, D., Yu, D., Qian, Y., Xu, Y., and Liu, Y. (2023). S2Former: Parallel Spectral–Spatial Transformer for Hyperspectral Image Classification. Electronics, 12.10.3390/electronics12183937 ↗
26. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., and Gelly, S. (2020). An image is worth 16 × 16 words: Transformers for image recognition at scale. arXiv.
27. Liu, B., Liu, Y., Zhang, W., Tian, Y., and Kong, W. (2023). Spectral swin transformer network for hyperspectral image classification. Remote Sens., 15.10.3390/rs15153721 ↗
28. Xie, J., Hua, J., Chen, S., Wu, P., Gao, P., Sun, D., Lyu, Z., Lyu, S., Xue, X., and Lu, J. (2023). HyperSFormer: A transformer-based end-to-end hyperspectral image classification method for crop classification. Remote Sens., 15.10.2139/ssrn.4425906 ↗
29. Mei, S., Song, C., Ma, M., and Xu, F. (2022). Hyperspectral image classification using group-aware hierarchical transformer. IEEE Trans. Geosci. Remote Sens., 60.10.1109/TGRS.2022.3207933 ↗
30. He, X., Chen, Y., and Li, Q. (2022). Two-branch pure transformer for hyperspectral image classification. IEEE Geosci. Remote Sens. Lett., 19.10.1109/LGRS.2022.3217775 ↗
31. Peng, Y., Zhang, Y., Tu, B., Li, Q., and Li, W. (2022). Spatial–spectral transformer with cross-attention for hyperspectral image classification. IEEE Trans. Geosci. Remote Sens., 60.10.1109/TGRS.2022.3203476 ↗
32. Zhang, J., Meng, Z., Zhao, F., Liu, H., and Chang, Z. (2022). Convolution transformer mixer for hyperspectral image classification. IEEE Geosci. Remote Sens. Lett., 19.10.1109/LGRS.2022.3208935 ↗
33. Zhang, Z., Li, T., Tang, X., Hu, X., and Peng, Y. (2022). CAEVT: Convolutional autoencoder meets lightweight vision transformer for hyperspectral image classification. Sensors, 22.10.3390/s22103902 ↗
34. Guo, J., Han, K., Wu, H., Tang, Y., Chen, X., Wang, Y., and Xu, C. (2022, January 18–24). Cmt: Convolutional neural networks meet vision transformers. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA.10.1109/CVPR52688.2022.01186 ↗
35. A multi-range spectral-spatial transformer for hyperspectral image classification10.1016/j.infrared.2023.104983 ↗
36. Zhang, K., Tan, Z., Sun, J., Zhu, B., Yang, Y., and Lv, Q. (2023). A Multidimensional Spectral Transformer with Channel-Wise Correlation for Hyperspectral Image Classification. Appl. Sci., 13.10.3390/app13095482 ↗
37. Wang, A., Xing, S., Zhao, Y., Wu, H., and Iwahori, Y. (2022). A hyperspectral image classification method based on adaptive spectral spatial kernel combined with improved vision transformer. Remote Sens., 14.10.3390/rs14153705 ↗
38. Bai, Z., Wan, L., Zhou, M., Jiang, S., and Zhang, H. (2024). Tripartite-structure transformer for hyperspectral image classification. Comput. Intell., 40.10.22541/au.169086249.98782131/v1 ↗
39. Zhang, X., Su, Y., Gao, L., Bruzzone, L., Gu, X., and Tian, Q. (2023). A lightweight transformer network for hyperspectral image classification. IEEE Trans. Geosci. Remote Sens., 61.10.1109/TGRS.2023.3297858 ↗
40. Yang, L., Yang, Y., Yang, J., Zhao, N., Wu, L., Wang, L., and Wang, T. (2022). FusionNet: A convolution–transformer fusion network for hyperspectral image classification. Remote Sens., 14.10.3390/rs14164066 ↗
41. Yang, H., Yu, H., Zheng, K., Hu, J., Tao, T., and Zhang, Q. (2023). Hyperspectral image classification based on interactive transformer and CNN with multilevel feature fusion network. IEEE Geosci. Remote Sens. Lett., 20.10.1109/LGRS.2023.3303008 ↗
42. Han, D., Ye, T., Han, Y., Xia, Z., Song, S., and Huang, G. (2023). Agent Attention: On the Integration of Softmax and Linear Attention. arXiv.
43. Krizhevsky, A., Sutskever, I., and Hinton, G.E. (2012, January 3–6). Imagenet classification with deep convolutional neural networks. Proceedings of the Advances in Neural Information Processing Systems, Lake Tahoe, NV, USA.
44. Deep learning: Methods and applications10.1561/2000000039 ↗
45. He, K., Zhang, X., Ren, S., and Sun, J. (2015, January 7–13). Delving deep into rectifiers: Surpassing human-level performance on imagenet classification. Proceedings of the IEEE International Conference on Computer Vision, Santiago, Chile.10.1109/ICCV.2015.123 ↗
46. Deeplab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected crfs10.1109/TPAMI.2017.2699184 ↗
47. Yu, F., and Koltun, V. (2015). Multi-scale context aggregation by dilated convolutions. arXiv.
48. Howard, A.G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., Andreetto, M., and Adam, H. (2017). Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv.
49. Chollet, F. (2017, January 21–26). Xception: Deep learning with depthwise separable convolutions. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.195 ↗
50. Ashish, V. (2017, January 4–9). Attention is all you need. Proceedings of the Advances in Neural Information Processing Systems, Long Beach, CA, USA.
51. CSiT: A multiscale vision transformer for hyperspectral image classification10.1109/JSTARS.2022.3216335 ↗
52. Qing, Y., Liu, W., Feng, L., and Gao, W. (2021). Improved transformer net for hyperspectral image classification. Remote Sens., 13.10.3390/rs13112216 ↗
53. Ba, J.L., Kiros, J.R., and Hinton, G.E. (2016). Layer normalization. arXiv.
54. Xu, H., Zeng, Z., Yao, W., and Lu, J. (2023). CS2DT: Cross spatial–spectral dense transformer for hyperspectral image classification. IEEE Geosci. Remote Sens. Lett., 20.10.1109/LGRS.2023.3321343 ↗
55. Multi-granularity vision transformer via semantic token for hyperspectral image classification10.1080/01431161.2022.2142078 ↗