Self-supervised Monocular Depth Estimation Robust to Reflective Surface Leveraged by Triplet Mining
Wonhyeok Choi, Kyumin Hwang, Wei Peng, Minwoo Choi, Sunghoon Im
TL;DR
This work tackles the failure of self-supervised monocular depth estimation on reflective/non-Lambertian surfaces by introducing a reflection-aware training paradigm. It localizes reflective regions via triplet mining that leverages cross-view photometric discrepancies, and uses a reflection-aware loss to suppress contaminated gradients in those regions. A two-teacher distillation scheme further preserves high-frequency depth details in non-reflective areas while enhancing reflective-surface accuracy. Across indoor and outdoor benchmarks, including ScanNet, KITTI, NYU-v2, and cross-dataset tests, the proposed approach yields robust depth estimates on reflective surfaces with minimal sacrifice to non-reflective regions, marking a notable step toward reliable, plug-in depth learning for real-world scenes.
Abstract
Self-supervised monocular depth estimation (SSMDE) aims to predict the dense depth map of a monocular image, by learning depth from RGB image sequences, eliminating the need for ground-truth depth labels. Although this approach simplifies data acquisition compared to supervised methods, it struggles with reflective surfaces, as they violate the assumptions of Lambertian reflectance, leading to inaccurate training on such surfaces. To tackle this problem, we propose a novel training strategy for an SSMDE by leveraging triplet mining to pinpoint reflective regions at the pixel level, guided by the camera geometry between different viewpoints. The proposed reflection-aware triplet mining loss specifically penalizes the inappropriate photometric error minimization on the localized reflective regions while preserving depth accuracy in non-reflective areas. We also incorporate a reflection-aware knowledge distillation method that enables a student model to selectively learn the pixel-level knowledge from reflective and non-reflective regions. This results in robust depth estimation across areas. Evaluation results on multiple datasets demonstrate that our method effectively enhances depth quality on reflective surfaces and outperforms state-of-the-art SSMDE baselines.
