一个模型搞定所有图文布局生成任务,还能自动适应各种约束条件。
UniLayDiff: A Unified Diffusion Transformer for Content-Aware Layout Generation
- 将布局约束当作独立模态,用扩散Transformer统一建模
- 在多个任务上达到当前最优效果,支持无条件到多条件生成
- 适合需要快速生成多样化设计布局的AI设计师或自动化工具
内容感知布局生成是图形设计自动化中的关键任务,旨在创建与背景图像自然融合的视觉吸引人元素排布。现实应用多样,使得开发一个能统一处理多种输入受限生成任务(如基于元素类型、尺寸或关系约束)的单一模型极具挑战性。现有方法要么仅覆盖部分任务,要么需为不同条件单独配置参数,难以实现真正统一。本文提出UniLayDiff:首个可端到端训练的统一扩散Transformer模型,首次实现对各类内容感知布局生成任务的统一建模。具体而言,将布局约束视为独立模态,采用多模态扩散Transformer捕捉背景图、布局元素与多样约束间的复杂交互;并通过在预训练后使用LoRA微调整合关系约束。该方案不仅实现统一条件生成,还提升整体布局质量。大量实验证明,UniLayDiff在从无条件到多种条件生成任务中均达到领先性能,据我们所知,是首个真正统一完整内容感知布局生成任务的模型。
原文摘要 · Abstract (English)
Content-aware layout generation is a critical task in graphic design automation, focused on creating visually appealing arrangements of elements that seamlessly blend with a given background image. The variety of real-world applications makes it highly challenging to develop a single model capable of unifying the diverse range of input-constrained generation sub-tasks, such as those conditioned by element types, sizes, or their relationships. Current methods either address only a subset of these tasks or necessitate separate model parameters for different conditions, failing to offer a truly unified solution. In this paper, we propose UniLayDiff: a Unified Diffusion Transformer, that for the first time, addresses various content-aware layout generation tasks with a single, end-to-end trainable model. Specifically, we treat layout constraints as a distinct modality and employ Multi-Modal Diffusion Transformer framework to capture the complex interplay between the background image, layout elements, and diverse constraints. Moreover, we integrate relation constraints through fine-tuning the model with LoRA after pretraining the model on other tasks. Such a schema not only achieves unified conditional generation but also enhances overall layout quality. Extensive experiments demonstrate that UniLayDiff achieves state-of-the-art performance across from unconditional to various conditional generation tasks and, to the best of our knowledge, is the first model to unify the full range of content-aware layout generation tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。