Diffusion Subspace Clustering for Hyperspectral Images

Code

• github.com/N1ceChen/DiffSC
  pytorch★ 0

Abstract

Hyperspectral image (HSI) subspace clustering remains a challenging task due to the poor spatial and rich spectral resolutions of HSIs. Most of the existing HSI subspace clustering approaches just extract the spatial and spectral features, ignoring the intrinsic distribution information of data and leading to low accuracy of clustering generally. To solve this problem, this article presents a diffusion subspace clustering model (DiffSC) that learns distribution information of HSI data simultaneously through a diffusion module (DM). Specifically, due to the diffusion probabilistic model (DPM) learning raw object data distribution to generate data of the same distribution, which has received wide attention in generation tasks and outperforms other generative models significantly, we attempt to apply the DPM in the field of feature extraction. DiffSC performs distribution information extraction of HSIs by the DM and fuses them with spatial-spectral features extracted by deep subspace clustering for training jointly. Experiment outcomes demonstrate that intermediate activation of specific timestep in the inverse diffusion process captures latent distribution information of images effectively and improves the HSI clustering accuracy significantly. Since the DPM is simplified, it can be easily trained from scratch. We evaluate the presented DiffSC on five real HSI datasets, and the experiments indicate that DiffSC can obtain the most advanced clustering outcomes that notably outperform most existing HSI subspace clustering approaches.

Tasks

Reproductions