Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Lotus-2: Advancing Geometric Dense Prediction with Powerful Image Generative Model

About

Recovering pixel-wise geometric properties from a single image is fundamentally ill-posed due to appearance ambiguity and non-injective mappings between 2D observations and 3D structures. While discriminative regression models achieve strong performance through large-scale supervision, their success is bounded by the scale, quality, and diversity of available data, as well as by limited physical reasoning. Recent diffusion models exhibit powerful world priors that encode geometry and semantics learned from massive image-text data, yet directly reusing their stochastic generative formulation is suboptimal for deterministic geometric inference: the former is optimized for diverse and high-fidelity image generation, whereas the latter requires stable and accurate predictions. In this work, we propose Lotus-2, a two-stage deterministic framework for stable, accurate and fine-grained geometric dense prediction, aiming to provide an optimal adaptation protocol to fully exploit the pre-trained generative priors. Specifically, in the first stage, the core predictor employs a single-step deterministic formulation with a clean-data objective and a lightweight local continuity module (LCM) to generate globally coherent structures without grid artifacts. In the second stage, the detail sharpener performs a constrained multi-step rectified-flow refinement within the manifold defined by the core predictor, enhancing fine-grained geometry through noise-free deterministic flow matching. Using only 59K training samples, less than 1% of existing large-scale datasets, Lotus-2 establishes new state-of-the-art results in monocular depth estimation and highly competitive surface normal prediction. These results demonstrate that diffusion models can serve as deterministic world priors, enabling high-quality geometric reasoning beyond traditional discriminative and generative paradigms.

Jing He, Haodong Li, Mingzhi Sheng, Ying-Cong Chen• 2025

Related benchmarks

TaskDatasetResultRank
Surface Normal PredictionNYU V2
Mean Error16.9
123
Surface Normal EstimationReal-world Average
MAE (°)19.642
14
Surface Normal EstimationScanNet
Mean Angle Error14.2
5
Surface Normal EstimationIndoor Average
Mean Error16.558
5
Surface Normal EstimationDIODE Indoor
Mean Error18.575
5
Surface Normal EstimationvKITTI
Mean Error28.894
3
Showing 6 of 6 rows

Other info

GitHub

Follow for update