跳转至

GeoDistill: Geometry-Guided Self-Distillation for Weakly Supervised Cross-View Localization

会议: ICCV 2025
arXiv: 2507.10935
代码: https://github.com/tongshw/GeoDistill
领域: Remote Sensing / Cross-View Localization
关键词: 跨视角定位, 弱监督学习, 自蒸馏, FoV遮挡, 方向估计

一句话总结

提出GeoDistill框架,通过基于视场角(FoV)遮挡的教师-学生自蒸馏范式增强局部判别性特征学习,在弱监督条件下(仅需粗略GPS标注)实现稳健的跨视角定位,性能提升超过10%且可即插即用于不同定位框架。

研究背景与动机

跨视角定位旨在通过匹配地面图像与卫星图像来估计相机的3自由度位姿(平面位置+偏航角),在自动驾驶和增强现实等大规模户外应用中至关重要。现有SOTA方法主要依赖全监督学习,需要精确的地面相机位姿标注——这通常需要装载昂贵传感器的专用测绘车辆遍历环境来获取,成本极高且跨区域泛化性差。

弱监督提供了更实际的替代方案:手机GPS就能获取粗略位置信息。但已有弱监督方法存在局限:(1)需要测试区域数据进行微调,实际中无法收集所有区域数据;(2)基于图像级度量学习无法为精确定位提供有力监督。核心问题在于:如何在仅有配对图像(无精确位置标注)的条件下学习判别性局部特征?

GeoDistill的核心洞察是:全景图和其部分视角描绘的是同一地理位置,因此必须映射到相同的卫星坐标——这种几何一致性要求本身就是一个强大的监督信号。

方法详解

整体框架

GeoDistill分为两阶段流水线: - 阶段1 - 方向估计: 预测地面图像与卫星图像的相对偏航角 \(\hat{\theta}\),通过水平平移全景图完成方向对齐。 - 阶段2 - 位置估计: 在任意跨视角定位网络上应用所提出的几何引导自蒸馏范式。推理时使用经EMA持续优化的教师模型。

关键设计

  1. 方向估计网络: 要对齐地面和航拍视角中共享的结构线索(如道路布局),首先用球面变换将全景图 \(I_g\) 投影为鸟瞰图(BEV)\(I_g^b\),矫正因透视投影导致的道路弯曲。然后分别用非共享权重的EfficientNet-B0提取卫星和BEV特征: \(\mathbf{F}_s = \mathcal{E}_s(\mathbf{I}_s), \quad \mathbf{F}_g = \mathcal{E}_g(\mathbf{I}_g^b)\) 通道拼接后通过MLP进行分类(每度一个类别),用交叉熵损失训练。

  2. FoV-based遮挡策略: 与随机patch遮挡或激活值遮挡不同,FoV遮挡模拟有限视场角的相机:\(\tilde{I_g}^m = I_g \odot M_{\text{mask}}\),其中mask保持一段连续的水平视场(如180°-240°),丢弃其余部分。关键优势在于:FoV遮挡总是保留一个连贯的场景几何结构(包含车道线、建筑等判别性特征),而patch遮挡可能破坏关键结构或保留无用区域(如天空)。值得注意的是,FoV遮挡作为单纯的数据增强反而会损害性能,只有在教师-学生蒸馏框架中才能发挥作用。

  3. 几何引导自蒸馏:

    • 教师模型 \(f_t\):接收完整全景图 \(\tilde{I_g}\) 和卫星图 \(I_s\),利用全局上下文生成热力图
    • 学生模型 \(f_s\):同架构但接收FoV遮挡图像 \(\tilde{I_g}^m\)\(I_s\)
    • 不确定性保留: 用低温度 \(\tau < 1\) 的softmax锐化双方热力图:\(P_t = \text{Softmax}(H_t/\tau)\), \(P_s = \text{Softmax}(H_s/\tau)\)。锐化的双重作用:压制噪声低置信度信号,同时保留教师的"暗知识"(分布形状和相对激活强度)
    • 自蒸馏损失: 交叉熵 \(\mathcal{L}_{SD} = \mathbb{E}[-\sum_i P_t(i)\log P_s(i)]\)
    • 双向知识流: 教师权重通过EMA吸收学生学到的鲁棒特征:\(\theta_t \leftarrow \alpha\theta_t + (1-\alpha)\theta_s\)\(\alpha=0.9\)),实现持续自我改进

损失函数 / 训练策略

学生通过梯度下降最小化自蒸馏损失更新参数,教师通过EMA机制更新。训练使用batch size 8,学习率0.0001,Adam优化器。温度 \(\tau=0.06\),EMA比率 \(\alpha=0.9\)。FoV范围在180°-240°之间随机采样。在单张NVIDIA 4090 GPU上完成所有实验。

