Heng Wang 0009

dblp:61/5618-9 · DBLP profile ↗
← Back
5ranked-venue papers
1as first author
5since 2021 · last 2025
0000-0002-5678-7186ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Applied, interdisciplinary, general and emerging computing · 5 · 1 first-author · 5 since 2021
YearPublicationVenuePosition
2025 Masked Vision Transformer for Fast Hyperspectral Image Classification
abstract
Vision Transformer (ViT) has been thoroughly explored in hyperspectral image (HSI) classification (HIC). Nevertheless, current ViT-based approaches still acquire discriminative features, resulting in relatively limited generalization capabilities when confronted with the challenges posed by high intraclass variances and interclass similarities commonly observed in HSI data. Moreover, most of these methods fail to adequately emphasize the significance of the central pixel in HIC. To address the aforementioned challenges, we introduce a masked ViT (MViT) for HIC. First and foremost, MViT endeavors for the first time to introduce the masking operations in supervised models to learn more robust patterned features instead of distinguishable features, thereby bestowing it with outstanding generalization performance. Secondly, during the training phase of MViT, when conducting the random masking operations on the embedded features, we deliberately retain the embedding corresponding to the central pixel to guarantee the effectiveness of the model and emphasize the importance of the central pixel in HIC. Finally, MViT will deactivate the masking operations during the testing phase and utilize all the embedded features to accomplish the classification task, thereby enabling the model to fully exploit its recognition capabilities. On top of that, MViT is an extremely lightweight model, and by introducing the masking operations during the training phase, its training speed becomes unprecedentedly rapid. Experiments conducted on four publicly accessible datasets demonstrate that MViT can consistently achieve excellent or even the optimal classification results in comparison with the most advanced methods. The source code was powered by Jupyter and released at https://github.com/swiftest/MViT.
Liguo Wang 0001, Heng Wang 0009, Shoulin Yin, Lifeng Wang 0005
IEEE Trans. Geosci. Remote. Sens.2
2024 Regularized Masked Auto-Encoder for Semi-Supervised Hyperspectral Image Classification
abstract
As the most prevalent self-supervised representation learning (SSRL) model, the masked auto-encoder (MAE) has been gradually investigated in semi-supervised hyperspectral image classification (SHIC). However, the majority of the current approaches augment MAE merely from the application perspective or by introducing a weak regularization term, and do not comprehensively consider the challenges posed by the high intraclass variances and interclass similarities that often appear in hyperspectral image (HSI) data. In this article, we present a regularized MAE (RMAE) to address the aforementioned problems. Specifically, within the framework of MAE, we introduce a self-designed induced transformer block, using a small number of visible patches to learn the embeddings of patches with larger receptive fields. The learned embeddings are used to reconstruct the corresponding patches and an induced reconstruction loss is calculated. This strategy creates a much harder task for masked image modeling (MIM), and the induced transformer block is lightweight and imposes negligible computational burden overhead the underlying MAE framework. In addition, by rethinking the masking operations, we develop a masked convolutional neural network (MCNN), uncovering the principle of MAE and affirming the efficacy of RMAE. Finally, we present two metrics: the mean intraclass distance, and the mean interclass distance. Based on the metrics we give two criteria to evaluate the performance of an SSRL model, providing a new coordinate for the research in SSRL-based SHIC. Experiments conducted on four publicly accessible datasets show that RMAE outperforms state-of-the-art methods. The source code was powered by Jupyter and released athttps://github.com/swiftest/RMAE.
Liguo Wang 0001, Heng Wang 0009, Peng Wang 0030, Lifeng Wang 0005
IEEE Trans. Geosci. Remote. Sens.2
2023 Collaborative Active Learning Based on Improved Capsule Networks for Hyperspectral Image Classification
abstract
For hyperspectral image classification (HIC) tasks, most uncertainty-based active learning (AL) methods only consider the uncertainty, without considering the diversity of actively selected samples and the budget of expert labeling. In this paper, we propose a collaborative active learning (CAL) framework to address this problem. The proposed framework consists of two well-designed base classifiers and an ingenious CAL scheme that takes into account both the uncertainty and diversity of actively selected samples and the cost of expert annotation. Specifically, get benefit from the capsule networks’ ability to accurately identify and locate features, we design two improved capsule networks. For these two networks, we call the first CapsViT (Capsule Vision Transformer), which introduces Vision Transformer (ViT) into the capsule network (CapsNet) to learn the global relationship between the capsule features. We call the second CapsGLOM (Capsule GLOM), the basic structure of this network is derived from the GLOM system proposed by Geoffrey Hinton, we learn from the way CapsNet constructs the primary capsules to improve its implementation details. CapsViT and CapsGLOM are used as the two base classifiers in the proposed CAL framework to select the most informative samples according to the CAL scheme under the premise of fully considering the cost of expert annotation. Experimental results on four benchmark hyperspectral image data sets show that our proposed CAL framework can achieve satisfactory classification results. At the same time, compared with other advanced deep models, our proposed CapsViT and CapsGLOM are also competitive in the supervised HIC tasks. The source code can be available online (https://github.com/swiftest/CAL).
Heng Wang 0009, Liguo Wang 0001
IEEE Trans. Geosci. Remote. Sens.1
2022 SPCNet: A Subpixel Convolution-Based Change Detection Network for Hyperspectral Images With Different Spatial Resolutions
abstract
The very high spectral resolution in hyperspectral images (HSIs) offers an opportunity to detect subtle land-cover changes. However, the availability of HSIs acquired from different platforms requires the development of change detection (CD) methods capable of processing HSIs with different spatial resolutions. In this paper, we propose a general end-to-end subpixel convolution-based residual network (SPCNet) to accomplish the CD task between high spatial resolution (HR) and low spatial resolution (LR) HSIs. To effectively tackle the resolution matching issue, a super resolution (SR) block with an efficient subpixel convolution layer is introduced to upscale the LR feature maps into HR maps. The subpixel convolution layer can fully explore the subpixel context information by learning an array of upscaling filters. Moreover, the designed SPC module is embedded into the LR branch to generate more discriminative representations. More importantly, the SPC module as a plug-and-play unit has the potential to be embedded into other baseline networks to enhance the feature learning capability. Experimental results on four HSI datasets demonstrate the effectiveness of the proposed SPCNet.
Lifeng Wang 0005, Liguo Wang 0001, Heng Wang 0009, Xiaoyi Wang 0004, Lorenzo Bruzzone
IEEE Trans. Geosci. Remote. Sens.3
2022 RSSGL: Statistical Loss Regularized 3-D ConvLSTM for Hyperspectral Image Classification
abstract
Researches on the classification of hyperspectral images (HSIs) based on deep learning are in full swing, especially the spectral-spatial dependent global learning (SSDGL) framework, which is both efficient and robust. However, the global convolutional long short-term memory (GCL) module under this framework fails to take full consideration of the spectral characteristics contained in HSIs, and the hierarchically balanced (H-B) sampling strategy introduced in this framework prevents the training process from converging smoothly. In this article, we develop a novel regularized spectral-spatial global learning (RSSGL) framework. Compared with SSDGL, the proposed framework mainly makes three improvements. Above all, aiming at the problem that the GCL module used in SSDGL cannot fully tap the local spectral dependence, we apply 3D convolution to the gated units of long short-term memory (LSTM) as an alternative to the GCL module for adjacent and non-adjacent spectral dependencies learning. Furthermore, to extract the most discriminative features, an improved statistical loss regularization term is developed, in which we introduce a simple but effective diversity-promoting condition to make it more reasonable and suitable for deep metric learning in HSI classification. Finally, to effectively address the performance oscillation caused by the H-B sampling strategy, the proposed framework adopts an early stopping strategy to save and restore the optimal model parameters, making it more flexible and stable. Experiments conducted on three representative data sets show that the proposed RSSGL has superior classification performance compared with the existing relatively excellent research methods. The source code is released at https://github.com/swiftest/RSSGL.
Liguo Wang 0001, Heng Wang 0009, Lifeng Wang 0005, Xiaoyi Wang 0004, Yao Shi 0003
IEEE Trans. Geosci. Remote. Sens.2