跳转至

ConGeo: Robust Cross-View Geo-Localization Across Ground View Variations

会议: ECCV2024
arXiv: 2403.13965
代码: eceo-epfl/ConGeo
领域: remote_sensing
关键词: cross-view geo-localization, contrastive learning, orientation invariance, field of view, image retrieval

一句话总结

提出 ConGeo,一种模型无关的单视图+跨视图对比学习框架,通过强制同一地点不同地面视角变体之间的特征一致性,使单一模型即可在任意朝向和任意视场角(FoV)下实现鲁棒的跨视图地理定位。

背景与动机

跨视图地理定位(Cross-View Geo-Localization, CVGL)的目标是将地面图像与地理标记的航拍图像匹配以确定位置。现实场景中,用户拍摄的地面图像朝向不定、视场角各异(如手机或车载相机只有 70°–180°),但现有方法存在严重局限:

  1. 朝向/FoV 特定训练:现有方法如 DSM、SAIG-D 需要为每种朝向配置和 FoV 分别训练独立模型,无法泛化到未见过的视角变化
  2. 过度依赖空间对应关系:北对齐训练数据中的道路方向等几何特征成为模型的"捷径"(shortcut),一旦朝向改变,模型性能骤降(如 Sample4Geo 在未知朝向下 R@1 从 98.7% 暴跌至 16.3%)
  3. FoV 先验要求:部分方法(如 DSM)在训练和测试时都需要已知 FoV 信息,这在真实场景中往往不可获得

核心问题

如何用单一模型在不同朝向和不同 FoV 条件下都保持鲁棒的跨视图检索性能?核心挑战在于让模型学习到朝向不变(orientation-invariant)且对 FoV 变化有韧性(FoV-resilient)的特征表示。

方法详解

整体架构

ConGeo 基于经典的 Siamese 网络架构(地面编码器 + 航拍编码器),在此基础上引入三路输入:原始北对齐地面图像 \(I_q\)、变换后的地面图像 \(I_q^*\)(随机朝向偏移 + FoV 裁剪)以及航拍参考图像 \(I_r\)。ConGeo 是模型无关的学习目标,可插入任意 CVGL 基线模型。

地面视图变换

对北对齐全景图施加变换 \(T\):先用随机角度 \(\theta\) 进行水平循环偏移模拟未知朝向,再用 FoV 角度 \(\alpha\) 裁剪模拟有限视场:\(I_q^* = T_q(I_q | \theta, \alpha)\)。训练时 \(\theta \in [0°, 360°)\)\(\alpha = 180°\)

单视图对比学习 (Single-view Contrastive Learning)

  • 地面视图对比损失 \(\mathcal{L}_{\text{single-q}}\):对变换后的地面特征 \(q^*\) 与原始地面特征集合 \(Q\) 施加 InfoNCE 损失,强制同一位置不同视角变体在特征空间中靠近
  • 航拍视图对比损失 \(\mathcal{L}_{\text{single-r}}\):对同一航拍图像的两种随机数据增强版本施加对比损失,增强航拍特征的鲁棒性

跨视图对比学习 (Cross-view Contrastive Learning)

  • 基线损失 \(\mathcal{L}_{\text{vanilla}}\):原始地面图像与对应航拍图像的跨视图对齐损失(沿用基线方法,如 Sample4Geo 的 InfoNCE)
  • 跨视图对比损失 \(\mathcal{L}_{\text{cross}}\):变换后的地面图像 \(q^*\) 与航拍参考 \(R\) 之间的对比对齐,打破几何捷径,迫使模型关注语义一致特征

总损失

\[\mathcal{L} = \mathcal{L}_{\text{vanilla}} + w_1 \mathcal{L}_{\text{single-q}} + w_2 \mathcal{L}_{\text{single-r}} + w_3 \mathcal{L}_{\text{cross}}\]

其中 \(w_1 = 0.5\)\(w_2 = 0.5\)\(w_3 = 0.25\),单视图对比权重高于跨视图对比。所有温度参数均可学习。

实验关键数据

主实验:CVUSA 数据集(单模型,无 FoV 特定训练)

方法 FoV=360° R@1 FoV=180° R@1 FoV=90° R@1 FoV=70° R@1 平均 R@1
Sample4Geo(基线) 16.3% 4.1% 2.5% 1.5% 6.1%
Sample4Geo + DA 93.2% 84.6% 45.1% 28.4% 62.8%
ConGeo 85.2% 92.3% 55.9% 37.1% 67.6%

FoV 特定训练时更强

方法 FoV=360° R@1 FoV=180° R@1 FoV=90° R@1 FoV=70° R@1
SAIG-D 72.0% 52.5% 26.7% 20.9%
Sample4Geo 93.3% 84.6% 55.1% 40.9%
ConGeo 96.6% 92.3% 55.5% 49.1%