实验关键数据

主实验

数据集 方法 Cross-Area Mean(m) Cross-Area Median(m) Same-Area Mean(m)
VIGOR CCVPE 4.97 1.68 3.60
VIGOR +GeoDistill 4.05(↓18.5%) 1.57(↓6.5%) 3.21(↓10.8%)
VIGOR G2SWeakly(VGG) 5.20 1.44 4.81
VIGOR +GeoDistill(VGG) 4.49(↓13.6%) 1.22(↓15.3%) 4.26(↓11.4%)
VIGOR G2SWeakly(DINO) 3.58 1.45 3.61
VIGOR +GeoDistill(DINO) 2.68(↓25.1%) 1.20(↓17.2%) 3.08(↓14.7%)

与全监督方法对比(VIGOR Cross-Area, 0°噪声):

方法 监督类型 Mean(m) Median(m)
HC-Net(当前SOTA) 全监督* 3.35 1.59
CCVPE 全监督* 4.97 1.68
GeoDistill(DINO) 弱监督 2.68 1.20

消融实验

遮挡策略 骨干网络 Mean(m) Median(m) 说明
FoV遮挡 CNN(VGG) 4.49 1.22 最优
最大激活值遮挡 CNN 5.14 1.42 破坏结构
随机Patch遮挡 CNN 5.21 1.44 无改善
FoV遮挡 ViT(DINO) 2.68 1.20 骨干无关性验证
随机Patch遮挡 ViT 3.10 1.33 -
不确定性策略 Mean(m) Median(m) 说明
带锐化 4.49 1.22 平衡降噪与暗知识保留
不锐化 5.23 1.44 原始暗知识太噪声,无法收敛
单模式 4.96 1.36 丢弃暗知识但可用
基线(无蒸馏) 5.20 1.44 -
FoV作为… G2SWeakly CCVPE 说明
数据增强 5.64(↑损害) 5.37(↑损害) 单纯增强有害
教师-学生蒸馏 4.49(↓提升) 4.05(↓提升) 框架中才有效

关键发现

  • 弱监督GeoDistill(DINO)在Cross-Area定位上超越了所有全监督SOTA方法,Mean误差2.68m < HC-Net的3.35m
  • FoV遮挡作为单纯数据增强会损害性能(+0.44m Mean),但在蒸馏框架中带来显著提升(-0.71m Mean),验证了教师-学生框架的必要性
  • 基础模型越强(如DINO vs VGG),GeoDistill带来的提升幅度越大(DINO: ↓25.1% vs VGG: ↓13.6%)
  • 最优FoV范围为180°-240°,过窄(信息不足)或过宽(与教师差异太小)都会劣化

亮点与洞察

  • 几何一致性约束转化为自监督信号的思路非常优雅——同一位置的全景和局部视角必须指向相同坐标,这个"免费"的约束就能驱动局部特征学习
  • FoV遮挡 vs patch遮挡的对比揭示了一个重要原则:遮挡策略应保持"可辨识的场景一致性"而非随机破坏
  • 锐化温度参数 \(\tau\) 对"暗知识"的精炼效果优于两个极端(全暗知识 vs 完全丢弃),体现了信息理论中噪声-信号的trade-off
  • 即插即用特性:不修改架构,仅通过蒸馏训练范式就能提升多个已有方法

局限与展望

  • 仅在全景图输入上验证FoV遮挡策略,对本身就是有限FoV的输入(如KITTI的pinhole相机),GeoDistill在KITTI上的提升较VIGOR小
  • 方向估计网络依赖BEV投影的道路结构对齐,在非道路场景(如山地、森林)可能效果减弱
  • EMA更新系数 \(\alpha\) 固定为0.9,可探索自适应调整
  • 仅在VIGOR和KITTI两个数据集上验证,多样性有限

相关工作与启发

  • 自蒸馏范式(Born-Again Networks)在知识蒸馏领域已有成功先例,本文将其扩展到跨视角定位是首创
  • FoV遮挡的思想与MAE的随机遮挡在理念上相通,但根据任务特性做了关键适配
  • 弱监督超越全监督的结果表明:在某些任务上,精心设计的自监督信号比昂贵的标注数据更有效

评分

  • 新颖性: ⭐⭐⭐⭐ FoV遮挡+自蒸馏的组合简洁有效,几何一致性约束的利用巧妙
  • 实验充分度: ⭐⭐⭐⭐ 多基线方法验证即插即用性,消融全面但数据集仅两个
  • 写作质量: ⭐⭐⭐⭐⭐ 动机清晰,消融设计精心,每个设计选择都有对照实验支撑
  • 价值: ⭐⭐⭐⭐ 弱监督超全监督,为跨视角定位的部署提供了实际可行的低成本方案

相关论文