用小参数扩展感受野,提升高光谱图像分类精度
3D Wavelet Convolutions with Extended Receptive Fields for Hyperspectral Image Classification
- 引入小波卷积,通过多频带分层处理扩展感受野
- 在IN/UP/KSC数据集上超越主流方法,精度显著提升
- 适合处理高维稀疏数据,轻量级设计利于部署
深度神经网络在高光谱图像分类中面临高维数据、地物分布稀疏和光谱冗余等挑战,常导致过拟合与泛化能力差。本文提出WCNet,一种结合小波变换的改进3D-DenseNet模型。通过级联小波变换实现波浪卷积,有效扩展卷积感受野,引导网络更关注低频成分。各卷积层聚焦输入信号不同频段,有效范围逐层递增,增强对空间结构的响应能力,同时仅增加少量可训练参数。该动态机制使模型能灵活适应不同区域特征,无需固定感受野。波浪卷积模块通过3D小波变换扩展感受野,不增加网络深度或宽度。实验表明,在IN、UP和KSC数据集上性能优于主流方法。
原文摘要 · Abstract (English)
Deep neural networks face numerous challenges in hyperspectral image classification, including high-dimensional data, sparse ground object distributions, and spectral redundancy, which often lead to classification overfitting and limited generalization capability. To better adapt to ground object distributions while expanding receptive fields without introducing excessive parameters and skipping redundant information, this paper proposes WCNet, an improved 3D-DenseNet model integrated with wavelet transforms. We introduce wavelet transforms to effectively extend convolutional receptive fields and guide CNNs to better respond to low frequencies through cascading, termed wavelet convolution. Each convolution focuses on different frequency bands of the input signal with gradually increasing effective ranges. This process enables greater emphasis on low-frequency components while adding only a small number of trainable parameters. This dynamic approach allows the model to flexibly focus on critical spatial structures when processing different regions, rather than relying on fixed receptive fields of single static kernels. The Wavelet Conv module enhances model representation capability by expanding receptive fields through 3D wavelet transforms without increasing network depth or width. Experimental results demonstrate superior performance on the IN, UP, and KSC datasets, outperforming mainstream hyperspectral image classification methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。