Instructional Mask Autoencoder: A Scalable Learner for Hyperspectral Image Classification

Abstract

Nowadays, an increasing number of hyperspectral images (HSIs) are becoming available. However, the utilization of unlabeled HSIs is extremely low due to the expensive annotation costs. Thus, it is crucial to figure out how to use these unlabeled HSIs and enhance the classification performance. Fortunately, self-supervised training enables us acquire latten features from unlabeled HSIs, thereby enhancing network performance via transfer learning. Whereas, most current networks for HSIs are inflexible, challenging to perform transfer learning and accommodate multimodal HSIs. Therefore, we devise a scalable self-supervised network called Instructional Mask Autoencoder(IMAE), which can extract general patterns of HSIs by these unannotated data. It primarily consists of a spatial-spectral embedding block and a transformer-based masked autoencoder, which are utilized for projecting input samples into the same latent space and learning higher-level semantic information, respectively. Moreover, we utilize a random token called <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$ins\_{t}oken$</tex-math></inline-formula> , to instruct the model learn components of global information which are highly correlated with the target pixel in HSI samlpe. In fine-tuning stage, we design a learnable aggregation mechanism to put all tokens into full play. The obtained results illustrate that our method exhibits robust generalization performance and accelerates convergence across diverse datasets. In cases of limited samples, we conducted experiments on three structurally distinct HSIs, all of which achieved competitive performance. Compared to state-of-the-art methods, our approach demonstrated respective improvements of 1.97%, 0.44% and 3.35% on these three datasets.