Transformer based Endmember Fusion with Spatial Context for Hyperspectral Unmixing

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ratnayake, R. M. K. L., Sumanasekara, D. M. U. P., Wickramathilaka, H. M. K. D., Godaliyadda, G. M. R. I., Ekanayake, M. P. B., Herath, H. M. V. R.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909275141439488
author Ratnayake, R. M. K. L.
Sumanasekara, D. M. U. P.
Wickramathilaka, H. M. K. D.
Godaliyadda, G. M. R. I.
Ekanayake, M. P. B.
Herath, H. M. V. R.
author_facet Ratnayake, R. M. K. L.
Sumanasekara, D. M. U. P.
Wickramathilaka, H. M. K. D.
Godaliyadda, G. M. R. I.
Ekanayake, M. P. B.
Herath, H. M. V. R.
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_03835
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Transformer based Endmember Fusion with Spatial Context for Hyperspectral Unmixing
Ratnayake, R. M. K. L.
Sumanasekara, D. M. U. P.
Wickramathilaka, H. M. K. D.
Godaliyadda, G. M. R. I.
Ekanayake, M. P. B.
Herath, H. M. V. R.
Image and Video Processing
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.
title Transformer based Endmember Fusion with Spatial Context for Hyperspectral Unmixing
topic Image and Video Processing
url https://arxiv.org/abs/2402.03835