arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.03398eess.IVcs.CVcs.LGcs.NE

使用多任务学习的深度非线性高光谱解混

Deep Nonlinear Hyperspectral Unmixing Using Multi-task Learning

University of Notre Dame(圣母大学)

Saeid Mehrdad, Seyed AmirHossein Janani

AI总结:

本文提出一种基于深度学习的无监督非线性高光谱解混方法,采用无假设的通用模型,通过双分支结构和多任务学习提升解混性能,实验验证其优于现有方法。

摘要:

非线性高光谱解混近来受到广泛关注,因为线性混合模型在某些问题中无法获得可接受的分辨率。实际上,大多数非线性解混方法都是基于对非线性模型的特定假设而设计的,这随后限制了解混性能。在本文中,我们提出了一种基于深度学习的无监督非线性解混方法,该方法融合了一个无特殊假设的通用非线性模型。该模型包含两个分支。在第一个分支中,通过使用若干隐藏层重构高光谱图像的行来学习端元,而在第二个分支中,基于相应图像的列来学习丰度值。然后,利用多任务学习,我们引入一个辅助任务以强制两个分支协同工作。该技术可被视为一种缓解过拟合的正则化器,从而提升整个网络的性能。在合成数据和真实数据上的大量实验验证了所提方法相较于一些最先进的高光谱解混方法的有效性。

英文摘要:

Nonlinear hyperspectral unmixing has recently received considerable attention, as linear mixture models do not lead to an acceptable resolution in some problems. In fact, most nonlinear unmixing methods are designed by assuming specific assumptions on the nonlinearity model which subsequently limits the unmixing performance. In this paper, we propose an unsupervised nonlinear unmixing approach based on deep learning by incorporating a general nonlinear model with no special assumptions. This model consists of two branches. In the first branch, endmembers are learned by reconstructing the rows of hyperspectral images using some hidden layers, and in the second branch, abundance values are learned based on the columns of respective images. Then, using multi-task learning, we introduce an auxiliary task to enforce the two branches to work together. This technique can be considered as a regularizer mitigating overfitting, which improves the performance of the total network. Extensive experiments on synthetic and real data verify the effectiveness of the proposed method compared to some state-of-the-art hyperspectral unmixing methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