hk-code / gc-ressenet

Graph-Constrained Residual Self-Expressive Subspace Clustering Network for Hyperspectral Images

Kun Huang, Xin Li, Yingdong Pi, Hao Cheng, Guowei Xu

Abstract

Hyperspectral images are gradually being used in various industries because of their rich spectral information. Meanwhile, the difficult acquisition of data labels makes unsupervised classification attracts attention. Subspace clustering as an unsupervised classification method is widely used for hyperspectral image analysis because of its excellent performance and robustness. However, conventional subspace clustering does not consider the nonlinear structure of hyperspectral data, and deep subspace clustering tends to ignore the intrinsic structure of hyperspectral data. To address these problems, we developed a self-expressive learning network, ResSENet, for hyperspectral data; we then proposed the application of ResSENet under graph constraints (GC-ResSENet), considering the intrinsic graph structure of the data. Unlike conventional deep subspace clustering, our model discards the self-expressive layer; self-expressive coefficients between datasets are directly solved by the data using network parameters. Hyperparameters are used in the joint loss to balance the self-expressive properties of the data and the graph constraint terms. We evaluated GC-ResSENet by applying it to four well-known datasets, and our network achieved optimal performance. Additionally, because of its abandonment of the self-expressive layer, ResSENet is theoretically capable of clustering with large datasets; thus, we evaluated it using two large datasets. The source code is accessible at https://github.com/HK-code/GC-ResSENet.

Architecture

Dataset

SalinasA, Indian Pines and pavia university datasets you can download from here, and Houston2013 dataset you can download from here.

Requirements

matplotlib==3.5.2
munkres==1.1.4
numpy==1.24.4
numpy==1.22.4
scikit_learn==1.2.2
scipy==1.9.1
torch==1.11.0+cu113
torchsummary==1.5.1
tqdm==4.61.2

Result

Training & Evaluation

Train

python main.py

Evaluation

python loadmodel.py

More

If you have any questions and needs, you can contact me, my email is: [email protected].

Cite

@ARTICLE{10319068,
  author={Huang, Kun and Li, Xin and Pi, Yingdong and Cheng, Hao and Xu, Guowei},
  journal={IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing}, 
  title={Graph-Constrained Residual Self-Expressive Subspace Clustering Network for Hyperspectral Images}, 
  year={2024},
  volume={17},
  number={},
  pages={941-955},
  doi={10.1109/JSTARS.2023.3333281}}