ADMM-Based Hyperspectral Unmixing Networks for Abundance and Endmember Estimation
in IEEE Transactions on Geoscience and Remote Sensing, vol. 60, pp.
1-18, 2022, Art no. 5520018, doi: 10.1109/TGRS.2021.3136336.
Hyperspectral image (HSI) unmixing is an increasingly studied problem in various areas, including
remote sensing. It has been tackled using both physical model-based approaches and more recently
machine learning-based ones. In this article, we propose a new HSI unmixing algorithm combining
both model- and learning-based techniques, based on algorithm unrolling approaches, delivering
improved unmixing performance. Our approach unrolls the alternating direction method of
multipliers (ADMMs) solver of a constrained sparse regression problem underlying a linear mixture
model. We then propose a neural network structure for abundance estimation that can be trained
using supervised learning techniques based on a new composite loss function. We also propose
another neural network structure for blind unmixing that can be trained using unsupervised
learning techniques. Our proposed networks are also shown to possess a lighter and richer
structure containing less learnable parameters and more skip connections compared with other
competing architectures. Extensive experiments show that the proposed methods can achieve much
faster convergence and better performance even with a very small training dataset size when
compared with other unmixing methods, such as model-inspired neural network for abundance
estimation (MNN-AE), model-inspired neural network for blind unmixing (MNN-BU), unmixing using
deep image prior (UnDIP), and endmember-guided unmixing network (EGU-Net).