Abstract:In scenarios where sample collection is limited, studying few-shot learning algorithms such as prototypical networks (P-Net) is a keynote topic for supervised multiple tree species classification. In a previous study, we improved the P-Net by combining the feature enhancement algorithm based on the convolutional block attention module and several popular data augmentation methods in the computer vision domain, the classification accuracy can be significantly increased, and the degree of model overfitting can be reduced. However, there is a clear boundary between the data augmentations and the feature enhancement algorithm, which is manifested in that data augmentations are only used to enrich the diversity of the learned samples, but cannot directly affect the construction of the objective function, thus limiting the ability of data augmentations. In fact, in the supervised contrastive learning research, data augmentation methods are often used to generate positive samples of an anchor image to construct the objective function, i.e. supervised contrastive loss. The core idea for solving such a boundary problem is to use contrastive learning to make the anchor image close to its positive samples and the negative samples away from each other. Inspired by this, we introduced supervised contrastive learning in the P-Net, i.e., SCL-P-Net, which takes the discriminative feature representations as the constraints of the prototype clustering algorithm. In SCL-P-Net, data augmentation methods can not only extend the sample distribution, but also be used to construct the supervised contrastive loss directly. The study involves four airborne hyperspectral image datasets related to tree species classification, including the GFF-A and GFF-B datasets collected from Gaofeng Forest Farm in Nanning City, Guangxi Province, South China, the Xiongan dataset from Matiwan Village in Xiongan New Area, Hebei Province, North China, and the Tea Farm dataset from Fanglu Tea Farm in Changzhou City, Jiangsu Province, East China. The highest overall accuracy (OA) for the four datasets is 99.23% for GFF-A, 98.39% for GFF-B, 99.30% for Xiongan, and 99.54% for Tea Farm. It is concluded that the proposed SCL-P-Net classification framework can achieve multiple tree species classification with high-precision. Without changing the basic classification framework of P-Net, the introduction of supervised contrastive learning makes the combination of the data augmentations and the feature enhancement algorithm and plays a positive role in improving the distinguishability between samples.

Discriminative feature constraints via supervised contrastive learning for few-shot forest tree species classification using airborne hyperspectral images

Data Augmentation in Prototypical Networks for Forest Tree Species Classification Using Airborne Hyperspectral Images

An Improved Res-UNet Model for Tree Species Classification Using Airborne High-Resolution Images

ContrastNet: Unsupervised Feature Learning by Autoencoder and Prototypical Contrastive Learning for Hyperspectral Imagery Classification.

A New CBAM-P-Net Model for Few-Shot Forest Species Classification Using Airborne Hyperspectral Images

Few-Shot Remote Sensing Scene Classification With Spatial Affinity Attention and Class Surrogate-Based Supervised Contrastive Learning

Enhancing Hyperspectral Image Classification: Leveraging Unsupervised Information With Guided Group Contrastive Learning

Deeply supervised network for airborne LiDAR tree classification incorporating dual attention mechanisms

Nearest Neighbor-Based Contrastive Learning for Hyperspectral and LiDAR Data Classification

Classification of Typical Tree Species in Laser Point Cloud Based on Deep Learning

Improved Prototypical Network Model for Forest Species Classification in Complex Stand

Purified Contrastive Learning With Global and Local Representation for Hyperspectral Image Classification

Cross-Domain Few-Shot Hyperspectral Image Classification With Cross-Modal Alignment and Supervised Contrastive Learning

More appropriate DenseNetBL classifier for small sample tree species classification using UAV-based RGB imagery

SCL-MLNet: Boosting Few-Shot Remote Sensing Scene Classification via Self-Supervised Contrastive Learning

A deep-learning-based tree species classification for natural secondary forests using unmanned aerial vehicle hyperspectral images and LiDAR

A Hybrid convolution neural network for the classification of tree species using hyperspectral imagery

CTF-SSCL: CNN-Transformer for Few-Shot Hyperspectral Image Classification Assisted by Semisupervised Contrastive Learning

Multi-pretext-task Prototypes Guided Dynamic Contrastive Learning Network for Few-shot Remote Sensing Scene Classification

Two-Stream Networks for Contrastive Learning in Hyperspectral Image Classification

Contrastive Graph Convolution Network With Skip Connections for Few-Shot Hyperspectral Image Classification