arXiv:2510.22607cs.CV2025-10

用小波网络自监督解混高光谱图像,无需真实标签也能准确分离物质成分。

SWAN: Self-supervised Wavelet Neural Network for Hyperspectral Image Unmixing

  • 将高光谱数据转为小波域,通过自监督学习提取多尺度特征。
  • 在合成与真实数据集上均优于现有方法,尤其在低信噪比下表现稳定。
  • 适合遥感、地质勘探等需无标签解混的工程场景。

本文提出SWAN:一种三阶段自监督小波神经网络,用于从高光谱图像中联合估计端元和丰度。连续重叠的高光谱波段首先被扩展到双正交小波基空间,实现稀疏、分布且多尺度的表示。利用由此获得的不变与协变特征中的潜在对称性,采用自监督学习范式。第一阶段SWANencoder将输入小波系数映射至低维紧凑隐空间;第二阶段SWANdecoder基于该隐表示重建输入小波系数;第三阶段SWANforward学习高光谱图像的底层物理机制。设计了三阶段联合损失函数,作用于图像采集域,无需真实标签即可实现自监督训练。使用Adam优化,引入Sigmoid激活与0.3丢弃率防止过拟合,核正则化器约束系数幅度并保留空间变化。推理时SWANencoder输出丰度图,SWANdecoder权重用于提取端元。在两个具有不同信噪比的基准合成数据集及三个真实高光谱数据集上进行实验,结果优于多种先进神经网络解混方法。定性、定量与消融分析表明,该方法提升了鲁棒解混能力,强化了自监督机制与紧凑网络参数,适用于实际应用。

原文摘要 · Abstract (English)

In this article, we present SWAN: a three-stage, self-supervised wavelet neural network for joint estimation of endmembers and abundances from hyperspectral imagery. The contiguous and overlapping hyperspectral band images are first expanded to Biorthogonal wavelet basis space that provides sparse, distributed, and multi-scale representations. The idea is to exploit latent symmetries from thus obtained invariant and covariant features using a self-supervised learning paradigm. The first stage, SWANencoder maps the input wavelet coefficients to a compact lower-dimensional latent space. The second stage, SWANdecoder uses the derived latent representation to reconstruct the input wavelet coefficients. Interestingly, the third stage SWANforward learns the underlying physics of the hyperspectral image. A three-stage combined loss function is formulated in the image acquisition domain that eliminates the need for ground truth and enables self-supervised training. Adam is employed for optimizing the proposed loss function, while Sigmoid with a dropout of 0.3 is incorporated to avoid possible overfitting. Kernel regularizers bound the magnitudes and preserve spatial variations in the estimated endmember coefficients. The output of SWANencoder represents estimated abundance maps during inference, while weights of SWANdecoder are retrieved to extract endmembers. Experiments are conducted on two benchmark synthetic data sets with different signal-to-noise ratios as well as on three real benchmark hyperspectral data sets while comparing the results with several state-of-the-art neural network-based unmixing methods. The qualitative, quantitative, and ablation results show performance enhancement by learning a resilient unmixing function as well as promoting self-supervision and compact network parameters for practical applications.

高光谱解混自监督学习小波网络遥感

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。