ORIGEN: Zero-Shot 3D Orientation Grounding in Text-to-Image Generation
Yunhong Min, Daehyeon Choi, Kyeongmin Yeo, Jihyun Lee, Minhyuk Sung
TL;DR
ORIGEN addresses zero-shot 3D orientation grounding in text-to-image generation for open-vocabulary, multi-object scenes by framing the problem as test-time reward-guided sampling with a one-step generative model. It defines a reward from GroundingDINO and OrientAnything and solves for latent vectors via Langevin dynamics, augmented with an adaptive time-rescaling mechanism to preserve realism while accelerating convergence. Empirical results on ORIBENCH-Single and ORIBENCH-Multi show ORIGEN achieving superior orientation grounding and strong text-to-image fidelity, supported by a user study endorsing its quality. The solution provides a practical, training-free pathway for precise 3D pose control in real-world, diverse image synthesis tasks, with broad applicability to layout-to-image and depth-guided generation as well.
Abstract
We introduce ORIGEN, the first zero-shot method for 3D orientation grounding in text-to-image generation across multiple objects and diverse categories. While previous work on spatial grounding in image generation has mainly focused on 2D positioning, it lacks control over 3D orientation. To address this, we propose a reward-guided sampling approach using a pretrained discriminative model for 3D orientation estimation and a one-step text-to-image generative flow model. While gradient-ascent-based optimization is a natural choice for reward-based guidance, it struggles to maintain image realism. Instead, we adopt a sampling-based approach using Langevin dynamics, which extends gradient ascent by simply injecting random noise--requiring just a single additional line of code. Additionally, we introduce adaptive time rescaling based on the reward function to accelerate convergence. Our experiments show that ORIGEN outperforms both training-based and test-time guidance methods across quantitative metrics and user studies.
