PCA-ViT: Hyperspectral Image Classification using Principal Component Analysis and Vision Transformer

Kian Ming Lim, Chin Poo Lee, Zaim Zahisham, Jit Yan Lim, Jashila Nair Mogan

Research output: Journal PublicationConference articlepeer-review

Abstract

Hyperspectral image classification is essential in remote sensing applications, aiming to accurately categorize land cover or materials depicted in hyperspectral data. This paper introduces PCA-ViT, a novel approach for hyperspectral image classification that integrates Principal Component Analysis (PCA) with the Vision Transformer (ViT) architecture. PCA serves as a spectral dimension reduction technique to alleviate the curse of dimensionality inherent in hyperspectral data, transforming the original data into a more manageable format. The transformed data are then input into the ViT model, which utilizes self-attention mechanisms to capture spatial dependencies among image patches, avoiding traditional convolutional layers. Extensive experiments on benchmark hyperspectral datasets, including Indian Pines, University of Pavia, and Salinas Scene, demonstrate PCA-ViT's superior performance. It achieves 99.95% accuracy on Indian Pines, 100% on University of Pavia, and 100% on Salinas Scene, showcasing the effectiveness of transformer-based architectures in hyperspectral image classification tasks.

Original languageEnglish
Pages (from-to)30-34
Number of pages5
JournalProceedings of the IEEE Conference on Systems, Process and Control, ICSPC
Issue number2024
DOIs
Publication statusPublished - 2024
Event12th IEEE Conference on Systems, Process and Control, ICSPC 2024 - Malacca, Malaysia
Duration: 7 Dec 2024 → …

Keywords

  • hyperspectral image classification
  • Principal Component Analysis
  • Vision Transformer

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Science Applications
  • Information Systems
  • Information Systems and Management
  • Safety, Risk, Reliability and Quality
  • Control and Optimization
  • Modelling and Simulation
  • Education

Cite this