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

学习神经特征函数用于无监督语义分割

Learning Neural Eigenfunctions for Unsupervised Semantic Segmentation

Shanghai Jiao Tong University(上海交通大学);Max Planck Institute for Intelligent Systems(马克斯·普朗克智能系统研究所)

Zhijie Deng, Yucen Luo

AI总结:

本工作提出一种基于神经网络特征函数的参数化谱聚类方法,通过输出离散向量直接进行聚类分配,实现了端到端的无监督语义分割,在多个基准上取得显著性能提升。

摘要:

无监督语义分割是计算机视觉中具有重大意义的长期挑战。谱聚类是一种有理论依据的解决方案,它通过计算像素的谱嵌入来构建不同的簇。尽管近期在利用强大的预训练模型增强谱聚类方面取得了进展,但当前方法在谱分解上仍存在低效性,且在应用于测试数据时缺乏灵活性。本工作通过将谱聚类转化为一种参数化方法来解决这些问题,该方法采用基于神经网络的特征函数来生成谱嵌入。神经特征函数的输出被进一步限制为直接指示聚类分配的离散向量。因此,形成了一种端到端的基于神经网络的谱聚类范式。在实践中,神经特征函数是轻量级的,并将预训练模型的特征作为输入,提高了训练效率并释放了预训练模型在密集预测中的潜力。我们进行了广泛的实证研究来验证我们方法的有效性,并在Pascal Context、Cityscapes和ADE20K基准上观察到相对于竞争基线的显著性能提升。

英文摘要:

Unsupervised semantic segmentation is a long-standing challenge in computer vision with great significance. Spectral clustering is a theoretically grounded solution to it where the spectral embeddings for pixels are computed to construct distinct clusters. Despite recent progress in enhancing spectral clustering with powerful pre-trained models, current approaches still suffer from inefficiencies in spectral decomposition and inflexibility in applying them to the test data. This work addresses these issues by casting spectral clustering as a parametric approach that employs neural network-based eigenfunctions to produce spectral embeddings. The outputs of the neural eigenfunctions are further restricted to discrete vectors that indicate clustering assignments directly. As a result, an end-to-end NN-based paradigm of spectral clustering emerges. In practice, the neural eigenfunctions are lightweight and take the features from pre-trained models as inputs, improving training efficiency and unleashing the potential of pre-trained models for dense prediction. We conduct extensive empirical studies to validate the effectiveness of our approach and observe significant performance gains over competitive baselines on Pascal Context, Cityscapes, and ADE20K benchmarks.

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

↑