Hyperspectral image classification using a new deep learning model based on pseudo-3D block and depth separable 2D-3D convolution

被引:6
作者
Rani, Kumi [1 ]
Kumar, Sunil [1 ]
机构
[1] Indian Inst Technol BHU, Dept Math Sci, Varanasi, Uttar Pradesh, India
关键词
Deep learning; Convolutional neural network; Pseudo-3D block; Depth separable 2D-3D convolution; Spatial pyramid pooling; Hyperspectral image classification; NEURAL-NETWORKS;
D O I
10.1016/j.engappai.2023.107738
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In the task of classification of hyperspectral images (HSI), standard 2D and 3D convolutions increase the number of trainable parameters, making the model computationally complex, prone to overfitting when trained on smaller datasets, and demanding a large amount of memory. Thus, the cost of computation and memory consumption are bottlenecks of deep hybrid 2D-3D CNN models. This inspired us to propose a light weighted hybrid CNN model with fewer trainable parameters for efficient and accurate classification of HSI by introducing the concept of pseudo-3D block and depth separable 2D-3D convolution. The major components of the proposed framework are standard 3D-CNN, pseudo-3D block, depthwise separable 3D CNN, depth separable 2D CNN, and spatial pyramid pooling (SPP). Pseudo-3D block and depth separable 2D-3D convolution make the model computational and memory efficient by drastically reducing the number of trainable parameters. Further, the SPP layer is placed between the convolution layers and fully connected layers to generate a feature vector of fixed length at different scales. The proposed model is applied to three publicly available datasets. Experimental analysis proves the efficacy of the proposed approach over the state-of-the-art methods.
引用
收藏
页数:14
相关论文
共 49 条
[31]   Towards better exploiting convolutional neural networks for remote sensing scene classification [J].
Nogueira, Keiller ;
Penatti, Otavio A. B. ;
dos Santos, Jefersson A. .
PATTERN RECOGNITION, 2017, 61 :539-556
[32]   Crop type classification with hyperspectral images using deep learning : a transfer learning approach [J].
Patel, Usha ;
Pathan, Mohib ;
Kathiria, Preeti ;
Patel, Vibha .
MODELING EARTH SYSTEMS AND ENVIRONMENT, 2023, 9 (02) :1977-1987
[33]   SSNET: an improved deep hybrid network for hyperspectral image classification [J].
Paul, Arati ;
Bhoumik, Sanghamita ;
Chaki, Nabendu .
NEURAL COMPUTING & APPLICATIONS, 2021, 33 (05) :1575-1585
[34]   Learning Spatio-Temporal Representation with Pseudo-3D Residual Networks [J].
Qiu, Zhaofan ;
Yao, Ting ;
Mei, Tao .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :5534-5542
[35]   Hyperspectral Image Classification Model Using Squeeze and Excitation Network with Deep Learning [J].
Rajendran, T. ;
Valsalan, Prajoona ;
Amutharaj, J. ;
Jenifer, M. ;
Rinesh, S. ;
Latha, G. Charlyn Pushpa ;
Anitha, T. .
COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2022, 2022
[36]   Effective Feature Extraction and Data Reduction in Remote Sensing Using Hyperspectral Imaging [J].
Ren, Jinchang ;
Zabalza, Jaime ;
Marshall, Stephen ;
Zheng, Jiangbin .
IEEE SIGNAL PROCESSING MAGAZINE, 2014, 31 (04) :149-154
[37]   A study of deep convolutional auto-encoders for anomaly detection in videos [J].
Ribeiro, Manasses ;
Lazzaretti, Andre Eugenio ;
Lopes, Heitor Silverio .
PATTERN RECOGNITION LETTERS, 2018, 105 :13-22
[38]   HybridSN: Exploring 3-D-2-D CNN Feature Hierarchy for Hyperspectral Image Classification [J].
Roy, Swalpa Kumar ;
Krishna, Gopal ;
Dubey, Shiv Ram ;
Chaudhuri, Bidyut B. .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2020, 17 (02) :277-281
[39]  
Salakhutdinov R., 2009, P ART INT STAT, P448, DOI DOI 10.1109/CVPR.2009.5206577
[40]   Deep neural networks-based relevant latent representation learning for hyperspectral image classification [J].
Sellami, Akrem ;
Tabbone, Salvatore .
PATTERN RECOGNITION, 2022, 121