TY  - JOUR
PB  - Institute of Electrical and Electronics Engineers
N1  - This version is the author accepted manuscript. For information on re-use, please refer to the publisher?s terms and conditions.
TI  - ADMM-Based Hyperspectral Unmixing Networks for Abundance and Endmember Estimation
Y1  - 2021/12/16/
A1  - Zhou, C
A1  - Rodrigues, MRD
VL  - 60
UR  - https://doi.org/10.1109/TGRS.2021.3136336
N2  - 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).
AV  - public
JF  - IEEE Transactions on Geoscience and Remote Sensing
ID  - discovery10140134
KW  - Estimation

KW  - 
Neural networks

KW  - 
Training

KW  - 
Optimization

KW  - 
Reflectivity

KW  - 
Network architecture

KW  - 
Convergence
ER  -