让扩散模型直接预测标注,提升密集预测的精度与速度。
Lotus: Diffusion-based Visual Foundation Model for High-quality Dense Prediction
- 直接预测标签而非噪声,减少有害方差。
- 单步扩散过程,推理速度显著提升。
- 无需增大数据或模型规模,零样本表现领先。
利用预训练文生图扩散模型的视觉先验,可有效提升密集预测任务中的零样本泛化能力。然而,现有方法往往盲目沿用图像生成的原始扩散范式,因二者本质差异而效果不佳。本文系统分析了扩散模型在密集预测中的适用性,发现图像生成中学习噪声的参数化方式对密集预测有害,多步加噪/去噪过程也冗余且难优化。基于此,我们提出Lotus,一种针对密集预测设计的简化扩散基础模型。Lotus直接预测标注而非噪声,避免方差干扰;将扩散过程重构为单步操作,简化优化并大幅提速。此外,引入新颖的细节保留调优策略,实现更精细的预测。不增加训练数据或模型容量,Lotus在多个数据集上实现深度与法向量估计的零样本性能新纪录,并显著优于多数现有扩散方法。其高效高质特性还可支持联合估计、单/多视图三维重建等实际应用。
原文摘要 · Abstract (English)
Leveraging the visual priors of pre-trained text-to-image diffusion models offers a promising solution to enhance zero-shot generalization in dense prediction tasks. However, existing methods often uncritically use the original diffusion formulation, which may not be optimal due to the fundamental differences between dense prediction and image generation. In this paper, we provide a systemic analysis of the diffusion formulation for the dense prediction, focusing on both quality and efficiency. And we find that the original parameterization type for image generation, which learns to predict noise, is harmful for dense prediction; the multi-step noising/denoising diffusion process is also unnecessary and challenging to optimize. Based on these insights, we introduce Lotus, a diffusion-based visual foundation model with a simple yet effective adaptation protocol for dense prediction. Specifically, Lotus is trained to directly predict annotations instead of noise, thereby avoiding harmful variance. We also reformulate the diffusion process into a single-step procedure, simplifying optimization and significantly boosting inference speed. Additionally, we introduce a novel tuning strategy called detail preserver, which achieves more accurate and fine-grained predictions. Without scaling up the training data or model capacity, Lotus achieves SoTA performance in zero-shot depth and normal estimation across various datasets. It also enhances efficiency, being significantly faster than most existing diffusion-based methods. Lotus' superior quality and efficiency also enable a wide range of practical applications, such as joint estimation, single/multi-view 3D reconstruction, etc. Project page: https://lotus3d.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。