cs.CVNov 1, 2026

UA-Depth: Uncertainty-Aware Depth Restoration for Material-Agnostic Object Manipulation

Authors: Rui Cao, Jia-Hui Pan, Yin Zhang, Qiwei Meng, Jiapeng Li, Gang Wang, Runsong Zhu, Wei Chen, 

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

Appendix figures & tables25 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. AISPO: Enhancing Depth Reliability for Robotic Manipulation of Non-Lambertian Objects via Affine-Invariant Shape Prior

    Jun 24, 2026Zhiming Chen, Linfang Zheng, Kun Zhang +4Depth CompletionRobotic Manipulation

  2. AnchorD: Metric Grounding of Monocular Depth Using Factor Graphs

    May 4, 2026Simon Dorer, Martin Büchner, Nick Heppert +1Metric Depth EstimationDepth Estimation

  3. UfM*: Uncertainty from Motion* for DNN Depth Estimation Using Gaussians

    May 21, 2026Soumya Sudhakar, Sertac Karaman, Vivienne SzeUncertainty QuantificationDepth Estimation