返回研究文献库

物体位姿估计 · 2024 · CVPR · CCF A

Confronting Ambiguity in 6D Object Pose Estimation via Score-Based Diffusion on SE(3)

该方法直接学习条件分布 p(R,t|I),把 score diffusion 定义在 SE(3) 而非把旋转和平移独立扩散,并用 surrogate Stein score 避开昂贵 Jacobian。T-LESS 上 SE(3) 模型的 MSPD / R@5 为 93.16 / 86.94;100 步为 20 FPS,5 步仍有 250 FPS。

研究方向

物体位姿估计

任务层级

Instance-level

物体设定

Seen objects

输入模态

single RGB object cropvisible-object segmentation mask in the T-LESS protocol

关键依赖

ground-truth pose labels for trainingground-truth bounding boxes and visible masks in the T-LESS evaluationknown test-object identities represented during trainingiterative reverse diffusionno CAD model or symmetry annotations
完整论文元数据 作者、单位、研究方向、数据集、指标、方法标签和阅读状态 展开
作者
Tsu-Ching Hsiao, Hao-Wei Chen, Hsuan-Kung Yang, Chun-Yi Lee
作者单位
Elsa Lab, National Tsing Hua University
会议 / 期刊
会议 · CVPR · CCF A
研究方向
物体位姿估计
开源状态
未开源 / 未整理
阅读状态
粗读:已生成 精读:已生成
任务设定
Instance-level · Seen objects
评估指标
Rotation Error Translation Error MSPD MSSD VSD

论文原始写法:distribution spread / MSPD / MSSD / VSD / R@2 / R@5 / R@10 / T@2 / T@5 / T@10 / FPS

关键依赖
ground-truth pose labels for trainingground-truth bounding boxes and visible masks in the T-LESS evaluationknown test-object identities represented during trainingiterative reverse diffusionno CAD model or symmetry annotations

Reading Notes

粗读 / 精读

已生成 src/paper-notes/se3-diffusion-pose-cvpr2024/coarse.md

论文粗读笔记

用于快速判断论文定位、问题设定、核心方法、实验结论和是否值得精读。

论文粗读笔记:SE(3) Score Diffusion

1. 基本信息

  • 标题:Confronting Ambiguity in 6D Object Pose Estimation via Score-Based Diffusion on SE(3)
  • 作者:Tsu-Ching Hsiao, Hao-Wei Chen, Hsuan-Kung Yang, Chun-Yi Lee
  • 机构:Elsa Lab, National Tsing Hua University
  • 年份 / 会议:2024 / CVPR
  • 论文:CVF 正式页面 · 正式 PDF
  • 阅读模式:粗读

2. 一句话定位

该工作不把单图 pose 强行回归成唯一 R,tR,t,而是在刚体运动群 SE(3)SE(3) 上学习条件 score,并通过反向扩散采样多个合理姿态;surrogate Stein score 解决 SE(3)SE(3) Jacobian 使真实 score 昂贵且去噪方向偏离的问题。

3. 图表导读

条件 SE3 score diffusion 网络和去噪路径
原论文 Figure 2 的方法图:RGB 只编码一次得到条件 \(c\),pose 与时间噪声分别做 positional embedding,MLP score network 迭代更新 Lie algebra 中的 noisy pose。右侧示意子步越多,曲面上的去噪轨迹越接近真实逆变换。来源:CVPR 2024 正式论文;图片版权归论文作者及 IEEE/CVF 所有,本站仅作论文解读展示。
SE3 diffusion 在 SYMSOL SYMSOL-T 和 T-LESS 的结果
原论文 Tables 2–4:SYMSOL 平均角度 spread 为 0.37°;SYMSOL-T 上联合 SE(3) 显著改善 icosahedron 的旋转多模态;T-LESS 的 MSPD / R@5 为 93.16 / 86.94。来源:CVPR 2024 正式论文;表格版权归论文作者及 IEEE/CVF 所有,本站仅作论文解读展示。

4. 问题设定

  • 输入:单张 RGB object crop;T-LESS 协议还使用真值 bounding box 与 visible-object mask。
  • 输出:条件分布 p(X∣I)p(X\mid I) 的 pose samples,X=(R,T)∈SE(3)X=(R,T)\in SE(3)。
  • 训练监督:RGB 与 GT pose;不需要 3D CAD 或 symmetry annotations。
  • 数据:SYMSOL、论文新建的 SYMSOL-T、真实 T-LESS 30 objects。
  • 目标:覆盖对称、遮挡、自遮挡与透视引起的多解,而不是仅输出一个平均 pose。

5. 方法概览

  1. 在 Lie group 上以 Gaussian perturbation kernel 给 GT pose 加噪。
  2. ResNet 从 RGB 提取一次条件 cc,避免每个去噪 step 重算图像特征。
  3. MLP score network 在对应 Lie algebra 中接收 noisy pose、noise level 和图像条件。
  4. 用 Geodesic Random Walk 的反向过程逐步生成 pose samples。
  5. 对 SO(3)SO(3) 与 R3SO(3)\mathbb R^3SO(3),Stein score 可简化为 −z/σ2-z/\sigma^2;对 SE(3)SE(3) 不满足同样 Jacobian 对称性,作者以该量作为 surrogate target,并用更小子步逼近逆变换。

6. 实验速览

  • SYMSOL 五形状平均 spread:ResNet34 0.42°,ResNet50 0.37°;Normalizing Flows 0.70°。
  • SYMSOL-T:SE(3)SE(3) 模型五形状 rotation spread 为 0.59、0.58、0.64、0.54、0.41°;R3SO(3)\mathbb R^3SO(3) 在 icosahedron 上为 29.35°。
  • T-LESS:SE(3) 的 MSPD 93.16、MSSD 60.17、VSD 56.88、R@2 47.21、R@5 86.94、R@10 94.78。
  • GDRNPP translation 更好:T@2 90.31 vs 71.72;说明无 3D geometry 的 diffusion 深度仍弱。
  • SE(3) 100 / 10 / 5 steps:20 / 161 / 250 FPS;5 steps 的 MSPD 仍为 92.40。

7. 快速判断

  • 亮点:将歧义建模成 pose posterior,而不是靠 symmetry-aware 单值 loss 规避。
  • 最强证据:T-LESS 可视化会随可见部件变化产生离散或连续 mode,超过静态 symmetry annotation 的表达能力。
  • 限制:真实实验依赖 GT box / mask;T-LESS 对象在训练中已知;translation 明显落后有 3D geometry 的 GDRNPP。
  • 适合读者:Lie-group diffusion、pose uncertainty、对称性、多假设 6DoF、score-based generative modeling。

Related Papers

相关论文