arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2404.07112cs.CV

展开ADMM以增强高光谱图像的子空间聚类性能

Unfolding ADMM for Enhanced Subspace Clustering of Hyperspectral Images

Ghent University(根特大学);China University of Geosciences(中国地质大学)

Xianlu Li, Nicolas Nadisic, Shaoguang Huang, Aleksandra Pižurica

AI总结:

针对深度子空间聚类过拟合、可解释性差的问题,本文提出将ADMM迭代求解器展开的高光谱图像聚类架构,结合K近邻保留数据结构,在三个数据集上性能优于现有先进方法。

摘要:

深度子空间聚类方法目前在聚类领域应用广泛,通常采用全连接网络与自表示损失函数。但这类方法常存在过拟合问题,且可解释性不足。本文探索了一种基于深度展开的替代聚类方法。该方法通过将迭代优化方法展开为神经网络,与数据驱动的深度学习方法相比,可解释性与可靠性更强;与基于模型的方法相比,适应性与泛化能力更优。因此,展开技术已在图像复原、重建、超分辨率等成像逆问题中得到广泛应用,但在聚类场景下的研究仍不充分。本研究提出了一种创新的高光谱图像(HSI)聚类架构,将基于交替方向乘子法(ADMM)的迭代求解器展开,用于稀疏子空间聚类。据我们所知,这是首次尝试将展开ADMM应用于子空间聚类中的自表示矩阵计算。此外,该方法将K近邻算法作为结构保留模块的组成部分,能够很好地捕捉HSI数据的结构特征。在三个经典HSI数据集上的实验评估清晰表明,展开方法在HSI聚类中颇具潜力,其性能甚至优于当前最先进的技术。

英文摘要:

Deep subspace clustering methods are now prominent in clustering, typically using fully connected networks and a self-representation loss function. However, these methods often struggle with overfitting and lack interpretability. In this paper, we explore an alternative clustering approach based on deep unfolding. By unfolding iterative optimization methods into neural networks, this approach offers enhanced interpretability and reliability compared to data-driven deep learning methods, and greater adaptability and generalization than model-based approaches. Hence, unfolding has become widely used in inverse imaging problems, such as image restoration, reconstruction, and super-resolution, but has not been sufficiently explored yet in the context of clustering. In this work, we introduce an innovative clustering architecture for hyperspectral images (HSI) by unfolding an iterative solver based on the Alternating Direction Method of Multipliers (ADMM) for sparse subspace clustering. To our knowledge, this is the first attempt to apply unfolding ADMM for computing the self-representation matrix in subspace clustering. Moreover, our approach captures well the structural characteristics of HSI data by employing the K nearest neighbors algorithm as part of a structure preservation module. Experimental evaluation of three established HSI datasets shows clearly the potential of the unfolding approach in HSI clustering and even demonstrates superior performance compared to state-of-the-art techniques.

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

↑