Transformer based Endmember Fusion with Spatial Context for Hyperspectral Unmixing
Abstract
In recent years, transformer-based deep learning networks have gained popularity in Hyperspectral (HS) unmixing applications due to their superior performance. The attention mechanism within transformers facilitates input-dependent weighting and enhances contextual awareness during training. Drawing inspiration from this, we propose a novel attention-based Hyperspectral Unmixing algorithm called Transformer-based Endmember Fusion with Spatial Context for Hyperspectral Unmixing (FusionNet). This network leverages an ensemble of endmembers for initial guidance, effectively addressing the issue of relying on a single initialization. This approach helps avoid suboptimal results that many algorithms encounter due to their dependence on a singular starting point. The FusionNet incorporates a Pixel Contextualizer (PC), introducing contextual awareness into abundance prediction by considering neighborhood pixels. Unlike Convolutional Neural Networks (CNNs) and traditional Transformer-based approaches, which are constrained by specific kernel or window shapes, the Fusion network offers flexibility in choosing any arbitrary configuration of the neighborhood. We conducted a comparative analysis between the FusionNet algorithm and eight state-of-the-art algorithms using three widely recognized real datasets and one synthetic dataset. The results demonstrate that FusionNet offers competitive performance compared to the other algorithms.
Cite
@article{arxiv.2402.03835,
title = {Transformer based Endmember Fusion with Spatial Context for Hyperspectral Unmixing},
author = {R. M. K. L. Ratnayake and D. M. U. P. Sumanasekara and H. M. K. D. Wickramathilaka and G. M. R. I. Godaliyadda and M. P. B. Ekanayake and H. M. V. R. Herath},
journal= {arXiv preprint arXiv:2402.03835},
year = {2024}
}