高光谱分类论文解读分享之基于生成对抗性少数过采样的高光谱图像分类

IEEE TGRS 2022:基于生成对抗性少数过采样的高光谱图像分类

题目

Generative Adversarial Minority Oversampling for Spectral--Spatial Hyperspectral Image Classification

作者

Swalpa Kumar Roy , Student Member, IEEE, Juan M. Haut , Senior Member, IEEE, Mercedes E. Paoletti , Senior Member, IEEE, Shiv Ram Dubey , Member, IEEE, and Antonio Plaza , Fellow, IEEE

关键词

Convolutional neural networks (CNNs), deep learning, spectral--spatial hyperspectral image (HSI)

classification.

研究动机

现有方法在将生成对抗网络应用于高光谱影像分类任务时,缺乏对类别不均衡问题的考虑和解决,限制了分类精度。

模型

近年来,卷积神经网络(CNN)在高光谱图像分类中表现出了良好的性能,一般情况下,每类需要大量的样本才能正确地训练CNN。然而,现有的HSI数据集存在严重的类不平衡问题,许多类没有足够的样本来表征光谱信息,现有的CNN模型的性能偏向于拥有更多训练样本的大多数类,本文研究了HSI分类中的数据不平衡问题。特别地,提出了一种新的3D-HyperGAMO模型,该模型采用生成性对抗性少数族裔过采样。该模型利用少数族裔类的已有样本,在训练时自动生成更多的少数族裔类样本。样本以三维高光谱斑块的形式生成。3D-HyperGAMO模型使用不同于生成器和鉴别器的分类器,使用原始样本和生成的样本来训练新生成的样本,以确定它们实际属于的类别。最后,将生成的数据与原始训练数据集进行分类组合,以学习类别的网络参数。最后,将生成的数据与原始训练数据集进行分类组合,以学习类别的网络参数训练好的三维分类器网络使用测试集验证了模型的性能。我们的实验考虑了四个基准HSI数据集,即印第安纳松树(IP)、肯尼迪航天中心(KSC)、帕维亚大学(UP)和博茨瓦纳(BW)。该模型在训练过程中表现出了出色的数据生成能力,显著提高了分类性能。

亮点

提出并构建了一个convex 3-D hyperspectral patch generator unit (PGU)来生成伪样本。

论文以及代码

论文链接: link

代码链接: link

相关推荐
子燕若水2 小时前
Unreal Engine 5中的AI知识
人工智能
极限实验室3 小时前
Coco AI 实战(一):Coco Server Linux 平台部署
人工智能
杨过过儿3 小时前
【学习笔记】4.1 什么是 LLM
人工智能
巴伦是只猫3 小时前
【机器学习笔记Ⅰ】13 正则化代价函数
人工智能·笔记·机器学习
大千AI助手3 小时前
DTW模版匹配:弹性对齐的时间序列相似度度量算法
人工智能·算法·机器学习·数据挖掘·模版匹配·dtw模版匹配
AI生存日记3 小时前
百度文心大模型 4.5 系列全面开源 英特尔同步支持端侧部署
人工智能·百度·开源·open ai大模型
LCG元4 小时前
自动驾驶感知模块的多模态数据融合:时序同步与空间对齐的框架解析
人工智能·机器学习·自动驾驶
why技术4 小时前
Stack Overflow,轰然倒下!
前端·人工智能·后端
彭祥.4 小时前
Jetson边缘计算主板:Ubuntu 环境配置 CUDA 与 cudNN 推理环境 + OpenCV 与 C++ 进行目标分类
c++·opencv·分类