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 条
[1]  
Agarap A. Fred., 2018, PREPRINT, DOI DOI 10.48550/ARXIV.1803.08375
[2]   Aircraft Engines Remaining Useful Life Prediction Based on A Hybrid Model of Autoencoder and Deep Belief Network [J].
Al-Khazraji, Huthaifa ;
Nasser, Ahmed R. ;
Hasan, Ahmed M. ;
Al Mhdawi, Ammar K. ;
Al-Raweshidy, Hamed ;
Humaidi, Amjad J. .
IEEE ACCESS, 2022, 10 :82156-82163
[3]   Efficient 3D Deep Learning Model for Medical Image Semantic Segmentation [J].
Alalwan, Nasser ;
Abozeid, Amr ;
ElHabshy, AbdAllah A. ;
Alzahrani, Ahmed .
ALEXANDRIA ENGINEERING JOURNAL, 2021, 60 (01) :1231-1239
[4]   Multipath feature fusion for hyperspectral image classification based on hybrid 3D/2D CNN and squeeze-excitation network [J].
Ari, Ali .
EARTH SCIENCE INFORMATICS, 2023, 16 (01) :175-191
[5]   Deep Feature Extraction and Classification of Hyperspectral Images Based on Convolutional Neural Networks [J].
Chen, Yushi ;
Jiang, Hanlu ;
Li, Chunyang ;
Jia, Xiuping ;
Ghamisi, Pedram .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2016, 54 (10) :6232-6251
[6]   Remote Sensing Image Scene Classification: Benchmark and State of the Art [J].
Cheng, Gong ;
Han, Junwei ;
Lu, Xiaoqiang .
PROCEEDINGS OF THE IEEE, 2017, 105 (10) :1865-1883
[7]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[8]   Hyperspectral Imaging Applications in Agriculture and Agro-Food Product Quality and Safety Control: A Review [J].
Dale, Laura M. ;
Thewis, Andre ;
Boudry, Christelle ;
Rotar, Ioan ;
Dardenne, Pierre ;
Baeten, Vincent ;
Pierna, Juan A. Fernandez .
APPLIED SPECTROSCOPY REVIEWS, 2013, 48 (02) :142-159
[9]   Deep Hashing Neural Networks for Hyperspectral Image Feature Extraction [J].
Fang, Leyuan ;
Liu, Zhiliang ;
Song, Weiwei .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (09) :1412-1416
[10]   A New Spatial-Spectral Feature Extraction Method for Hyperspectral Images Using Local Covariance Matrix Representation [J].
Fang, Leyuan ;
He, Nanjun ;
Li, Shutao ;
Plaza, Antonio J. ;
Plaza, Javier .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2018, 56 (06) :3534-3546