RoboTAG: End-to-end Robot Configuration Estimation via Topological Alignment Graph
About
Estimating robot pose from a monocular RGB image is a challenge in robotics and computer vision. Existing methods typically build networks on top of 2D visual backbones and depend heavily on labeled data for training, which is often scarce in real-world scenarios, causing a sim-to-real gap. Moreover, these approaches reduce the 3D-based problem to 2D domain, neglecting the 3D priors. To address these, we propose Robot Topological Alignment Graph (RoboTAG), which incorporates a 3D branch to inject 3D priors while enabling co-evolution of the 2D and 3D representations, alleviating the reliance on labels. Specifically, the RoboTAG consists of a 3D branch and a 2D branch, where nodes represent the states of the camera and robot system, and edges capture the dependencies between these variables or denote alignments between them. Closed loops are then defined in the graph, on which a consistency supervision across branches can be applied. Experimental results demonstrate that our method is effective across robot types, suggesting new possibilities of alleviating the data bottleneck in robotics.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Robot Pose Estimation | DREAM-real Panda 3CAM-AK | AUC83.1 | 19 | |
| Robot Pose Estimation | DREAM-real Panda 3CAM-RS | -- | 12 | |
| Robot Pose Estimation | DREAM Baxter DR | AUC58.8 | 8 | |
| Robot Pose Estimation | DREAM Panda Photo | AUC (ADD Curve)84.3 | 7 | |
| Robot Pose Estimation | DREAM Panda ORB | AUC (ADD)77.5 | 7 | |
| Robot Pose Estimation | DREAM Panda 3C-XK | AUC (ADD Curve)75.7 | 7 | |
| Robot Pose Estimation | DREAM Panda DR | AUC (ADD Curve)82.5 | 7 | |
| Robot Pose Estimation | DREAM Kuka Photo | AUC (ADD Curve)76.6 | 5 | |
| Robot Pose Estimation | DREAM Kuka DR | AUC (ADD curve)75 | 5 | |
| Robot Joint Angle Estimation | Panda DREAM (DR) | J1 Error4.7 | 4 |