arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.01799cs.CV

面向高光谱图像的语义不变增强超像素图对比聚类

Superpixel Graph Contrastive Clustering with Semantic-Invariant Augmentations for Hyperspectral Images

School of Software Engineering, Southeast University(东南大学软件学院);School of Computer Science and Engineering, Southeast University(东南大学计算机科学与工程学院);Key Laboratory of New Generation Artificial Intelligence Technology and Its Interdisciplinary Applications (Southeast University), Ministry of Education, China(新一代人工智能技术及交叉应用关键实验室(东南大学));Department of Computing and Information Sciences, Saint Francis University(圣弗朗西斯大学计算机与信息科学系);Department of Computer Science, City University of Hong Kong(香港城市大学计算机科学系)

Jianhan Qi, Yuheng Jia, Hui Liu, Junhui Hou

AI总结:

本文提出超像素图对比聚类模型SPGCC,通过混合卷积预训练提取高光谱图像高阶空谱特征,并设计像素采样与模型权重两种语义不变增强,交替优化聚类与网络,在多个HSI数据集上优于现有方法。

摘要:

高光谱图像(HSI)聚类是一项重要但具有挑战性的任务。当前最先进(SOTA)的方法通常依赖超像素,然而它们没有充分利用HSI三维结构中的空间和光谱信息,并且其优化目标并非面向聚类。在本工作中,我们首先使用三维和二维混合卷积神经网络通过预训练提取HSI的高阶空间和光谱特征,然后设计了一个超像素图对比聚类(SPGCC)模型来学习具有判别力的超像素表示。合理的增强视图对于对比聚类至关重要,而传统对比学习可能会破坏聚类结构,因为不同样本在嵌入空间中会被推开,即使它们属于同一类别。在SPGCC中,我们为HSI超像素设计了两种语义不变的数据增强:像素采样增强和模型权重增强。然后进行样本级对齐和聚类中心级对比,以获得更好的超像素嵌入类内相似性和类间差异性。我们交替执行聚类和网络优化。在多个HSI数据集上的实验结果验证了所提出的SPGCC相比SOTA方法的优势。我们的代码可在 https://github.com/jhqi/spgcc 获取。

英文摘要:

Hyperspectral images (HSI) clustering is an important but challenging task. The state-of-the-art (SOTA) methods usually rely on superpixels, however, they do not fully utilize the spatial and spectral information in HSI 3-D structure, and their optimization targets are not clustering-oriented. In this work, we first use 3-D and 2-D hybrid convolutional neural networks to extract the high-order spatial and spectral features of HSI through pre-training, and then design a superpixel graph contrastive clustering (SPGCC) model to learn discriminative superpixel representations. Reasonable augmented views are crucial for contrastive clustering, and conventional contrastive learning may hurt the cluster structure since different samples are pushed away in the embedding space even if they belong to the same class. In SPGCC, we design two semantic-invariant data augmentations for HSI superpixels: pixel sampling augmentation and model weight augmentation. Then sample-level alignment and clustering-center-level contrast are performed for better intra-class similarity and inter-class dissimilarity of superpixel embeddings. We perform clustering and network optimization alternatively. Experimental results on several HSI datasets verify the advantages of the proposed SPGCC compared to SOTA methods. Our code is available at https://github.com/jhqi/spgcc.

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

↑