返回
Adversarial Prototype Learning for Hyperspectral Image Classification
DOI:10.1109/TGRS.2021.3100496.png)
摘要
En 中文
In hyperspectral image (HSI) classification, the training set often contains a very limited number of high-dimensional samples, which can cause overfitting problems, especially in deep learning (DL) frameworks. This situation worsens when a bias exists between the feature distributions of the training and testing sets. In this article, we propose a novel method, referred to as adversarial prototype learning (APL), for learning an accurate HSI classification model in a uniform manner when the training set contains few, high-dimensional, and biased samples. APL consists of a prototype learning module (PLM) and an adversarial alignment module (AAM). The PLM aims to alleviate overfitting by training prototypical classifiers with a simple inductive bias in the initial feature space. The AAM aims to reduce the bias between the feature distributions of the training and testing sets using two adversarial prototypical classifiers learned by the PLM. Iteratively training the PLM and AAM results in alignment of the feature distributions between the training and testing sets while improving the generalization ability of the prototypical classifiers. The theoretical analysis indicates that APL is able to lower the upper error bound when classifying testing samples. We further apply APL in a DL framework to establish the adversarial prototypical network (APNet) architecture. Experimental results on four publicly available HSI datasets demonstrate that the proposed APNet alleviates overfitting, aligns the feature distributions between the training and testing sets, and achieves state-of-the-art performance.
Keyword:
Feature extraction
Training
Testing
Prototypes
Active appearance model
Task analysis
Hyperspectral imaging
Adversarial learning
feature distribution alignment
hyperspectral image (HSI) classification
prototype learning
期刊
IF:
8.6
论文数:
2.2W
被引数:
10.7W
机构
引用论文
Deep Pyramidal Residual Networks for Spectral-Spatial Hyperspectral Image Classification用于光谱-空间高光谱图像分类的深度金字塔残差网络
Attention Multibranch Convolutional Neural Network for Hyperspectral Image Classification Based on Adaptive Region Search基于自适应区域搜索的注意力多分支卷积神经网络高光谱图像分类

