用最优传输提升图像语义对应,速度更快效果不差
Gromov Wasserstein Optimal Transport for Semantic Correspondences
- 用格罗莫夫-瓦瑟斯坦最优传输替代近邻匹配,引入空间一致性先验
- 在DINOv2基础上性能显著提升,效率比扩散模型方法快5-10倍
- 适合需要高效高精度对应的应用,如配准、跨模态匹配
图像对间建立语义对应是计算机视觉的经典问题。近年来,基于大模型(如DINOv2和Stable Diffusion)的内部特征在零样本任务中表现优异,被用于语义对应。研究发现,DINOv2特征准确但稀疏,而Stable Diffusion特征空间一致但不够精确。当前最先进方法通过融合两者特征实现高性能,但计算开销巨大。本文提出新思路:用更优的匹配算法替代Stable Diffusion特征,即引入具有空间平滑性的格罗莫夫-瓦瑟斯坦最优传输(Gromov Wasserstein Optimal Transport),替代标准最近邻匹配。实验表明,该方法显著提升DINOv2基线性能,且在效率上比使用Stable Diffusion特征的方法快5–10倍,同时保持甚至超越现有先进水平。代码已开源。
原文摘要 · Abstract (English)
Establishing correspondences between image pairs is a long studied problem in computer vision. With recent large-scale foundation models showing strong zero-shot performance on downstream tasks including classification and segmentation, there has been interest in using the internal feature maps of these models for the semantic correspondence task. Recent works observe that features from DINOv2 and Stable Diffusion (SD) are complementary, the former producing accurate but sparse correspondences, while the latter produces spatially consistent correspondences. As a result, current state-of-the-art methods for semantic correspondence involve combining features from both models in an ensemble. While the performance of these methods is impressive, they are computationally expensive, requiring evaluating feature maps from large-scale foundation models. In this work we take a different approach, instead replacing SD features with a superior matching algorithm which is imbued with the desirable spatial consistency property. Specifically, we replace the standard nearest neighbours matching with an optimal transport algorithm that includes a Gromov Wasserstein spatial smoothness prior. We show that we can significantly boost the performance of the DINOv2 baseline, and be competitive and sometimes surpassing state-of-the-art methods using Stable Diffusion features, while being 5--10x more efficient. We make code available at https://github.com/fsnelgar/semantic_matching_gwot .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。