中文

DecomPose: Disentangling Cross-Category Optimization Contention for Category-Level 6D Object Pose Estimation

计算机视觉与模式识别 2026-05-18 v1 人工智能

摘要

Category-level 6D object pose estimation is typically formulated as a multi-category joint learning problem with fully shared model parameters. However, pronounced geometric heterogeneity across categories entangles incompatible optimization signals in shared modules, resulting in gradient conflicts and negative transfer during training. To address this challenge, we first introduce gradient-based diagnostics to quantify module-level cross-category contention. Building on results of diagnostics, we propose DecomPose, a difficulty-aware decomposition framework that mitigates optimization contention via: (1) difficulty-aware gradient decoupling, which groups categories using a data-driven difficulty proxy and routes each instance to a group-specific correspondence branch to isolate incompatible updates; and (2) stability-driven asymmetric branching, which assigns higher-capacity branches to structurally simple categories as stable optimization anchors while constraining complex categories with lightweight branches to suppress noisy updates and alleviate negative transfer. Extensive experiments on REAL275, CAMERA25, and HouseCat6D demonstrate that DecomPose effectively reduces cross-category optimization contention and delivers superior pose estimation performance across multiple benchmarks.

关键词

引用

@article{arxiv.2605.15728,
  title  = {DecomPose: Disentangling Cross-Category Optimization Contention for Category-Level 6D Object Pose Estimation},
  author = {Yifan Gao and Lu Zou and Zhangjin Huang and Guoping Wang},
  journal= {arXiv preprint arXiv:2605.15728},
  year   = {2026}
}