Title
Deep Reinforcement Learning for Semisupervised Hyperspectral Band Selection
Abstract
Band selection is an important step in efficient processing of hyperspectral images (HSIs), which can be seen as the combination of powerful band search technique and effective evaluation criterion. The existing deep-learning-based methods make the network parameters sparse to search the spectral bands using threshold-based functions or regularization terms. These methods may lead to an intractable optimization problem. Furthermore, these methods need to repeatedly train deep networks for evaluating candidate band subsets. In this article, we formalize hyperspectral band selection as a reinforcement learning (RL) problem. Band search is regarded as a sequential decision-making process, where each state in the search space is a feasible band subset. To evaluate each state, a semisupervised convolutional neural network (CNN), called EvaluateNet, is constructed by adding the intraclass compactness constraint of both limited labeled and sufficient unlabeled samples. A simple stochastic band sampling method is designed to train EvaluateNet, making it possible to efficiently evaluate without any fine-tuning. In RL, new reward functions are defined by taking the EvaluateNet and the penalty of repeated selection into account. Finally, advantage actorx2013;critic algorithms are designed to explore in the state space and select the band subset according to the expected accumulated reward. The experimental results on HSI data sets demonstrate the effectiveness and efficiency of the proposed algorithms for hyperspectral band selection.
Year
DOI
Venue
2022
10.1109/TGRS.2021.3049372
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING
Keywords
DocType
Volume
Hyperspectral imaging, Reinforcement learning, Optimization, Deep learning, Task analysis, Neural networks, Convolutional neural networks, Actor-critic algorithm, band selection, deep reinforcement learning (DRL), hyperspectral image (HSI) classification, semisupervised learning
Journal
60
Issue
ISSN
Citations 
99
0196-2892
0
PageRank 
References 
Authors
0.34
0
7
Name
Order
Citations
PageRank
Jie Feng165.17
Di Li200.34
Gu Jing3377.42
Xianghai Cao4216.01
Ronghua Shang500.34
Xiangrong Zhang600.34
Licheng Jiao75698475.84