PrimeDepth: Efficient Monocular Depth Estimation with a Stable Diffusion Preimage

TL;DR

PrimeDepth introduces an efficient, diffusion-inspired monocular depth estimation method that leverages a single denoising step to produce detailed depth maps with superior robustness and speed, advancing zero-shot depth estimation.

Contribution

It proposes PrimeDepth, a novel approach that extracts a rich image representation from Stable Diffusion with one denoising step, significantly improving efficiency and robustness over prior diffusion-based methods.

Findings

PrimeDepth is two orders of magnitude faster than Marigold.

It is more robust in challenging scenarios.

It achieves marginally better quantitative results than Marigold.

Abstract

This work addresses the task of zero-shot monocular depth estimation. A recent advance in this field has been the idea of utilising Text-to-Image foundation models, such as Stable Diffusion. Foundation models provide a rich and generic image representation, and therefore, little training data is required to reformulate them as a depth estimation model that predicts highly-detailed depth maps and has good generalisation capabilities. However, the realisation of this idea has so far led to approaches which are, unfortunately, highly inefficient at test-time due to the underlying iterative denoising process. In this work, we propose a different realisation of this idea and present PrimeDepth, a method that is highly efficient at test time while keeping, or even enhancing, the positive aspects of diffusion-based approaches. Our key idea is to extract from Stable Diffusion a rich, but frozen, image representation by running a single denoising step. This representation, we term preimage, is then fed into a refiner network with an architectural inductive bias, before entering the downstream task. We validate experimentally that PrimeDepth is two orders of magnitude faster than the leading diffusion-based method, Marigold, while being more robust for challenging scenarios and quantitatively marginally superior. Thereby, we reduce the gap to the currently leading data-driven approach, Depth Anything, which is still quantitatively superior, but predicts less detailed depth maps and requires 20 times more labelled data. Due to the complementary nature of our approach, even a simple averaging between PrimeDepth and Depth Anything predictions can improve upon both methods and sets a new state-of-the-art in zero-shot monocular depth estimation. In future, data-driven approaches may also benefit from integrating our preimage.