An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images

Band selection (BS) is an efficacious approach to reduce hyperspectral information redundancy while preserving the physical meaning of hyperspectral images (HSIs). Recently, deep learning-based BS methods have received widespread interest due to their ability to model the nonlinear relationship betw...

Full description

Bibliographic Details
Main Authors: Xiaorun Li, Yufei Liu, Ziqiang Hua, Shuhan Chen
Format: Article
Language:English
Published: MDPI AG 2023-11-01
Series:Remote Sensing
Subjects:
Online Access:https://www.mdpi.com/2072-4292/15/23/5495
_version_ 1797399648969687040
author Xiaorun Li
Yufei Liu
Ziqiang Hua
Shuhan Chen
author_facet Xiaorun Li
Yufei Liu
Ziqiang Hua
Shuhan Chen
author_sort Xiaorun Li
collection DOAJ
description Band selection (BS) is an efficacious approach to reduce hyperspectral information redundancy while preserving the physical meaning of hyperspectral images (HSIs). Recently, deep learning-based BS methods have received widespread interest due to their ability to model the nonlinear relationship between bands, with existing methods typically relying on generative algorithms. However, the process of generating images with pixel-level detail required by generative algorithm-based BS methods is computationally expensive. To alleviate this issue, we propose a contrastive learning-based unsupervised BS architecture, termed ContrastBS, in this article. With the help of contrastive learning, the proposed architecture avoids the costly generation step in pixel space by learning to distinguish data at the abstract semantic level of the feature space. Specifically, ContrastBS combines an attention mechanism with contrastive learning to extract the importance of each band. Furthermore, we design a novel loss function, which is able to constrain the symmetric loss while ensuring attention to the most valuable bands, for the contrastive learning-based BS network. Experimental results indicate that ContrastBS has excellent classification performance and competitive time cost compared to the comparison methods.
first_indexed 2024-03-09T01:44:10Z
format Article
id doaj.art-90214972245149cb83b6bf04327ae7b1
institution Directory Open Access Journal
issn 2072-4292
language English
last_indexed 2024-03-09T01:44:10Z
publishDate 2023-11-01
publisher MDPI AG
record_format Article
series Remote Sensing
spelling doaj.art-90214972245149cb83b6bf04327ae7b12023-12-08T15:24:48ZengMDPI AGRemote Sensing2072-42922023-11-011523549510.3390/rs15235495An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral ImagesXiaorun Li0Yufei Liu1Ziqiang Hua2Shuhan Chen3College of Electrical Engineering, Zhejiang University, Hangzhou 310027, ChinaCollege of Electrical Engineering, Zhejiang University, Hangzhou 310027, ChinaCollege of Electrical Engineering, Zhejiang University, Hangzhou 310027, ChinaCollege of Electrical Engineering, Zhejiang University, Hangzhou 310027, ChinaBand selection (BS) is an efficacious approach to reduce hyperspectral information redundancy while preserving the physical meaning of hyperspectral images (HSIs). Recently, deep learning-based BS methods have received widespread interest due to their ability to model the nonlinear relationship between bands, with existing methods typically relying on generative algorithms. However, the process of generating images with pixel-level detail required by generative algorithm-based BS methods is computationally expensive. To alleviate this issue, we propose a contrastive learning-based unsupervised BS architecture, termed ContrastBS, in this article. With the help of contrastive learning, the proposed architecture avoids the costly generation step in pixel space by learning to distinguish data at the abstract semantic level of the feature space. Specifically, ContrastBS combines an attention mechanism with contrastive learning to extract the importance of each band. Furthermore, we design a novel loss function, which is able to constrain the symmetric loss while ensuring attention to the most valuable bands, for the contrastive learning-based BS network. Experimental results indicate that ContrastBS has excellent classification performance and competitive time cost compared to the comparison methods.https://www.mdpi.com/2072-4292/15/23/5495unsupervised band selectioncontrastive learninghyperspectral imageattention mechanism
spellingShingle Xiaorun Li
Yufei Liu
Ziqiang Hua
Shuhan Chen
An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
Remote Sensing
unsupervised band selection
contrastive learning
hyperspectral image
attention mechanism
title An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
title_full An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
title_fullStr An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
title_full_unstemmed An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
title_short An Unsupervised Band Selection Method via Contrastive Learning for Hyperspectral Images
title_sort unsupervised band selection method via contrastive learning for hyperspectral images
topic unsupervised band selection
contrastive learning
hyperspectral image
attention mechanism
url https://www.mdpi.com/2072-4292/15/23/5495
work_keys_str_mv AT xiaorunli anunsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT yufeiliu anunsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT ziqianghua anunsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT shuhanchen anunsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT xiaorunli unsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT yufeiliu unsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT ziqianghua unsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages
AT shuhanchen unsupervisedbandselectionmethodviacontrastivelearningforhyperspectralimages