AI Navigate

WS-Net: Weak-Signal Representation Learning and Gated Abundance Reconstruction for Hyperspectral Unmixing via State-Space and Weak Signal Attention Fusion

arXiv cs.CV / 3/11/2026

Signals & Early TrendsModels & Research

Key Points

  • WS-Net is a novel deep unmixing framework designed to improve hyperspectral unmixing by addressing weak-signal collapse using state-space modelling and Weak Signal Attention fusion.
  • The architecture incorporates a multi-resolution wavelet-fused encoder and a hybrid backbone combining a Mamba state-space branch for modelling long-range dependencies with a Weak Signal Attention branch to enhance low-similarity spectral signals.
  • A learnable gating mechanism fuses the two representations adaptively, and the decoder uses KL-divergence-based regularisation to better separate dominant and weak endmembers.
  • Experimental results on synthetic and real datasets show WS-Net achieves significant improvements over six state-of-the-art baselines, reducing RMSE and SAD by up to 55% and 63% respectively, especially under low signal-to-noise ratio conditions.
  • WS-Net establishes a robust and computationally efficient benchmark for weak-signal hyperspectral unmixing, effectively preserving weak spectral features that are commonly lost due to noise and dominant signals.

Computer Science > Computer Vision and Pattern Recognition

arXiv:2603.09037 (cs)
[Submitted on 10 Mar 2026]

Title:WS-Net: Weak-Signal Representation Learning and Gated Abundance Reconstruction for Hyperspectral Unmixing via State-Space and Weak Signal Attention Fusion

View a PDF of the paper titled WS-Net: Weak-Signal Representation Learning and Gated Abundance Reconstruction for Hyperspectral Unmixing via State-Space and Weak Signal Attention Fusion, by Zekun Long and 4 other authors
View PDF HTML (experimental)
Abstract:Weak spectral responses in hyperspectral images are often obscured by dominant endmembers and sensor noise, resulting in inaccurate abundance estimation. This paper introduces WS-Net, a deep unmixing framework specifically designed to address weak-signal collapse through state-space modelling and Weak Signal Attention fusion. The network features a multi-resolution wavelet-fused encoder that captures both high-frequency discontinuities and smooth spectral variations with a hybrid backbone that integrates a Mamba state-space branch for efficient long-range dependency modelling. It also incorporates a Weak Signal Attention branch that selectively enhances low-similarity spectral cues. A learnable gating mechanism adaptively fuses both representations, while the decoder leverages KL-divergence-based regularisation to enforce separability between dominant and weak endmembers. Experiments on one simulated and two real datasets (synthetic dataset, Samson, and Apex) demonstrate consistent improvements over six state-of-the-art baselines, achieving up to 55% and 63% reductions in RMSE and SAD, respectively. The framework maintains stable accuracy under low-SNR conditions, particularly for weak endmembers, establishing WS-Net as a robust and computationally efficient benchmark for weak-signal hyperspectral unmixing.
Subjects: Computer Vision and Pattern Recognition (cs.CV); Artificial Intelligence (cs.AI)
Cite as: arXiv:2603.09037 [cs.CV]
  (or arXiv:2603.09037v1 [cs.CV] for this version)
  https://doi.org/10.48550/arXiv.2603.09037
Focus to learn more
arXiv-issued DOI via DataCite

Submission history

From: Ali Zia [view email]
[v1] Tue, 10 Mar 2026 00:12:33 UTC (12,604 KB)
Full-text links:

Access Paper:

    View a PDF of the paper titled WS-Net: Weak-Signal Representation Learning and Gated Abundance Reconstruction for Hyperspectral Unmixing via State-Space and Weak Signal Attention Fusion, by Zekun Long and 4 other authors
  • View PDF
  • HTML (experimental)
  • TeX Source
Current browse context:
cs.CV
< prev   |   next >
Change to browse by:

References & Citations

export BibTeX citation Loading...

BibTeX formatted citation

×
Data provided by:

Bookmark

BibSonomy logo Reddit logo
Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle
Bibliographic Explorer (What is the Explorer?)
Connected Papers Toggle
Connected Papers (What is Connected Papers?)
Litmaps Toggle
Litmaps (What is Litmaps?)
scite.ai Toggle
scite Smart Citations (What are Smart Citations?)
Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle
alphaXiv (What is alphaXiv?)
Links to Code Toggle
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub Toggle
DagsHub (What is DagsHub?)
GotitPub Toggle
Gotit.pub (What is GotitPub?)
Huggingface Toggle
Hugging Face (What is Huggingface?)
Links to Code Toggle
Papers with Code (What is Papers with Code?)
ScienceCast Toggle
ScienceCast (What is ScienceCast?)
Demos

Demos

Replicate Toggle
Replicate (What is Replicate?)
Spaces Toggle
Hugging Face Spaces (What is Spaces?)
Spaces Toggle
TXYZ.AI (What is TXYZ.AI?)
Related Papers

Recommenders and Search Tools

Link to Influence Flower
Influence Flower (What are Influence Flowers?)
Core recommender toggle
CORE Recommender (What is CORE?)
About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.