RealmDreamer: Text-Driven 3D Scene Generation with Inpainting and Depth Diffusion

TL;DR

RealmDreamer leverages inpainting and depth diffusion for text-driven 3D scene synthesis, achieving 95.5% user preference in evaluations.

cs.CV 🔴 Advanced 2024-04-11 39 views
Jaidev Shriram Alex Trevithick Lingjie Liu Ravi Ramamoorthi
3D generation diffusion models deep learning image inpainting scene synthesis

Key Findings

Methodology

The approach combines 3D Gaussian splatting with pretrained diffusion models, employing a multi-stage pipeline: scene initialization using monocular depth and scene estimates, 2D inpainting conditioned on partial views for missing regions, and depth diffusion for geometric fidelity. Scene initialization leverages text-to-image models and monocular depth liftings, followed by inpainting distillation that reduces variance in unknown regions. Geometric refinement uses depth diffusion to align 3D geometry with appearance. The process avoids multi-view or video data, relying solely on single images and text prompts, with iterative optimization ensuring high-fidelity results.

Key Results

  • Quantitative metrics show superior performance: CLIP score of 31.69, depth Pearson correlation of 0.89, and user preference of 95.5%. Compared to ProlificDreamer (88%) and LucidDreamer, RealmDreamer produces more coherent, detailed, and realistic 3D scenes across various styles and layouts. Ablation studies confirm the importance of scene initialization, low-variance inpainting, and depth distillation for quality.
  • In single-image to 3D synthesis, the method effectively fills occluded regions, producing consistent geometry and appearance comparable to multi-view trained models. User studies and quantitative scores validate its robustness, especially in complex scenes with diverse layouts and styles.
  • The approach does not require multi-view data, enabling rapid synthesis from a single image. Multi-stage optimization ensures geometric accuracy and visual fidelity, outperforming existing baselines in both qualitative and quantitative assessments.

Significance

This work addresses the longstanding challenge of high-quality 3D scene generation without multi-view or video data. By integrating pretrained 2D diffusion models with geometric distillation, it overcomes limitations of prior methods that suffered from poor geometry and lack of detail. The ability to generate complex, realistic scenes from a single image or text prompt significantly broadens application potential in virtual reality, gaming, and robotics. Its generalization to diverse styles and layouts demonstrates a major step forward in the field, reducing reliance on extensive 3D datasets and multi-view setups.

Technical Contribution

The main innovations include repurposing 2D inpainting diffusion models conditioned on scene estimates for low-variance supervision, and integrating monocular depth diffusion for geometric fidelity. The multi-stage pipeline—initialization, inpainting distillation, depth refinement, and fine-tuning—ensures high-quality, coherent 3D reconstructions. The method avoids training on 3D data or fine-tuning diffusion models, offering a flexible, plug-and-play framework that enhances existing 3D Gaussian splatting techniques with diffusion-based priors.

Novelty

This is the first work to utilize pre-trained 2D inpainting diffusion models as scene-aware priors for 3D scene synthesis, combined with depth diffusion for geometric accuracy. Unlike prior methods relying on multi-view or video data, it achieves high-quality results from a single image and text prompt, representing a significant leap in efficiency and applicability. Its multi-stage optimization and low-variance supervision introduce new paradigms in text-to-3D generation.

Limitations

  • The process requires several hours per scene, limiting real-time applications. Handling highly complex scenes with severe occlusion still results in blurry or incomplete reconstructions. Dependence on monocular depth models can introduce errors in geometry, especially in textureless or ambiguous regions. Future work should focus on speeding up inference and improving geometric detail in challenging scenarios.
  • Current results are limited in dynamic or highly detailed environments. The method's reliance on scene initialization means poor initial estimates can degrade final quality. Extending to full 360-degree scenes and reducing computational costs remain open challenges.

Future Work

Future directions include accelerating the pipeline with more efficient diffusion models, extending to full 360-degree scene generation, and integrating multi-modal cues like audio or tactile data. Improving geometric accuracy in complex scenes and reducing computational costs will be key. Additionally, exploring real-time applications and broader generalization to outdoor environments are promising avenues.

AI Executive Summary

Generating realistic 3D scenes from text prompts has long been a goal in computer graphics and AI, yet existing methods often rely on multi-view data, videos, or extensive 3D datasets, limiting their practicality. In this context, RealmDreamer introduces a novel framework that leverages pretrained 2D diffusion models—specifically inpainting models conditioned on partial views—and monocular depth diffusion to synthesize high-fidelity 3D scenes from a single image or text prompt.

The core innovation lies in a multi-stage pipeline: starting with scene initialization via monocular depth liftings and scene estimates, followed by inpainting distillation that fills missing regions with low variance supervision, and finally geometric refinement using depth diffusion. This process circumvents the need for multi-view or video data, making it highly flexible and scalable. The scene initialization employs a text-to-image model to generate a reference image, which is then lifted to 3D using monocular depth models, with outpainting to expand the scene. The inpainting stage conditions on the current scene estimate, filling occluded or disoccluded regions plausibly, guided by a loss that minimizes variance and perceptual differences.

Experimental results demonstrate that RealmDreamer outperforms state-of-the-art approaches such as ProlificDreamer and LucidDreamer, achieving a user preference rate of 95.5%. Quantitative metrics, including CLIP scores and depth correlation, confirm the high quality of generated scenes. Notably, the method can produce diverse styles and complex layouts, even from a single image, with results validated through extensive user studies and ablation experiments.

This work significantly advances the field by removing the dependency on multi-view data, enabling rapid, high-quality scene synthesis from minimal input. Its implications span virtual reality, gaming, and robotics, where flexible and scalable scene generation is crucial. Future efforts will focus on reducing computational costs, extending to full 360-degree environments, and integrating multi-modal cues, pushing the boundary of what AI can achieve in automatic scene creation.

Deep Dive

Abstract

We introduce RealmDreamer, a technique for generating forward-facing 3D scenes from text descriptions. Our method optimizes a 3D Gaussian Splatting representation to match complex text prompts using pretrained diffusion models. Our key insight is to leverage 2D inpainting diffusion models conditioned on an initial scene estimate to provide low variance supervision for unknown regions during 3D distillation. In conjunction, we imbue high-fidelity geometry with geometric distillation from a depth diffusion model, conditioned on samples from the inpainting model. We find that the initialization of the optimization is crucial, and provide a principled methodology for doing so. Notably, our technique doesn't require video or multi-view data and can synthesize various high-quality 3D scenes in different styles with complex layouts. Further, the generality of our method allows 3D synthesis from a single image. As measured by a comprehensive user study, our method outperforms all existing approaches, preferred by 88-95%. Project Page: https://realmdreamer.github.io/

cs.CV cs.AI cs.GR cs.LG