Diffusion-Model-Based Hyperspectral Unmixing Using Spectral Prior Distribution
Keli Deng,Yuntao Qian,Jie Nie,Jun Zhou
DOI: https://doi.org/10.1109/tgrs.2024.3408475
IF: 8.2
2024-06-15
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Hyperspectral unmixing is a crucial task for identifying the constituent materials and their respective distributions in a scene. Utilizing known spectral libraries as prior information, semiblind unmixing methods (also known as spectral-library-based methods) have been proven advantageous over unblind or blind unmixing approaches. However, such methods encounter two main challenges: difficulty in handling large-scale spectral libraries and vulnerability to variabilities stemming from differences between the underlying signatures and those in the spectral library. To address these challenges, a novel approach named diffusion-model-based unmixing method (DiffUn), based on a diffusion model (DM), is proposed in this article for semiblind hyperspectral unmixing. DiffUn considers hyperspectral unmixing as a sampling process from a posterior distribution, where the prior distribution is learned from a spectral library, and the likelihood distribution is estimated from the observed data by the linear spectral mixture model. Specifically, the approach first learns the spectral prior distribution from a spectral library through an unconditional DM, then integrates this prior knowledge into the reverse process of the DM, and finally samples the underlying endmembers and corresponding abundances from the posterior distribution. Since spectral prior distribution estimation is not sensitive to library size, DiffUn exhibits superior unmixing performance even in a large-scale library. Furthermore, DiffUn permits sampling spectral signatures from a continuous probabilistic distribution, whereas conventional semiblind unmixing methods only allow endmembers selected from the library, which is a discrete space. Thus, DiffUn shows greater robustness to spectral variations. The experimental results on synthetic and real-world datasets demonstrate DiffUn outperforming the state-of-the-art semiblind unmixing methods. The code is available at https://github.com/Dmsw/DiffUn.git.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics