CVPR|2025|RCTDistill:基于时间融合的雷达—相机三维目标检测跨模态知识蒸馏框架

  • 论文:https://arxiv.org/pdf/2410.00871
  • 会议:CVPR
  • 年份:2025

背景

雷达 - 相机融合是自动驾驶 3D 目标检测的低成本方案,但性能始终落后于激光雷达(LiDAR)基方法,核心瓶颈在于 BEV(鸟瞰图)特征存在空间错位与时间错位。

现有方法的局限

核心挑战

  1. 传感器不确定性:相机深度估计模糊、雷达水平定位精度低,融合时难以抵消两者固有误差;
  2. 时间特征错位:动态目标独立运动导致多帧 BEV 特征对齐困难,传统时间融合需复杂运动估计,增加延时;
  3. 特征判别性不足:雷达 - 相机融合模型难以像激光雷达模型那样清晰区分前景目标与背景

在这里插入图片描述

  • 基线模型的预测框与静态 GT(绿色框)存在明显偏移,部分目标甚至出现 “漏检” 或 “定位偏差过大”。
  • 基线模型对动态 GT(红色框)的预测框出现 “拖影” 或 “位置滞后”,与真实运动轨迹偏差较大,甚至误将运动轨迹判定为多个目标。
  • RCTDistill 通过在距离-方位方向上对齐空间特征以及在动态物体轨迹上对齐时间特征,提高了 BEV 特征的三维目标检测质量。

方法

img

  • 教师模型,基于 LiDAR 的 CenterPoint 检测器,输出低维(BlowL)和高维(BhighL)BEV 特征,作为知识来源。
  • 学生模型,基于改进 BEVFusion-R 的雷达 - 相机融合模型,接收雷达 + 相机输入,经多阶段处理输出检测结果。
    雷达骨干网络(Radar Backbone),提取雷达原始数据的特征,为 BEV 转换提供基础
  • 门控模块(Gating Module),自适应融合雷达与相机的低维 BEV 特征,动态调节两模态权重(替代传统 1×1 卷积)。本质是一个 “权重生成器”,输出范围通常在 [0,1](通过 Sigmoid 激活),用于对输入特征 X 进行逐元素加权
  • 记忆库(Memory Bank),存储历史帧的低维 BEV 特征(Bt−N−1 Bt−1),支持时间融合,通道维上拼接
  • HA-Net(Historical Alignment Network),将 “历史 BEV 特征” 与 “当前 BEV 特征” 做时间聚合。

1.时间融合方法旨在通过结合过去帧的额外信息并将其与当前特征融合,从而弥补单时刻可能遗漏的信息来提升性能。
2.三大 KD 模块(RAKD/TKD/RDKD),分别在低维特征、时间聚合特征、高维特征阶段,实现师生模型知识对齐
3.图中紫色(Teacher)、黄色(Student)特征图里,不同红色区块(代表不同目标)之间的虚线箭头,对应 “同一帧中多个目标的特征区分关系”—— 比如两个相邻目标的特征需要有足够差异,避免混淆,RDKD 会让 Student 学习 Teacher 中这种 “不同目标间的特征区分度”。

img

RAKD 和 TKD 的椭圆高斯掩模区域。RAKD(中间)和 TKD(右侧)区域的椭圆掩模用于知识蒸馏,分别针对距离-方位不确定性和时间错位。

RAKD-距离方向角蒸馏

r 1 i = l i ⋅ ( α l l i ) β , r 2 i = w i ⋅ ( α w w i ) β , (1) r_{1}^{i} = l^{i} \cdot \left( \frac{\alpha_{l}}{l^{i}} \right)^{\beta}, \quad r_{2}^{i} = w^{i} \cdot \left( \frac{\alpha_{w}}{w^{i}} \right)^{\beta}, \tag{1} r1i=li(liαl)β,r2i=wi(wiαw)β,(1)

含义

  • 第i个 GT 目标的长度 / 宽度;
  • 距离归一化系数,用于平衡不同距离下的掩码尺度。核心是把距离值映射到固定范围(通常是 [0,1])
  • 超参数,分别控制 “距离对长度方向半径”“距离对方位方向半径” 的影响
  • 根据目标的尺寸、与自车的距离,动态确定椭圆高斯掩码的长半轴,短半轴。
    使用这些参数,我们生成第i个GT框在位置(x,y)处的椭圆高斯掩膜 E

E i , x , y = exp ⁡ ( − 1 2 ( ( x ′ ) 2 ( r 1 i ) 2 + ( y ′ ) 2 ( r 2 i ) 2 ) ) , (2) E_{i,x,y} = \exp\left( -\frac{1}{2} \left( \frac{(x')^2}{(r_1^i)^2} + \frac{(y')^2}{(r_2^i)^2} \right) \right), \tag{2} Ei,x,y=exp(21((r1i)2(x)2+(r2i)2(y)2)),(2)

