3 个月前

基于注意力机制的自适应光谱-空间核ResNet用于高光谱图像分类

基于注意力机制的自适应光谱-空间核ResNet用于高光谱图像分类

摘要

高光谱图像(HSI)通过堆叠数百个连续的窄波段,提供了丰富的光谱-空间信息。然而,由于噪声的存在以及波段间的高度相关性,如何有效选择具有判别性的光谱-空间核特征仍是一个挑战。传统方法通常采用具有固定感受野(Receptive Field, RF)的卷积神经网络(CNN)来应对这一问题,但这类方法在前向与反向传播优化过程中,难以使神经元自适应地调整感受野大小,也无法有效建模跨通道依赖关系。本文提出了一种基于注意力机制的自适应光谱-空间核改进型残差网络(A²S²K-ResNet),并引入光谱注意力机制,以端到端的方式捕获高光谱图像分类任务中具有判别性的光谱-空间特征。具体而言,所提出的网络通过改进的三维残差块(3-D ResBlock)学习具有选择性的三维卷积核,实现光谱与空间特征的联合提取;同时,设计了一种高效的特征重校准(Efficient Feature Recalibration, EFR)机制,进一步提升分类性能。在三个广泛使用的高光谱数据集(IP、KSC 和 UP)上进行了大量实验验证,结果表明,与现有方法相比,所提出的 A²S²K-ResNet 在总体分类精度(OA)、平均分类精度(AA)以及 Kappa 系数等关键指标上均取得了更优的表现。相关源代码将公开发布于:https://github.com/suvojit-0x55aa/A2S2K-ResNet。

基准测试

基准方法指标
hyperspectral-image-classification-on-indianA2S2K-ResNet
Overall Accuracy: 99.57 %
hyperspectral-image-classification-on-kennedyA2S2K-ResNet
Overall Accuracy: 99.34
hyperspectral-image-classification-on-paviaA2S2K-ResNet
Overall Accuracy: 99.85

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于注意力机制的自适应光谱-空间核ResNet用于高光谱图像分类 | 论文 | HyperAI超神经