Deep Induction Network for Small Samples Classification of Hyperspectral Images

Abstract

Recently, the deep learning models have achieved great success in hyperspectral images (HSI) classification. However, most of the deep learning models fail to obtain satisfactory results under the condition of small samples due to the contradiction between the large parameter space of the deep learning models and the insufficient labeled samples in HSI. To address the problem, a deep model based on the induction network is designed in this article to improve the classification performance of HSI under the condition of small samples. Specifically, the typical meta-training strategy is adopted, enabling the model to acquire stronger generalization ability, so as to accurately distinguish the new classes with only a few labeled samples (e.g., five samples per class). Moreover, in order to deal with the disturbance caused by the various characteristics of the samples in the same class in HSI, the class-wise induction module is introduced utilizing the dynamic routing algorithm, which can induce the sample-wise representations to the class-wise level representations. The obtained class-wise level representations possess better separability, allowing the designed model to generate more accurate and robust classification results. Extensive experiments are carried out on three public HSI to verify the effectiveness of the proposed method. The results demonstrate that our method outperforms existing deep learning methods under the condition of small samples.