arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.04754cs.CVcs.LG

通过带美学约束的扩散模型实现对齐的布局生成

Towards Aligned Layout Generation via Diffusion Model with Aesthetic Constraints

University at Buffalo(布法罗大学);Adobe Research(奥多比研究院);MBZUAI(Mohamed bin Zayed 人工智能大学)

Jian Chen, Ruiyi Zhang, Yufan Zhou, Rajiv Jain, Zhiqiang Xu, Ryan Rossi, Changyou Chen

AI总结:

针对扩散模型布局生成对齐偏差问题,提出带美学约束的连续扩散模型LACE,通过掩码输入实现条件生成,在多布局任务上性能优于现有最优基线。

摘要:

可控布局生成指的是在平面设计(如文档和网页设计)中,依据代表设计意图的约束条件,创建合理的视觉元素排布过程。尽管近期基于扩散的模型已取得当前最优的FID分数,但与早期基于Transformer的模型相比,它们往往表现出更明显的对齐偏差。本研究提出**LACE(布局约束扩散模型,Layout Constraint diffusion modEl)**,这是一个统一模型,可处理广泛的布局生成任务,例如排布具有指定属性的元素、优化或补全粗略的布局设计。该模型基于连续扩散模型,与使用离散扩散模型的现有方法相比,连续状态空间设计能够在训练中引入可微的美学约束函数。对于条件生成,我们通过掩码输入引入条件。大量实验结果表明,LACE可生成高质量布局,性能优于现有最优基线模型。

英文摘要:

Controllable layout generation refers to the process of creating a plausible visual arrangement of elements within a graphic design (e.g., document and web designs) with constraints representing design intentions. Although recent diffusion-based models have achieved state-of-the-art FID scores, they tend to exhibit more pronounced misalignment compared to earlier transformer-based models. In this work, we propose the $\textbf{LA}$yout $\textbf{C}$onstraint diffusion mod$\textbf{E}$l (LACE), a unified model to handle a broad range of layout generation tasks, such as arranging elements with specified attributes and refining or completing a coarse layout design. The model is based on continuous diffusion models. Compared with existing methods that use discrete diffusion models, continuous state-space design can enable the incorporation of differentiable aesthetic constraint functions in training. For conditional generation, we introduce conditions via masked input. Extensive experiment results show that LACE produces high-quality layouts and outperforms existing state-of-the-art baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