含义

  • 位置(x,y)相对于 GT 中心的局部坐标(经公式 3 的朝向对齐变换后);
  • 公式 1 计算的椭圆长 / 短半轴。
  • 特点:相机中的深度模糊和雷达中的距离-方位不确定性。高斯掩码应用的是从物体中心均匀递减的权重,这使得其难以适应每种模态的独特分布。椭圆的形状由距离和方位方向上的固有不确定性决定
    -** 生成第i个 GT 目标在位置(x,y)处的椭圆高斯掩码值,用于限定蒸馏的区域**
    [ x ′ y ′ ] = [ cos ⁡ θ i − sin ⁡ θ i sin ⁡ θ i cos ⁡ θ i ] ⋅ [ x − p x i y − p y i ] . (3) \begin{bmatrix} x' \\ y' \end{bmatrix} = \begin{bmatrix} \cos\theta^i & -\sin\theta^i \\ \sin\theta^i & \cos\theta^i \end{bmatrix} \cdot \begin{bmatrix} x - p_x^i \\ y - p_y^i \end{bmatrix}. \tag{3} [xy]=[cosθisinθisinθicosθi][xpxiypyi].(3)

含义

  • 核心作用:将像素坐标系下的位置(x,y),转换为 “以 GT 中心为原点、对齐目标朝向” 的局部坐标系下的偏移量(x’,y’)。
  • 算法意义:目标存在朝向差异(如横向 / 纵向车辆),通过旋转坐标,让椭圆掩码的方向与目标实际朝向一致,确保掩码能精准覆盖目标区域,避免因朝向错位导致的蒸馏区域偏差。

L RA = 1 ∣ N RA ∣ ∑ j = 1 H ∑ k = 1 W W RA , j , k ∥ B low , j , k L − B ˉ low , j , k RC ∥ 2 , (4) L_{\text{RA}} = \frac{1}{|N_{\text{RA}}|} \sum_{j=1}^{H} \sum_{k=1}^{W} W_{\text{RA},j,k} \left\| B_{\text{low},j,k}^{\text{L}} - \bar{B}_{\text{low},j,k}^{\text{RC}} \right\|_2, \tag{4} LRA=NRA1j=1Hk=1WWRA,j,k Blow,j,kLBˉlow,j,kRC 2,(4)

公式 4:RAKD 损失函数

  • 蒸馏掩码中非零元素的数量(即有效蒸馏区域的像素数)
  • W RA , j , k W_{\text{RA},j,k} WRA,j,k:融合所有 GT 椭圆掩码后的蒸馏掩码(取最大值实现掩码融合)
  • B low L B_{\text{low}}^{\text{L}} BlowL:教师模型的低维 BEV 特征; B ˉ low RC \bar{B}_{\text{low}}^{\text{RC}} BˉlowRC:学生模型低维 BEV 特征(对原始特征 B low RC B_{\text{low}}^{\text{RC}} BlowRC 应用 1×1 卷积,获得的经通道匹配后的结果)。
  • B ˉ low RC \bar{B}_{\text{low}}^{\text{RC}} BˉlowRC:学生模型低维 BEV 特征(对原始特征 B low RC B_{\text{low}}^{\text{RC}} BlowRC 应用 1×1 卷积,获得的经通道匹配后的结果)。
  • 算法意义:让学生模型在雷达 - 相机误差敏感的区域,精准对齐教师 LiDAR 的高保真空间特征,修正传感器不确定性导致的 BEV 特征空间错位

公式5 TKD-时间知识蒸馏

为动态目标(如运动的车辆 / 行人)构建时序掩码,并通过损失函数让学生模型的时序特征与教师模型对齐

