Enhancing Monocular Height Estimation via Weak Supervision from Imperfect Labels
Sining Chen, Yilei Shi, Xiao Xiang Zhu
TL;DR
This work tackles monocular height estimation under label scarcity and domain shift by training with imperfect, out-of-domain labels. It introduces an ensemble-based pipeline with a shared encoder and quality-specific branches, a domain classifier to weigh outputs, and loss terms including a balanced soft height loss $L_{BSH}$ and ordinal constraints $L_{OC}$, complemented by a ground truth augmentation module. The method demonstrates that training with mid-/low-quality labels yields better cross-domain generalization, achieving up to 22.94% RMSE reduction on DFC23 and 18.62% on GBH over strong baselines, with ablations confirming each component's contribution. The approach offers a practical path to scale monocular height estimation to regions lacking LiDAR data, with robust performance across diverse urban contexts.
Abstract
Monocular height estimation provides an efficient and cost-effective solution for three-dimensional perception in remote sensing. However, training deep neural networks for this task demands abundant annotated data, while high-quality labels are scarce and typically available only in developed regions, which limits model generalization and constrains their applicability at large scales. This work addresses the problem by leveraging imperfect labels from out-of-domain regions to train pixel-wise height estimation networks, which may be incomplete, inexact, or inaccurate compared to high-quality annotations. We introduce an ensemble-based pipeline compatible with any monocular height estimation network, featuring architecture and loss functions specifically designed to leverage information in noisy labels through weak supervision, utilizing balanced soft losses and ordinal constraints. Experiments on two datasets -- DFC23 (0.5--1 m) and GBH (3 m) -- show that our method achieves more consistent cross-domain performance, reducing average RMSE by up to 22.94% on DFC23 and 18.62% on GBH compared with baselines. Ablation studies confirm the contribution of each design component.
