Novel Spectral Loss Function for Unsupervised Hyperspectral Image Segmentation

Abstract

Neural networks (NNs) have gained importance in hyperspectral image (HSI) segmentation for earth observation due to its unparalleled data-driven feature extraction capability. However, in many real-life situations, ground truth is not available, and the performance of unsupervised NNs is still susceptible to enhancement. To overcome this challenge, this letter presents a new loss function to improve the performance of unsupervised HSI segmentation models. The spectral loss function, <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">Sl</i> , which can be included in different models, is based on the purity of the unmixing endmembers and the spectral similarity of the clusters provided by the NN to determine the classes. It is incorporated into a 3D convolutional autoencoder to validate its performance on four standard HSI benchmarks. Furthermore, its performance has been qualitatively examined in a real case study, an oil spill without ground truth. The results show that <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">Sl</i> is a breakthrough in unsupervised HS segmentation, obtaining the best overall performance and highlighting the importance of spectral signatures. Additionally, dimensional reduction is also vital in compacting the spectral information, which facilitates its segmentation. Source code available at [https://github.com/mhaut/HSI-3DSpLoss].