mlsanityData to 14 Oct 2026

3D Space as a Scratchpad for Editable Text-to-Image Generation

arXiv 2601.14602 ↗First posted 21 Jan 2026cs.CVcited by 1

On the map: LLM reasoning → Spatial and embodied reasoning

also in 1 more
Abstract (arXiv)

Recent progress in large language models (LLMs) has shown that reasoning improves when intermediate thoughts are externalized into explicit workspaces, such as chain-of-thought traces or tool-augmented reasoning. Yet, visual language models (VLMs) lack an analogous mechanism for spatial reasoning, limiting their ability to generate images that accurately reflect geometric relations, object identities, and compositional intent. We introduce the concept of a spatial scratchpad -- a 3D reasoning substrate that bridges linguistic intent and image synthesis. Given a text prompt, our framework parses subjects and background elements, instantiates them as editable 3D meshes, and employs agentic scene planning for placement, orientation, and viewpoint selection. The resulting 3D arrangement is rendered back into the image domain with identity-preserving cues, enabling the VLM to generate spatially consistent and visually coherent outputs. Unlike prior 2D layout-based methods, our approach supports intuitive 3D edits that propagate reliably into final images. Empirically, it achieves a 32% improvement in text alignment on GenAI-Bench, demonstrating the benefit of explicit 3D reasoning for precise, controllable image generation. Our results highlight a new paradigm for vision-language models that deliberate not only in language, but also in space. Code and visualizations at https://oindrilasaha.github.io/3DScratchpad/

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Spatial and embodied reasoning, one of 17 directions in LLM reasoning. Builds on 47 of its 53 references that are in the map. First posted on arXiv 21 Jan 2026.

Builds on

47 of 53 references.

References · 47

  1. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  2. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
  3. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
  4. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
  5. 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
  6. 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,300
  7. 2022An Image is Worth One Word: Personalizing Text-to-Image Generation using Textual Inversioncited by 2,245
  8. 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,407
  9. 2023T2I-Adapter: Learning Adapters to Dig out More Controllable Ability for Text-to-Image Diffusion Modelscited by 1,351
  10. 2021Blended Diffusion for Text-driven Editing of Natural Imagescited by 927
  11. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
  12. 2020Accelerating 3D Deep Learning with PyTorch3Dcited by 784
35 more
  1. 2023GLIGEN: Open-Set Grounded Text-to-Image Generationcited by 781
  2. 2022PAL: Program-aided Language Modelscited by 652
  3. 2023ViperGPT: Visual Inference via Python Execution for Reasoningcited by 603
  4. 2023MM-REACT: Prompting ChatGPT for Multimodal Reasoning and Actioncited by 517
  5. 2022Visual Programming: Compositional visual reasoning without trainingcited by 516
  6. 2023Q-Align: Teaching LMMs for Visual Scoring via Discrete Text-Defined Levelscited by 503
  7. 2024Evaluating Text-to-Visual Generation with Image-to-Text Generationcited by 474
  8. 2023LayoutGPT: Compositional Visual Planning and Generation with Large Language Modelscited by 347
  9. 2024OminiControl: Minimal and Universal Control for Diffusion Transformercited by 340
  10. 2023BoxDiff: Text-to-Image Synthesis with Training-Free Box-Constrained Diffusioncited by 282
  11. 2024Visual Sketchpad: Sketching as a Visual Chain of Thought for Multimodal Language Modelscited by 268
  12. 2024Mastering Text-to-Image Diffusion: Recaptioning, Planning, and Generating with Multimodal LLMscited by 224
  13. 2024InstanceDiffusion: Instance-level Control for Image Generationcited by 175
  14. 2020SceneFormer: Indoor Scene Generation with Transformerscited by 158
  15. 2024GenAI-Bench: Evaluating and Improving Compositional Text-to-Visual Generationcited by 135
  16. 2024SceneCraft: An LLM Agent for Synthesizing 3D Scene as Blender Codecited by 134
  17. 2021Simple multi-dataset detectioncited by 127
  18. 2025Hunyuan3D 2.5: Towards High-Fidelity 3D Assets Generation with Ultimate Detailscited by 125
  19. 2023Self-correcting LLM-controlled Diffusion Modelscited by 113
  20. 2025CAST: Component-Aligned 3D Scene Reconstruction from an RGB Imagecited by 106
  21. 2023LLM-grounded Video Diffusion Modelscited by 83
  22. 2025Scenethesis: A Language and Vision Agentic Framework for 3D Scene Generationcited by 67
  23. 2025Insert Anything: Image Insertion via In-Context Editing in DiTcited by 61
  24. 2025PromptEnhancer: A Simple Approach to Enhance Text-to-Image Models via Chain-of-Thought Prompt Rewritingcited by 43
  25. 2023Idea2Img: Iterative Self-Refinement with GPT-4V(ision) for Automatic Image Design and Generationcited by 38
  26. 2024Training-free Regional Prompting for Diffusion Transformerscited by 27
  27. 2024RePrompt: Planning by Automatic Prompt Engineering for Large Language Models Agentscited by 16
  28. 2024Whiteboard-of-Thought: Thinking Step-by-Step Across Modalitiescited by 16
  29. 2025OptiScene: LLM-driven Indoor Scene Layout Generation via Scaled Human-aligned Data Synthesis and Multi-Stage Preference Optimizationcited by 14
  30. 2025Precise Object and Effect Removal with Adaptive Target-Aware Attentioncited by 13
  31. 2025Seg2Any: Open-set Segmentation-Mask-to-Image Generation with Precise Shape and Semantic Controlcited by 10
  32. 2024MUSES: 3D-Controllable Image Generation via Multi-Modal Agent Collaborationcited by 8
  33. 2025InstanceGen: Image Generation with Instance-level Instructionscited by 6
  34. 2025SIGMA-GEN: Structure and Identity Guided Multi-subject Assembly for Image Generationcited by 5
  35. 2024Chain-of-Sketch: Enabling Global Visual Reasoningcited by 4

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 20263D-Layout-R1: Structured Reasoning for Language-Instructed Spatial Editing
  2. 2026SpatialGuard: Harness-Guided Verifiable Spatial Reasoning for Text-to-Image Generation
  3. 2026Autoregressive Mosaics: Probing 2D Spatial Reasoning in Text-Only Language Models
  4. 2024I Know About "Up"! Enhancing Spatial Reasoning in Visual Language Models Through 3D Reconstruction
  5. 2026SpatialImaginer: Towards Adaptive Visual Imagination for Spatial Reasoning
  6. 2026Flame3D: Zero-shot Compositional Reasoning of 3D Scenes with Agentic Language Models
  7. 2024REVISION: Rendering Tools Enable Spatial Fidelity in Vision-Language Models
  8. 2025GoT: Unleashing Reasoning Capability of Multimodal Large Language Model for Visual Generation and Editing