Solving Spectrum Unmixing as a Multi-Task Bayesian Inverse Problem with Latent Factors for Endmember Variability

Authors

  • Dong Wu Fudan University
  • Mingmin Chi Fudan University Zhongshan PoolNet Technology Co., Ltd
  • Xuan Zang Fudan University
  • Bo Peng Shanghai Ocean University

DOI:

https://doi.org/10.1609/aaai.v38i14.29518

Keywords:

ML: Classification and Regression, APP: Natural Sciences, ML: Bayesian Learning, ML: Deep Learning Algorithms

Abstract

With the increasing customization of spectrometers, spectral unmixing has become a widely used technique in fields such as remote sensing, textiles, and environmental protection. However, endmember variability is a common issue for unmixing, where changes in lighting, atmospheric, temporal conditions, or the intrinsic spectral characteristics of materials, can all result in variations in the measured spectrum. Recent studies have employed deep neural networks to tackle endmember variability. However, these approaches rely on generic networks to implicitly resolve the issue, which struggles with the ill-posed nature and lack of effective convergence constraints for endmember variability. This paper proposes a streamlined multi-task learning model to rectify this problem, incorporating abundance regression and multi-label classification with Unmixing as a Bayesian Inverse Problem, denoted as BIPU. To address the issue of the ill-posed nature, the uncertainty of unmixing is quantified and minimized through the Laplace approximation in a Bayesian inverse solver. In addition, to improve convergence under the influence of endmember variability, the paper introduces two types of constraints. The first separates background factors of variants from the initial factors for each endmember, while the second identifies and eliminates the influence of non-existent endmembers via multi-label classification during convergence. The effectiveness of this model is demonstrated not only on a self-collected near-infrared spectral textile dataset (FENIR), but also on three commonly used remote sensing hyperspectral image datasets, where it achieves state-of-the-art unmixing performance and exhibits strong generalization capabilities.

Published

2024-03-24

How to Cite

Wu, D., Chi, M., Zang, X., & Peng, B. (2024). Solving Spectrum Unmixing as a Multi-Task Bayesian Inverse Problem with Latent Factors for Endmember Variability. Proceedings of the AAAI Conference on Artificial Intelligence, 38(14), 15878–15887. https://doi.org/10.1609/aaai.v38i14.29518

Issue

Section

AAAI Technical Track on Machine Learning V