LLLMWIKI
正文研究地图阅读入口元数据

来源笔记 · 更新于 2026-07-12

LayoutDiffusion以对象框控制像素 U-Net 的 layout-to-image 生成

一句话结论

LayoutDiffusion 把对象类别+bounding boxes 与 image patches 映射到统一空间坐标,用 LFM 和 OaCA 将 layout 注入从头训练的 ADM-style pixel U-Net;它是 closed-vocabulary layout-to-image,不是同名 graphic-layout token diffusion、图像编辑或 DiT。

同名消歧

  • 本文:输入 natural-scene object boxes/categories,输出像素图像;主干是 U-Net。
  • LayoutDiffusion(graphic layout generation):输入/输出 UI/文档 layout tokens,用 BERT-base 风格 Transformer reverse process,不生成图像。

两篇论文不可共享 backbone、数据、指标或任务结论。

方法与接口

  • Layout 是对象集合 $\{(bbox,category)\}$,不含自由文本、属性、关系谓词或 reference image。
  • LFM 用 6-layer Transformer encoder 在对象间 self-attention;它只是 condition encoder。
  • 每个 U-Net image patch 获得归一化 box,与 layout object 处于同一坐标表示。
  • OaCA 在 32/16/8 resolution 将对象 category/box 注入 U-Net feature。
  • 训练为 1000-step pixel-space Gaussian diffusion noise MSE;CFG dropout 0.2,推理用 DPM-Solver 25 steps。

“统一”仅指 image patch/layout 的空间表示,不是统一 generation/editing architecture。

Backbone、数据与成本

维度全文核验
去噪 backboneADM-style convolutional U-Net;无 DiT
Full / Small569M / 142M 参数
数据COCO-Stuff:25,210 train/3,097 val;VG:62,565/5,062/5,096
条件训练 taxonomy 内 category IDs + boxes;COCO 不用 segmentation,VG 只用 boxes/categories
训练8×RTX 3090;COCO Full 1.15M steps、Small 1.4M;VG Full 1.45M
等价成本COCO Full 216.55 V100-days;Small 75.83 V100-days
推理吞吐Full 0.308 img/s;Small 0.608 img/s(25 solver steps)

正文“32×3090”与补充 per-run 表“8×3090”口径不一致;source 采用具体模型表,并保留该疑点。

主结果

256×256:

数据方法FID↓DS↑CAS↑YOLO↑
COCOLostGAN-v231.180.5640.0017.50
COCOPLGAN29.100.5237.6514.40
COCOLayoutDiffusion15.610.5747.7432.00
VGLostGAN-v232.080.5334.48--
VGLayoutDiffusion15.630.5948.90--

COCO 评测 3,097 layouts×5 images;VG 5,096×1。自动 evaluator 强调可检测/可分类对象,没有人评、open-vocabulary 或 relation consistency 测试。

消融与 trade-off

COCO 128、300K:无 LFM/OaCA 时 FID 29.94、CAS 3.83、YOLO 0;完整 LFM+OaCA 为 16.57/43.60/27.00。普通 cross-attention 的 FID 16.46 略优于 OaCA,但控制指标较弱(CAS 42.47、YOLO 23.60)。DS 从无控制的 0.70 降至完整模型 0.47,显示更严格 layout adherence 会压缩同-layout 多样性。

任务边界与局限

  • 仅控制 category/box;不控制文本属性、身份、style、像素形状或任意 region prompt。
  • 修改 layout 后从噪声重采样,不保持未改区域,不能称 image editing。
  • 目标域从头训练,依赖 bbox labels;不是 pretrained T2I adapter。
  • 词表封闭,对密集/域外 layout 的泛化未测。
  • Full 25-step 模型仍需 216.55 V100-days,且吞吐低于 Small/LDM-8;少步不等于低总成本。
  • 训练图像沿用 COCO/VG 许可与偏差,论文没有开放新数据集。

相关页面

备注

本库将本文定位为 closed-vocabulary layout-to-image pixel diffusion,已清理 vision-language/unified image model/DiT 误关联。