Spatial Chain-of-Thought: Bridging Understanding and Generation Models for Spatial Reasoning Generation
AI 摘要
提出Spatial Chain-of-Thought框架,提升扩散模型在空间理解和推理生成方面的能力。
主要贡献
- 提出SCoT框架,弥合MLLM推理和扩散模型生成能力
- 训练增强布局感知能力的扩散模型
- 利用MLLM生成布局规划
方法论
训练扩散模型以识别文本-坐标指令,利用MLLM生成布局规划,指导图像生成。
原文摘要
While diffusion models have shown exceptional capabilities in aesthetic image synthesis, they often struggle with complex spatial understanding and reasoning. Existing approaches resort to Multimodal Large Language Models (MLLMs) to enhance this capability. However, they either incur high computational costs through joint training or suffer from spatial information loss when relying solely on textual prompts. To alleviate these limitations, we propose a Spatial Chain-of-Thought (SCoT) framework, a plug-and-play approach that effectively bridges the reasoning capabilities of MLLMs with the generative power of diffusion models. Specifically, we first enhance the diffusion model's layout awareness by training it on an interleaved text-coordinate instruction format. We then leverage state-of-the-art MLLMs as planners to generate comprehensive layout plans, transferring their spatial planning capabilities directly to the generation process. Extensive experiments demonstrate that our method achieves state-of-the-art performance on image generation benchmarks and significantly outperforms baselines on complex reasoning tasks, while also showing strong efficacy in image editing scenarios.