当前路径:/wiki/page/simba
返回首页索引

Simba

Simba: Towards High-Fidelity and Geometrically-Consistent Point Cloud Completion via Transformation Diffusion(AAAI 2026)— 首将点云补全重新定义为变换场(transformation field)上的条件扩散生成,结合对称先验与 Mamba 架构。

Overview

  • 发布时间:AAAI 2026(arXiv 2511.16161v1)
  • 作者机构:南京航空航天大学
  • 代码I2-Multimedia-Lab/Simba
  • 核心洞察SymmCompletion 等回归方法有两大局限:① 过拟合训练分布,记忆特例而非泛化几何先验;② 逐点回归对噪声极度敏感

架构(两阶段)

Stage 1: SymmGT 预训练

  • 从部分点云和完整 GT 中提取关键点特征,通过交叉注意力回归目标变换场 $T^{gt} \in \mathbb{R}^{K \times 12}$(逐点 3×3 仿射矩阵 + 3D 平移向量)
  • 使用 CD 损失训练

Stage 2: Sym-Diffuser + MBA-Refiner

Sym-Diffuser(对称扩散模块)
- 核心思想:不扩散点坐标,而是扩散变换场
- 以部分点云特征 $F_k$ 为条件,学习变换场的条件分布 $p(T|F_k)$
- 100 步扩散,使用一致性模型风格训练目标
- 推理时从随机噪声去噪得到变换场,生成初始完整点云 $P_{init} = P_k \cup P_s$

MBA-Refiner(Mamba 基础细化器)
- 三级联:Blocks 1-2 使用 Cross-Attention 融合,Block 3 使用 Mamba Fusion
- Mamba Fusion 将多源特征排序后通过 Mamba 处理,避免 O(N²) 注意力复杂度
- 渐进式上采样:2× → 2× → 4×

关键创新

  1. 变换扩散:首次将点云补全建模为变换场的分布学习,而非坐标扩散
  2. MBA-Refiner:Mamba 替代 Transformer 做细化,降低计算复杂度
  3. 强跨域泛化:合成数据训练的模型在 KITTI 真实数据上无需微调即有竞争力

性能(PCN Dataset, CD × 10³ ↓)

方法 CD-AVG
SymmCompletion (AAAI 2025) 6.28
PCDreamer (CVPR 2025) 6.52
Simba 6.34

注:Simba 在部分类别(Lamp, Watercraft)有优势,整体与 SymmCompletion 接近但跨域泛化更强。

Sources