UA-Depth: Uncertainty-Aware Depth Restoration for Material-Agnostic Object Manipulation
Organizations: Department of Mechanical and Automation Engineering, The Chinese University of Hong Kong, Hong Kong, China. · Department of Computer Science and Engineering, The Chinese University of Hong Kong, Hong Kong, China. · The University of Hong Kong, Hong Kong, China. · Sino-German College of Intelligent Manufacturing, Shenzhen Technology University, Shenzhen, China.
Abstract
Robotic manipulation requires accurate metric depth, yet commodity RGB-D cameras often fail on non-Lambertian objects such as transparent, reflective, and translucent materials, producing holes, structured noise, and biased depth. Most restoration methods produce only a single depth map, treating noisy and reliable regions uniformly for downstream tasks while ignoring regional uncertainty. We present UA-Depth, an uncertainty-aware depth restoration framework that iteratively co-refines depth and pixel-wise uncertainty, and uses uncertainty-gated features to suppress unreliable cues for stable updates. To strengthen zero-shot behavior, UA-Depth leverages a frozen depth foundation model as a prior to initialize missing or unreliable depth, and applies two-stage MAD-weighted log-depth scale normalization to reduce outlier bias in relative-to-metric alignment. We introduce an eight-dataset zero-shot benchmark across diverse scenes, materials, and sensors. Across these benchmarks, UA-Depth achieves the best mean rank among representative baselines, with particularly strong gains on datasets dominated by transparent and reflective objects. Beyond depth metrics, we demonstrate downstream improvements in category-level 6D pose estimation and 6D grasp synthesis by exploiting the predicted uncertainty. Code, models, and data for training and evaluation will be released upon publication.
Figures & tables
| Split | Dataset | Size | Object | Scene | Depth Sensor |
| Training | DREDS-CatKnown [ 11 ] | 100K | 1536 objects, 7 categories | 74 scenes | Synthetic |
| ClearPose [ 56 ] | 222K | 63 objects, 7 categories | 24 scenes | RealSense L515 | |
| HISS [ 14 ] | 65K | 350 objects | 160 scenes | Synthetic | |
| Testing | HAMMER [ 26 ] | 1414 | 20 objects, 8 categories | 9 scenes | RealSense D435 |
| HouseCat6D [ 27 ] | 2929 | 50 objects, 10 categories | 5 scenes | RealSense D435 | |
| TransCG [ 25 ] | 11K | 46 objects | 52 scenes | RealSense D435 |
| Method | RMSE | REL | MAE | |||
| SwinDR [ 11 ] | 0.0433 | 0.0364 | 0.0282 | 78.13 | 90.06 | 98.24 |
| [ 14 ] | 0.0405 | 0.0494 | 0.0265 | 76.95 | 91.26 | 98.22 |
| DP [ 36 ] | 0.0587 | 0.0448 | 0.0274 | 82.98 | 89.97 | 95.89 |
| OGNI-DC [ 19 ] | 0.0491 | 0.0338 | 0.0242 | 82.17 | 92.73 | 98.44 |
| PromptDA [ 21 ] | 0.0523 | 0.0546 | 0.0317 | 69.73 | 85.89 | 95.39 |
| PPFT [ 58 ] | 0.0351 | 0.0302 | 0.0201 | 85.23 | 94.23 | 98.83 |
| Methods | RMSE | REL | MAE | |||
| NLSPN [ 35 ] | 0.010 | 0.009 | 0.006 | 97.48 | 99.51 | 99.97 |
| LIDF [ 59 ] | 0.016 | 0.018 | 0.011 | 93.60 | 98.71 | 99.92 |
| SwinDR [ 11 ] | 0.010 | 0.008 | 0.005 | 98.04 | 99.62 | 99.98 |
| ASGrasp [ 23 ] | 0.007 | 0.006 | 0.004 | - | - | - |
| RoMa [ 14 ] | 0.0101 | 0.0075 | 0.0050 | 98.11 | 99.49 | 99.97 |
| PromptDA [ 21 ] | 0.0138 | 0.0141 | 0.0090 | 95.96 | 99.12 | 99.93 |
| Methods | RMSE | REL | MAE | |||
| STD-CatKnown | ||||||
| SwinDR [ 11 ] | 0.015 | 0.013 | 0.008 | 96.66 | 99.03 | 99.92 |
| RoMa [ 14 ] | 0.0176 | 0.0133 | 0.0096 | 96.04 | 98.42 | 99.74 |
| PromptDA [ 21 ] | 0.3427 | 0.5497 | 0.3353 | 71.63 | 77.29 | 82.26 |
| Ours | 0.0155 | 0.0125 | 0.0083 | 96.92 | 98.87 | 99.89 |
| STD-CatNovel | ||||||
| Testset | Method | RMSE | REL | MAE | |||
| New Background | ImplicitDepth [ 59 ] | 0.07 | 0.05 | 0.04 | 67.00 | 87.03 | 97.50 |
| TransCG [ 25 ] | 0.03 | 0.03 | 0.02 | 86.50 | 97.02 | 99.74 | |
| [ 14 ] | 0.05 | 0.01 | 0.01 | 96.71 | 98.84 | 99.74 | |
| PromptDA [ 21 ] | 0.240 | 0.045 | 0.061 | 85.03 | 93.66 | 97.49 | |
| Ours | 0.057 | 0.009 | 0.011 | 96.18 | 98.55 | 99.55 | |
| Heavy Occlusion | ImplicitDepth [ 59 ] | 0.11 | 0.09 | 0.08 | 41.43 | 66.52 | 91.96 |
| Method | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | ||||||||||
| SwinDR [ 11 ] | 0.0429 | 76.35 | 0.0463 | 76.84 | 0.0307 | 81.10 | 0.1078 | 61.75 | 0.0241 | 88.82 | 0.0151 | 93.53 | 0.0247 | 85.19 | 0.0318 | 77.95 | 3.2 |
| [ 14 ] | 0.0655 | 58.43 | 0.0498 | 76.45 | 0.0419 | 67.56 | 0.1231 | 50.27 | 0.0494 | 81.61 | 0.0511 | 73.36 | 0.0650 | 47.27 | 0.0575 | 50.07 | 5.3 |
| PromptDA [ 21 ] | 0.5156 | 6.62 | 0.3444 | 24.75 | 0.2130 | 8.97 | 0.3535 | 17.28 | 0.4402 | 15.52 | 0.1351 | 62.97 | 0.3060 | 38.45 | 0.6614 | 10.03 | 7.0 |
| OMNI-DC [ 20 ] | 0.0586 | 67.47 | 0.0530 | 71.34 | 0.0360 | 79.06 | 0.1642 | 52.00 | 0.0215 | 94.59 | 0.0260 | 91.52 | 0.0047 | 99.03 | 0.0063 | 99.40 | 3.1 |
| PriorDA [ 22 ] | 0.0762 | 41.42 | 0.0720 | 45.51 | 0.0418 | 71.37 | 0.1492 | 46.55 | 0.0338 | 87.47 | 0.0420 | 73.41 | 0.0205 | 95.26 | 0.0182 | 95.15 | 4.9 |
| Model Variant | ADF | Unc. Modeling | UCFC | HAMMER | DREDS-CatKnown | ||||||
| MAE | RMSE | REL | MAE | RMSE | REL | ||||||
| Baseline | ✗ | ✗ | ✗ | 86.55 | 0.0182 | 0.0322 | 0.0272 | 98.15 | 0.0046 | 0.0094 | 0.0070 |
| + ADF | ✓ | ✗ | ✗ | 87.81 | 0.0170 | 0.0321 | 0.0263 | 98.17 | 0.0045 | 0.0095 | 0.0069 |
| + Unc. Modeling | ✓ | ✓ | ✗ | 86.46 | 0.0184 | 0.0329 | 0.0271 | 98.19 | 0.0045 | 0.0094 | 0.0069 |
| + UCFC (Ours) | ✓ | ✓ | ✓ | 89.56 | 0.0164 | 0.0309 | 0.0243 | 98.32 | 0.0044 | 0.0089 | 0.0067 |
| Model Variant | REL | RMSE | REL | RMSE | |||||
| AUSE | AURG | AUSE | AURG | AUSE | AURG | ||||
| Baseline + ADF | 0.00685 | 0.00945 | 98.17 | – | – | – | – | – | – |
| + | 0.00794 | 0.01038 | 97.70 | 0.00021 | 0.00055 | 0.00036 | 0.00075 | 0.00040 | 0.00259 |
| + | 0.00642 | 0.00924 | 98.26 | 0.00059 | 0.00003 | 0.00090 | 0.00011 | 0.00200 | 0.00020 |
| + Grad | 0.00653 | 0.00921 | 98.26 | 0.00021 | 0.00041 | 0.00028 | 0.00073 | 0.00043 | 0.00181 |
| + + Grad (Ours) | 0.00672 | 0.00898 | 98.32 | 0.00019 | 0.00046 | 0.00024 | 0.00073 | 0.00032 | 0.00188 |
| Model Variant | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | ||||||||||
| Plain refinement | 0.0365 | 78.05 | 0.0346 | 85.14 | 0.0221 | 86.98 | 0.0566 | 72.23 | 0.0295 | 91.30 | 0.0149 | 95.33 | 0.0090 | 97.87 | 0.0124 | 98.17 | 4.00 |
| Residual gate | 0.0393 | 75.92 | 0.0391 | 81.85 | 0.0228 | 86.24 | 0.0682 | 66.67 | 0.0267 | 93.31 | 0.0148 | 94.67 | 0.0065 | 99.38 | 0.0116 | 98.48 | 3.88 |
| Simple conv fusion | 0.0351 | 82.36 | 0.0330 | 85.80 | 0.0214 | 87.75 | 0.0727 | 64.39 | 0.0223 | 94.11 | 0.0144 | 95.07 | 0.0057 | 99.59 | 0.0114 | 98.47 | 2.50 |
| UCFC-PM | 0.0345 | 79.46 | 0.0322 | 86.83 | 0.0220 | 87.43 | 0.0545 | 72.47 | 0.0404 | 79.91 | 0.0146 | 95.40 | 0.0081 | 99.01 | 0.0121 | 98.54 | 2.75 |
| UCFC (Ours) | 0.0326 | 82.38 | 0.0334 | 86.86 | 0.0199 | 89.52 | 0.0701 | 67.28 | 0.0262 | 93.94 | 0.0137 | 95.73 | 0.0069 | 98.35 | 0.0103 | 98.74 | 1.88 |
| Model Variant | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | ||||||||||
| Vanilla (w/o SN) | 0.0349 | 80.84 | 0.0424 | 83.05 | 0.0221 | 87.13 | 0.0671 | 66.67 | 0.0252 | 93.60 | 0.0120 | 96.32 | 0.0078 | 99.20 | 0.0199 | 90.36 | 3.25 |
| SN-Median | 0.0327 | 83.19 | 0.0351 | 86.84 | 0.0198 | 89.46 | 0.0657 | 69.16 | 0.0285 | 93.85 | 0.0159 | 95.45 | 0.0077 | 98.90 | 0.0109 | 98.49 | 2.63 |
| SN-LogBias | 0.0343 | 81.52 | 0.0350 | 85.68 | 0.0204 | 88.95 | 0.0648 | 69.21 | 0.0255 | 94.01 | 0.0175 | 95.64 | 0.0073 | 99.06 | 0.0105 | 98.66 | 2.44 |
| SN-LogBias-Robust | 0.0324 | 82.96 | 0.0345 | 86.09 | 0.0193 | 89.66 | 0.0677 | 67.20 | 0.0259 | 94.35 | 0.0143 | 95.83 | 0.0063 | 99.25 | 0.0096 | 99.12 | 1.69 |
| w/o Unc | w/ Unc | ||||
| Dataset | REL | REL | AUSE(REL) | ||
| Scene pixels | |||||
| HAMMER | 0.0398 | 79.05 | 0.0379 ( 4.77%) | 80.98 ( 2.44%) | 0.00131 |
| TransCG | 0.0115 | 94.55 | 0.0085 ( 26.09%) | 96.81 ( 2.39%) | 0.00069 |
| ROBI | 0.0053 | 96.94 | 0.0024 ( 54.72%) | 99.61 ( 2.75%) | 0.00021 |
| Object pixels | |||||
| (a) Overall methods Method Time (ms) SwinDR [ 11 ] 73.0 PromptDA [ 21 ] 103.4 OMNI-DC [ 20 ] 277.0 PriorDA [ 22 ] 297.0 CDM [ 15 ] 383.4 D 3 RoMa [ 14 ] 2116.6 Ours 225.6 | (b) Deployment cost Item Value Params 349.2M Trainable params 13.8M Checkpoint size 1.43 GB FLOPs 1367.2G Infer. peak VRAM 2.09 GB Train peak VRAM 19.4 GB/GPU Training cost 24 GPU-days | (c) Latency breakdown Component Time (ms) DAV2 backbone 118.5 Robust SN 6.1 Context networks 57.4 ConvGRU 43.5 Total 225.6 |
| Depth Type | 2cm | 5cm | 2cm | 5cm | |||
| Raw | 77.3 | 52.6 | 17.2 | 6.7 | 8.9 | 17.6 | 25.4 |
| SwinDR [ 11 ] | 77.3 | 57.8 | 21.9 | 7.5 | 9.9 | 20.9 | 30.1 |
| D 3 RoMa [ 14 ] | 71.7 | 42.6 | 12.4 | 2.8 | 4.6 | 10.6 | 17.9 |
| CDM [ 15 ] | 80.7 | 58.4 | 18.4 | 10.0 | 14.7 | 22.4 | 35.6 |
| Ours | 79.9 | 60.5 | 30.3 | 13.3 | 15.0 | 28.9 | 36.7 |
| Ours w/ Unc - Sampling | 79.4 | 60.2 | 31.9 | 14.3 | 16.4 | 30.8 | 39.7 |
| Depth Type | Seen | Similar | Novel | Mean AP | ||||||
| AP | AP 0.8 | AP 0.4 | AP | AP 0.8 | AP 0.4 | AP | AP 0.8 | AP 0.4 | ||
| Raw | 23.15 | 27.73 | 19.81 | 17.97 | 21.86 | 15.15 | 6.48 | 8.05 | 3.36 | 15.87 |
| SwinDR [ 11 ] | 28.95 | 35.45 | 22.41 | 22.51 | 27.82 | 17.32 | 10.20 | 12.64 | 4.70 | 20.55 |
| (RGB - D) [ 14 ] | 26.52 | 33.86 | 16.19 | 22.45 | 28.38 | 15.02 | 9.53 | 11.59 | 3.79 | 19.50 |
| (RGB - D Stereo) [ 14 ] | 33.01 | 40.31 | 26.32 | 25.82 | 31.91 | 19.63 | 10.89 | 13.41 | 4.92 | 23.24 |
| CDM [ 15 ] | 28.28 | 35.66 | 20.42 | 24.56 | 29.27 | 17.62 | 9.78 | 11.80 | 5.52 | 20.87 |
| Input Depth | Success Rate | Completion Rate |
| Raw | 54.42% 9.49 | 88.00% 6.50 |
| D 3 RoMa (RGB-D-Stereo) | 69.05% 6.99 | 96.00% 3.50 |
| CDM (RGB-D) | 72.73% 6.07 | 98.00% 2.50 |
| Ours (RGB-D) | 76.65% 5.12 | 99.00% 1.50 |
| Ours + Unc-Reweighting (RGB-D) | 71.08% 6.74 | 94.50% 4.00 |
Appendix figures & tables25 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | , 1mm | , 2mm | , 5mm |
| YCB-V | 0.0111 / 0.0100 | 0.0190 / 0.0170 | 0.0356 / 0.0314 |
| T-LESS | 0.0136 / 0.0119 | 0.0219 / 0.0191 | 0.0385 / 0.0326 |
| Split | Dataset | #Cat. | Canonical Categories |
| Train | DREDS-CatKnown | 15 | airplane, bag, bottle, bowl, camera, can, car, clock, faucet, guitar, hat, knife, lamp, mug, phone |
| Train | HISS | 11 | airplane, bag, bottle, bowl, camera, can, car, guitar, hat, mug, phone |
| Train | ClearPose | 20 | beaker, bottle, bowl, container, cup, drinking glass, dropper, flask, fork, funnel, graduated cylinder, knife, mug, pan, pitcher, plate, reagent bottle, rod, spoon, syringe |
| Test | HAMMER | 12 | bottle, box, can, cup, jar, knife, remote control, shoe, spoon, teapot, trophy, tube |
| Test | HouseCat6D | 12 | bottle, box, can, cup, drinking glass, fork, knife, remote control, shoe, spoon, teapot, tube |
| Test | TransCG | 10 | bottle, bowl, can, cup, drinking glass, fruit dish, jelly container, soap container, spoon, wash container |
| Dataset | #Obj. | Exact Overlap | Category Cov. | Material Cov. | Category–Material Cov. |
| HAMMER | 20 | 0/20 | 7/20 (35.0%) | 20/20 (100.0%) | 4/20 (20.0%) |
| HouseCat6D | 50 | 0/50 | 25/50 (50.0%) | 50/50 (100.0%) | 17/50 (34.0%) |
| TransCG | 46 | 0/46 | 33/46 (71.7%) | 46/46 (100.0%) | 29/46 (63.0%) |
| GraspNet-Trans | 76 | 0/76 | 15/76 (19.7%) | 76/76 (100.0%) | 10/76 (13.2%) |
| YCB-V | 21 | 0/21 | 9/21 (42.9%) | 21/21 (100.0%) | 8/21 (38.1%) |
| T-LESS | 30 | 0/30 | 0/30 (0.0%) | 30/30 (100.0%) | 0/30 (0.0%) |
| Method | RMSE | REL | MAE | |||
| SwinDR [ 11 ] | 0.0503 | 0.0403 | 0.0321 | 74.23 | 83.61 | 98.70 |
| [ 14 ] | 0.0592 | 0.0544 | 0.0348 | 68.43 | 84.95 | 96.81 |
| DP [ 36 ] | 0.0484 | 0.0445 | 0.0271 | 70.59 | 87.95 | 98.64 |
| OGNI-DC [ 19 ] | 0.0506 | 0.0515 | 0.0302 | 67.35 | 86.86 | 98.32 |
| PromptDA [ 21 ] | 0.0500 | 0.0532 | 0.0308 | 63.05 | 86.84 | 98.42 |
| PPFT [ 58 ] | 0.0389 | 0.0435 | 0.0241 | 78.36 | 87.76 | 98.80 |
| Methods | RMSE | REL | MAE | |||
| NLSPN [ 35 ] | 0.026 | 0.039 | 0.015 | 78.90 | 89.02 | 97.86 |
| LIDF [ 59 ] | 0.082 | 0.183 | 0.069 | 23.70 | 42.77 | 75.44 |
| SwinDR [ 11 ] | 0.022 | 0.034 | 0.013 | 81.90 | 92.18 | 98.39 |
| RoMa [ 14 ] | 0.0247 | 0.0358 | 0.0141 | 79.56 | 89.94 | 98.29 |
| PromptDA [ 21 ] | 0.0463 | 0.0983 | 0.0365 | 42.79 | 66.71 | 91.75 |
| Ours | 0.0220 | 0.0322 | 0.0123 | 82.30 | 91.82 | 98.37 |
| Method | HAMMER | TransCG | GraspNet-Trans | XYZ-IBD | ROBI | Mean Rank | |||||
| REL | REL | REL | REL | REL | |||||||
| SwinDR [ 11 ] | 0.0492 | 72.81 | 0.0138 | 94.26 | 0.0438 | 85.96 | 0.0283 | 84.01 | 0.0099 | 95.05 | 3.3 |
| [ 14 ] | 0.0804 | 54.46 | 0.0330 | 81.70 | 0.0546 | 81.05 | 0.0664 | 50.28 | 0.0347 | 76.97 | 5.4 |
| PromptDA [ 21 ] | 0.3642 | 33.42 | 0.1250 | 54.73 | 0.2083 | 56.18 | 0.2998 | 32.94 | 0.2000 | 67.97 | 7.0 |
| OMNI-DC [ 20 ] | 0.0665 | 68.65 | 0.0121 | 95.54 | 0.0752 | 81.35 | 0.0055 | 98.56 | 0.0057 | 99.52 | 2.8 |
| PriorDA [ 22 ] | 0.0670 | 49.79 | 0.0371 | 74.70 | 0.0690 | 76.05 | 0.0312 | 82.69 | 0.0286 | 79.63 | 5.4 |
| Dataset | Method | RMSE | REL | |
| XYZ-IBD | CDM-D435 | 0.0128 | 0.0145 | 96.60 |
| CDM-L515 | 0.0146 | 0.0172 | 96.47 | |
| CDM-Kinect | 0.0163 | 0.0182 | 93.73 | |
| Ours | 0.0067 | 0.0063 | 99.25 | |
| YCB-V | CDM-D435 | 0.1216 | 0.0575 | 67.17 |
| CDM-L515 | 0.1202 | 0.0501 | 75.19 |
| Sensor | Method | RMSE | REL | |
| D435 (IR Stereo) | OMNI-DC | 0.0616 | 0.0586 | 67.47 |
| CDM-D435 | 0.0497 | 0.0427 | 77.95 | |
| Ours | 0.0338 | 0.0324 | 82.96 | |
| L515 (D-ToF) | OMNI-DC | 0.0661 | 0.0655 | 66.61 |
| CDM-L515 | 0.0438 | 0.0407 | 77.48 | |
| Ours | 0.0387 | 0.0320 | 83.83 |
| Method | Input Res. | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | ||||||||||
| SwinDR | 518 518 | 0.0429 | 76.35 | 0.0463 | 76.84 | 0.0307 | 81.10 | 0.1078 | 61.75 | 0.0241 | 88.82 | 0.0151 | 93.53 | 0.0247 | 85.19 | 0.0318 | 77.95 |
| D 3 RoMa | 518 518 | 0.0655 | 58.43 | 0.0498 | 76.45 | 0.0419 | 67.56 | 0.1231 | 50.27 | 0.0494 | 81.61 | 0.0511 | 73.36 | 0.0650 | 47.27 | 0.0575 | 50.07 |
| Ours | 518 518 | 0.0324 | 82.96 | 0.0345 | 86.09 | 0.0193 | 89.66 | 0.0677 | 67.20 | 0.0259 | 94.35 | 0.0143 | 95.83 | 0.0063 | 99.25 | 0.0096 | 99.12 |
| SwinDR | 360 640 | 0.0394 | 77.85 | 0.0441 | 76.90 | 0.0242 | 83.30 | 0.1017 | 62.09 | 0.0169 | 90.62 | 0.0134 | 93.87 | 0.0257 | 82.34 | 0.0393 | 66.71 |
| D 3 RoMa | 360 640 | 0.0393 | 75.16 | 0.0414 | 79.59 | 0.0505 | 58.26 | 0.0744 | 60.34 | 0.0625 | 73.53 | 0.0453 | 77.02 | 0.0869 | 28.59 | 0.0895 | 24.07 |
| Method | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | |||||||||||
| SwinDR | 0.0401 | 75.53 | 0.0503 | 74.77 | 0.0200 | 90.41 | 0.0815 | 64.81 | 0.0182 | 91.41 | 0.0175 | 92.35 | 0.0255 | 84.72 | 0.0346 | 74.21 | 3.1 | -0.06 |
| 0.0393 | 75.18 | 0.0414 | 79.59 | 0.0505 | 58.26 | 0.0744 | 60.34 | 0.0625 | 73.53 | 0.0453 | 77.02 | 0.0869 | 28.59 | 0.0895 | 24.07 | 4.8 | -0.50 | |
| PromptDA | 0.5383 | 6.70 | 0.3792 | 20.16 | 0.2960 | 14.69 | 0.3474 | 10.99 | 0.4364 | 14.96 | 0.1390 | 59.47 | 0.3373 | 33.47 | 0.7275 | 6.27 | 6.9 | -0.06 |
| OMNI-DC | 0.0585 | 67.84 | 0.0523 | 72.14 | 0.0356 | 79.14 | 0.1640 | 51.88 | 0.0214 | 94.59 | 0.0253 | 91.53 | 0.0047 | 99.03 | 0.0065 | 99.34 | 3.3 | +0.25 |
| PriorDA | 0.0762 | 41.41 | 0.0720 | 45.51 | 0.0418 | 71.36 | 0.1492 | 46.55 | 0.0338 | 87.47 | 0.0420 | 73.41 | 0.0205 | 95.26 | 0.0182 | 95.15 | 5.0 | +0.06 |
| Model Variant | Unc. Sup. | Feedback Signal | Fusion Type |
| Single-step depth | – | – | – |
| Single-step depth + Unc. | ✓ | – | – |
| Plain refinement | – | – | – |
| Uncertainty-only | ✓ | – | – |
| Static learned gate | ✓ | Learned static | Gate |
| Validity-mask gate | ✓ | Valid mask | Gate |
| Model Variant | Object Pixels | All-valid Pixels | ||||
| REL | REL-AUSE | REL | REL-AUSE | |||
| Single-step depth | 0.0341 | 80.60 | – | 0.0324 | 83.67 | – |
| Single-step depth + Unc. | 0.0372 | 76.19 | 0.00152 | 0.0346 | 80.95 | 0.00105 |
| Plain refinement | 0.0320 | 82.55 | – | 0.0259 | 88.15 | – |
| Uncertainty-only | 0.0340 | 80.16 | 0.00132 | 0.0316 | 83.85 | 0.00111 |
| Static learned gate | 0.0292 | 85.53 | 0.00117 | 0.0264 | 87.69 | 0.00097 |
| ADF+Robust-SN | UCFC | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | |||||||||||
| – | – | 0.0372 | 78.83 | 0.0355 | 84.25 | 0.0214 | 87.69 | 0.0773 | 64.53 | 0.0227 | 92.64 | 0.0161 | 95.17 | 0.0060 | 98.51 | 0.0106 | 98.32 | 3.41 |
| ✓ | 0.0349 | 81.50 | 0.0347 | 85.16 | 0.0206 | 88.56 | 0.0646 | 69.42 | 0.0234 | 94.06 | 0.0145 | 95.35 | 0.0054 | 99.50 | 0.0108 | 98.59 | 1.94 | |
| ✓ | – | 0.0334 | 80.56 | 0.0345 | 84.62 | 0.0208 | 87.70 | 0.0665 | 68.32 | 0.0230 | 91.44 | 0.0153 | 95.32 | 0.0060 | 99.43 | 0.0113 | 98.01 | 2.78 |
| ✓ | 0.0326 | 82.38 | 0.0334 | 86.86 | 0.0199 | 89.52 | 0.0701 | 67.28 | 0.0262 | 93.94 | 0.0137 | 95.73 | 0.0069 | 98.35 | 0.0103 | 98.74 | 1.88 | |
| Model Variant | HAMMER | HouseCat6D | TransCG | GraspNet-Trans | YCB-V | T-LESS | XYZ-IBD | ROBI | Mean Rank | ||||||||
| REL | REL | REL | REL | REL | REL | REL | REL | ||||||||||
| Single-step depth | 0.0482 | 70.02 | 0.0396 | 75.75 | 0.0259 | 83.09 | 0.0779 | 62.12 | 0.0409 | 72.35 | 0.0223 | 89.22 | 0.0094 | 98.68 | 0.0147 | 95.83 | 6.38 |
| Single-step depth + Unc. | 0.0509 | 73.96 | 0.0458 | 81.26 | 0.0261 | 86.16 | 0.0808 | 61.82 | 0.0423 | 83.86 | 0.0248 | 92.30 | 0.0083 | 99.12 | 0.0155 | 96.52 | 6.19 |
| Plain refinement | 0.0365 | 78.05 | 0.0346 | 85.14 | 0.0221 | 86.98 | 0.0566 | 72.23 | 0.0295 | 91.30 | 0.0149 | 95.33 | 0.0090 | 97.87 | 0.0124 | 98.17 | 4.19 |
| Residual gate | 0.0393 | 75.92 | 0.0391 | 81.85 | 0.0228 | 86.24 | 0.0682 | 66.67 | 0.0267 | 93.31 | 0.0148 | 94.67 | 0.0065 | 99.38 | 0.0116 | 98.48 | 3.88 |
| Simple conv fusion | 0.0351 | 82.36 | 0.0330 | 85.80 | 0.0214 | 87.75 | 0.0727 | 64.39 | 0.0223 | 94.11 | 0.0144 | 95.07 | 0.0057 | 99.59 | 0.0114 | 98.47 | 2.50 |
| Failure mode | Failed Pixels (% Obj.) | CW Rate |
| Biased sensor | 8.4% | 15.3% |
| Missing sensor, inaccurate init. | 4.5% | 7.0% |
| Missing sensor, accurate init. | 0.2% | 4.4% |
| Reliable sensor, prediction error | 3.8% | 9.9% |
| All failed pixels | 16.9% | 11.7% |
| Condition | Condition Pixels | Failed Pixels | Mean REL | Improved | Degraded | ||
| UCFC-PM | UCFC | UCFC-PM | UCFC | ||||
| Biased sensor | 19.1% | 10.3% | 8.4% | 0.0611 | 0.0546 | 4.4% | 2.1% |
| Missing sensor, inaccurate init. | 16.3% | 5.5% | 4.5% | 0.0473 | 0.0453 | 3.8% | 2.4% |
| Missing sensor, accurate init. | 2.0% | 0.2% | 0.2% | 0.0252 | 0.0267 | 0.2% | 0.2% |
| Reliable sensor | 62.7% | 2.9% | 3.8% | 0.0208 | 0.0212 | 2.5% | 3.1% |
| All object pixels | 100.0% | 18.8% | 16.9% | 0.0345 | 0.0326 | 10.8% | 7.8% |
| Train Depth | Test Depth | IoU@25 | IoU@50 | IoU@75 | 2cm | 5cm | 2cm | 5cm |
| Raw | Raw | 77.3 | 52.6 | 17.2 | 6.7 | 8.9 | 17.6 | 25.4 |
| Raw | Restored | 77.9 | 52.2 | 11.4 | 4.4 | 7.1 | 12.4 | 21.9 |
| Raw | Restored+Unc-Sampling | 78.4 | 52.3 | 11.0 | 4.2 | 7.1 | 12.0 | 22.1 |
| GT | Raw | 60.6 | 27.0 | 3.0 | 1.2 | 2.8 | 3.6 | 8.0 |
| GT | Restored | 60.7 | 25.5 | 3.3 | 2.7 | 4.2 | 6.6 | 11.9 |
| GT | Restored+Unc-Sampling | 69.0 | 43.9 | 13.7 | 4.3 | 5.6 | 13.4 | 18.1 |
| Depth Type | Seen | Similar | Novel | Mean | ||||||||
| AP | CR | CR | AP | CR | CR | AP | CR | CR | AP | CR | CR | |
| Ours (Restored) | 32.67 | 47.32 | 58.74 | 25.48 | 56.32 | 62.87 | 11.26 | 64.31 | 65.10 | 23.14 | 55.98 | 62.23 |
| Ours (Restored+Unc - Sampling) | 31.65 | 48.60 | 59.84 | 24.58 | 57.73 | 63.62 | 10.55 | 65.76 | 61.81 | 22.26 | 57.36 | 61.76 |
| Ours (Restored+Unc - Reweighting) | 33.54 | 45.44 | 58.41 | 26.62 | 53.98 | 62.51 | 11.89 | 62.36 | 64.42 | 24.02 | 53.93 | 61.78 |
| Input Depth | SR (%) | CR (%) |
| Raw | 54.42 [45.33, 64.31] | 88.00 [81.00, 94.00] |
| D 3 RoMa (RGB-D-Stereo) | 69.05 [62.36, 76.34] | 96.00 [92.00, 99.00] |
| CDM (RGB-D) | 72.73 [66.55, 78.69] | 98.00 [95.00, 100.00] |
| Ours (RGB-D) | 76.65 [71.57, 81.82] | 99.00 [97.00, 100.00] |
| Ours + Unc-Reweighting (RGB-D) | 71.08 [64.21, 77.69] | 94.50 [90.00, 98.00] |
| Comparison | SR (pp) | CR (pp) |
| Ours (RGB-D) Raw | +22.23 [12.00, 32.12] | +11.00 [5.00, 18.00] |
| Ours (RGB-D) D 3 RoMa (RGB-D-Stereo) | +7.60 [-0.15, 14.97] | +3.00 [-1.00, 7.00] |
| Ours (RGB-D) CDM (RGB-D) | +3.92 [-3.09, 12.25] | +1.00 [0.00, 3.00] |
| Ours (RGB-D) Ours + Unc-Reweighting (RGB-D) | +5.56 [-3.37, 13.33] | +4.50 [0.00, 9.50] |
| Scene | Raw | D 3 RoMa | CDM | Ours | Ours + Unc. |
| 01 | 6/5/5 | 7/4/5 | 7/5/5 | 8/5/5 | 9/4/5 |
| 02 | 11/3/5 | 8/5/5 | 6/5/5 | 6/5/5 | 6/5/5 |
| 03 | 6/5/5 | 6/5/5 | 6/5/5 | 6/5/5 | 6/5/5 |
| 04 | 12/4/5 | 8/5/5 | 6/5/5 | 7/5/5 | 8/4/5 |
| 05 | 6/5/5 | 6/5/5 | 6/5/5 | 7/5/5 | 6/3/4 |
| 06 | 10/3/5 | 8/4/5 | 8/5/5 | 7/5/5 | 6/5/5 |
Depth Estimation
Depth estimation infers scene distance from images and sensors, enabling 3D reconstruction, navigation, and spatial understanding.
See all 201 papers