SPARSE UNMIXING USING DEEP CONVOLUTIONAL NETWORKS

Abstract

This paper proposes a sparse unmixing technique using a convolutional neural network (SUnCNN). We reformulate the sparse unmixing problem into an optimization over the parameters of a convolutional network. Relying on a spectral library, the deep network learns in an unsupervised manner a mapping from a fixed input to the sparse abundances. Moreover, SUnCNN fulfills the sum-to-one constraint using a softmax activation layer. We compare SUnCNN with the state-of-the-art using a simulated and a real dataset. The experimental results show that the proposed deep learning-based unmixing method outperforms the others in terms of signal to reconstruction error. Additionally, SUnCNN is visually superior to the competing techniques. SUnCNN was implemented in Python (3.8) using PyTorch as the platform for the deep network and is available online: https://github.com/BehnoodRasti/SUnCNN.