arXiv:2409.09144cs.CV2024-09被引 17

用单步扩散生成图像预表示,实现高效零样本深度估计

PrimeDepth: Efficient Monocular Depth Estimation with a Stable Diffusion Preimage

  • 仅用一次去噪步骤提取稳定扩散模型的冻结图像特征
  • 测试速度比Marigold快100倍,且在复杂场景更鲁棒
  • 适合追求高效与高细节深度图的工业应用

本文解决零样本单目深度估计问题。近期进展利用如Stable Diffusion等文本到图像基础模型,因其丰富的通用图像表征,仅需少量训练数据即可转化为高精度深度估计模型。然而,现有方法因依赖迭代去噪过程,推理效率极低。为此,本文提出PrimeDepth,通过运行一次去噪步骤从Stable Diffusion中提取丰富但固定的图像表示(称作preimage),再输入具结构先验的精炼网络完成下游任务。实验表明,PrimeDepth比领先方法Marigold快两个数量级,且在挑战性场景下更具鲁棒性,定量性能略优。尽管仍略逊于数据驱动方法Depth Anything(需20倍标注数据),但其预测更精细。二者结合可超越各自表现,达到新基准。未来数据驱动方法亦可借鉴此预表示思想。

原文摘要 · Abstract (English)

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.

深度估计扩散模型高效推理零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。