p ^ i = { p i − t s 2 v i , if  ∥ v i ∥ 2 2 > τ v (5) \hat{\mathbf{p}}^i = \begin{cases} \mathbf{p}^i - \frac{t_s}{2} \mathbf{v}^i, & \text{if } \left\| \mathbf{v}^i \right\|_2^2 > \tau_v \end{cases} \tag{5} p^i={pi2tsvi,if  vi 22>τv(5)

p i \mathbf{p}^i pi第i个动态目标的当前中心

t s t_s ts:选择的时序持续时间(小于历史轨迹总时长)

τ v \tau_v τv:速度阈值,判断是否快速移动

作用:当目标速度超过阈值时,将椭圆掩码的中心向 “速度反方向” 偏移,让掩码能覆盖目标的历史运动轨迹(而非仅当前位置),适配动态目标的轨迹范围。

r ^ 1 i = l i + ( p ^ x i − p x i ) 2 + ( p ^ y i − p y i ) 2 (6) \hat{r}_1^i = l^i + \sqrt{(\hat{p}_x^i - p_x^i)^2 + (\hat{p}_y^i - p_y^i)^2} \tag{6} r^1i=li+(p^xipxi)2+(p^yipyi)2 (6)
r ^ 2 i = w i , (7) \hat{r}_2^i = w^i, \tag{7} r^2i=wi,(7)
T i , x , y = exp ⁡ ( − 1 2 ( ( x ′ ′ ) 2 ( r ^ 1 i ) 2 + ( y ′ ′ ) 2 ( r ^ 2 i ) 2 ) ) , (8) T_{i,x,y} = \exp\left( -\frac{1}{2} \left( \frac{(x'')^2}{(\hat{r}_1^i)^2} + \frac{(y'')^2}{(\hat{r}_2^i)^2} \right) \right), \tag{8} Ti,x,y=exp(21((r^1i)2(x′′)2+(r^2i)2(y′′)2)),(8)

[ x ′ ′ y ′ ′ ] = [ cos ⁡ θ i − sin ⁡ θ i sin ⁡ θ i cos ⁡ θ i ] [ x − p ^ x i y − p ^ y i ] . (9) \begin{bmatrix} x'' \\ y'' \end{bmatrix} = \begin{bmatrix} \cos \theta^i & -\sin \theta^i \\ \sin \theta^i & \cos \theta^i \end{bmatrix} \begin{bmatrix} x - \hat{p}_x^i \\ y - \hat{p}_y^i \end{bmatrix}. \tag{9} [x′′y′′]=[cosθisinθisinθicosθi][xp^xiyp^yi].(9)

L T = 1 ∣ N T ∣ ∑ j = 1 H ∑ k = 1 W W T , j , k ∥ B low , j , k L − B ^ low , j , k RC ∥ 2 , (10) L_T = \frac{1}{|N_T|} \sum_{j=1}^{H} \sum_{k=1}^{W} W_{T,j,k} \left\| B_{\text{low},j,k}^{\text{L}} - \hat{B}_{\text{low},j,k}^{\text{RC}} \right\|^2, \tag{10} LT=NT1j=1Hk=1WWT,j,k Blow,j,kLB^low,j,kRC 2,(10)

W:时序掩码的有效区域(取多个椭圆掩码的最大值,再过滤出大于阈值(\tau)的区域);

RDKD-区域解耦知识蒸馏

S j k RC = f j RC ⋅ f k RC ∥ f j RC ∥ 2 ∥ f k RC ∥ 2 , for  j , k = 1 , 2 , … , K , (11) S_{jk}^{\text{RC}} = \frac{\mathbf{f}_j^{\text{RC}} \cdot \mathbf{f}_k^{\text{RC}}}{\|\mathbf{f}_j^{\text{RC}}\|_2 \|\mathbf{f}_k^{\text{RC}}\|_2}, \quad \text{for } j, k = 1, 2, \dots, K, \tag{11} SjkRC=fjRC2fkRC2fjRCfkRC,for j,k=1,2,,K,(11)

  • 求余弦相似度函数

  • f:高层特征图输入检测头,生成分类得分图,取 “所有类别得分中的最大值”,得到置信度得分图,设定一个阈值,大于阈值共K个,对应位置上提分类得分图,得到向量。

L RD = 1 ∣ K 2 ∣ ∑ j = 1 K ∑ k = 1 K ∣ S j k L − S j k RC ∣ . (12) L_{\text{RD}} = \frac{1}{|K^2|} \sum_{j=1}^{K} \sum_{k=1}^{K} |S_{jk}^L - S_{jk}^{\text{RC}}|. \tag{12} LRD=K21j=1Kk=1KSjkLSjkRC∣.(12)

总Loss

L t o t a l = L d e t + λ RA L RA + λ T L T + λ RD L RD , (13) L_{total} = L_{det} + \lambda_{\text{RA}} L_{\text{RA}} + \lambda_{\text{T}} L_{\text{T}} + \lambda_{\text{RD}} L_{\text{RD}}, \tag{13} Ltotal=Ldet+λRALRA+λTLT+λRDLRD,(13)

实验

img

  • RCTDistill与现有相机-雷达融合及纯相机3D 目标检测器在nuScenes验证集和测试集上的表现。

  • 在测试集上,本方法超越所有现有雷达-相机融合 模型。

  • RCTDistill在所有评估配置中均实现了最高的推理速度

消融实验

img

  • 对比不同模块区域的KD

img

  • 对比现有时间蒸馏方法

  • 对比传统知识蒸馏(整个区域)

img

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