北对齐设定保持竞争力

ConGeo 在 CVUSA 上 R@1=98.3%(vs Sample4Geo 98.7%),在 CVACT Test 上 R@1=71.7%(vs Sample4Geo 71.5%),性能几乎无损。

跨区域鲁棒性(VIGOR Cross-Area)

方法 FoV=360° R@1 FoV=90° R@1
Sample4Geo 9.0% 0.5%
ConGeo 16.2% 3.9%

未见过的变换泛化(CVUSA)

ConGeo 在 Random Zooming (68.7% vs 基线 48.2%)、Gaussian Noise (45.8% vs DA 的 0.2%) 等未见变换上大幅超越数据增强方法,表明对比学习带来的不变性具有可迁移性。

消融实验核心发现

  • 地面视图对比损失(\(\mathcal{L}_{\text{single-q}}\))是最关键组件,显著提升各 FoV 表现
  • 跨视图对比损失(\(\mathcal{L}_{\text{cross}}\))在单视图损失基础上进一步提升 FoV=90° 表现约 35%
  • 纯数据增强在北对齐设定上性能大幅下降(98.7%→88.9%),ConGeo 仅降至 98.3%

亮点

  1. 单模型通吃所有设定:无需为每种朝向和 FoV 训练独立模型,一个 ConGeo 模型即可处理从 70° 到 360° 的全部场景
  2. 模型无关的即插即用设计:成功应用于 CNN(Sample4Geo)和 ViT(TransGeo)两种架构以及混合架构(SAIG-D),均带来显著提升
  3. 深入的可解释性分析:通过朝向不变性曲线和 Grad-CAM 激活图清晰展示了 ConGeo 如何从依赖空间捷径转向利用语义一致特征(如从关注道路方向转向关注树木等语义对象)
  4. 对比学习 > 数据增强:系统性证明了对比目标优于简单数据增强,尤其在未见变换上泛化能力远超 DA
  5. 训练高效:仅需单张 RTX 4090,60 个 epoch 即可完成训练

局限性 / 可改进方向

  1. 北对齐性能小幅下降:ConGeo 在北对齐设定下 R@1 从 98.7% 降至 98.3%,这是打破空间捷径的代价,但在实际应用中朝向通常未知
  2. 训练 FoV 选择敏感:训练时固定 \(\alpha = 180°\),虽然在不同 FoV 上均表现良好,但未探索自适应 FoV 采样策略
  3. 仅探索对比学习范式:论文提到冗余缩减(redundancy reduction)等其他模态对齐方式也值得探索
  4. 极端低 FoV 仍有挑战:FoV=70° 时性能虽大幅提升但绝对值仍不高(37.1%),语义信息极少时的定位仍是开放问题
  5. 未考虑时序信息:真实场景可利用连续帧提供额外约束

与相关工作的对比

维度 DSM SAIG-D Sample4Geo ConGeo
是否需要 FoV 先验
单模型多 FoV
未知朝向 R@1 (CVUSA) 78.1% 72.0% 93.3% 96.6%
FoV=90° R@1 (CVUSA) 16.2% 26.7% 55.1% 55.9%
未见变换泛化

ConGeo 的核心优势在于将对比学习从传统的跨视图对齐扩展到同一视图内不同变体之间的对齐,这与 SimCLR/BYOL 等自监督方法的思想一脉相承,但专门针对地理定位场景设计。

启发与关联

  1. 对比学习在跨模态检索中的新应用模式:不仅对齐两个模态,还对齐同一模态内的变体,这种范式可推广到其他跨模态任务(如文本-图像检索中对同一查询的改述变体做对比)
  2. "打破捷径"的通用思路:模型依赖训练数据中的空间对应捷径是一个普遍问题,ConGeo 通过对比学习显式破坏这种捷径的做法值得在其他任务中借鉴
  3. 与遥感领域的关联:跨视图定位对自动驾驶导航和无人机定位有直接价值,ConGeo 提高了方法的实用性
  4. 可扩展到视频定位:结合时序信息和 ConGeo 的视角不变性,有潜力用于视频级地理定位

评分

  • 新颖性: ⭐⭐⭐⭐ — 思路清晰直觉自然,对比学习框架设计合理,但核心组件(InfoNCE)并非全新
  • 实验充分度: ⭐⭐⭐⭐⭐ — 四个数据集、三种基线模型、丰富的消融和可解释性分析,实验非常全面
  • 写作质量: ⭐⭐⭐⭐ — 结构清晰,问题动机阐述到位,分析部分(Sec 6)尤其出色
  • 价值: ⭐⭐⭐⭐ — 实用性强,即插即用设计降低了应用门槛,单模型处理多设定在工程上很有吸引力