arXiv:2510.04125cs.CV2025-10ICCV被引 1

通过联合回归与扩散模型,提升6D姿态估计的精度与效率。

Joint Learning of Pose Regression and Denoising Diffusion with Score Scaling Sampling for Category-level 6D Pose Estimation

  • 先用回归头预训练编码器,再联合优化扩散与回归网络,加速收敛。
  • 在REAL275等数据集上达到最新最优性能,单次推理即达顶尖水平。
  • 无需额外筛选网络,时间依赖得分缩放实现高效采样探索与利用。

最新扩散模型通过深度图像输入建模类别级6D姿态分布,取得良好效果。然而现有方法存在训练收敛慢、端到端联合训练编码器与去噪网络、以及需额外网络评估采样姿态候选等问题。本文提出新框架,包含两个关键组件:首先,采用直接姿态回归头预训练编码器,并联合优化回归头与去噪扩散头,显著加快训练收敛并提升精度;其次,引入时间依赖得分缩放采样策略,有效平衡探索与利用,无需额外评估网络。该策略在早期去噪步骤保持对称物体的多模态特性,最终生成高质量姿态。在REAL275、HouseCat6D和ROPE等多个基准上实验表明,本方法简单而有效,在单姿态推理下仍达到最先进精度,且训练与推理更高效。

原文摘要 · Abstract (English)

Latest diffusion models have shown promising results in category-level 6D object pose estimation by modeling the conditional pose distribution with depth image input. The existing methods, however, suffer from slow convergence during training, learning its encoder with the diffusion denoising network in end-to-end fashion, and require an additional network that evaluates sampled pose hypotheses to filter out low-quality pose candidates. In this paper, we propose a novel pipeline that tackles these limitations by two key components. First, the proposed method pretrains the encoder with the direct pose regression head, and jointly learns the networks via the regression head and the denoising diffusion head, significantly accelerating training convergence while achieving higher accuracy. Second, sampling guidance via time-dependent score scaling is proposed s.t. the exploration-exploitation trade-off is effectively taken, eliminating the need for the additional evaluation network. The sampling guidance maintains multi-modal characteristics of symmetric objects at early denoising steps while ensuring high-quality pose generation at final steps. Extensive experiments on multiple benchmarks including REAL275, HouseCat6D, and ROPE, demonstrate that the proposed method, simple yet effective, achieves state-of-the-art accuracies even with single-pose inference, while being more efficient in both training and inference.

6D姿态估计扩散模型联合学习姿态回归

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。