通过动态专家卷积提升高光谱图像分类精度,避免过拟合。
Expert Kernel Generation Network Driven by Contextual Mapping for Hyperspectral Image Classification
- 基于上下文映射生成动态卷积核,自适应融合多维特征。
- 在IN、UP、KSC数据集上准确率优于主流方法,最高提升2.1%。
- 无需增加网络深度或宽度,适合资源受限场景使用。
深度神经网络在高光谱图像分类中面临高维数据、地物分布稀疏和光谱冗余等挑战,常导致过拟合与泛化能力不足。本文提出EKGNet,基于改进的3D-DenseNet架构,包含上下文感知映射网络与动态卷积核生成模块。该模块将输入的全局上下文信息转化为基卷积核组合指令,动态卷积核由K组基础卷积构成,类比于专精于不同维度基础模式的K个专家。映射模块根据输入生成有意义的组合权重,动态卷积模块则利用这些权重构建自适应专家卷积系统。此动态机制使模型能灵活聚焦于不同区域的关键空间结构,而非依赖单一静态卷积核的固定感受野。EKGNet通过3D动态专家卷积系统增强表示能力,不增加网络深度或宽度。在IN、UP、KSC数据集上表现优异,超越主流高光谱分类方法。
原文摘要 · Abstract (English)
Deep neural networks face several challenges in hyperspectral image classification, including high-dimensional data, sparse distribution of ground objects, and spectral redundancy, which often lead to classification overfitting and limited generalization capability. To more efficiently adapt to ground object distributions while extracting image features without introducing excessive parameters and skipping redundant information, this paper proposes EKGNet based on an improved 3D-DenseNet model, consisting of a context-aware mapping network and a dynamic kernel generation module. The context-aware mapping module translates global contextual information of hyperspectral inputs into instructions for combining base convolutional kernels, while the dynamic kernels are composed of K groups of base convolutions, analogous to K different types of experts specializing in fundamental patterns across various dimensions. The mapping module and dynamic kernel generation mechanism form a tightly coupled system - the former generates meaningful combination weights based on inputs, while the latter constructs an adaptive expert convolution system using these weights. This dynamic approach enables the model to focus more flexibly on key spatial structures when processing different regions, rather than relying on the fixed receptive field of a single static convolutional kernel. EKGNet enhances model representation capability through a 3D dynamic expert convolution system without increasing network depth or width. The proposed method demonstrates superior performance on IN, UP, and KSC datasets, outperforming mainstream hyperspectral image classification approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。