Paper Reading · Diffusion Models · SIGGRAPH 2024

LooseControl

Lifting ControlNet for Generalized Depth Conditioning
从精确深度图,走向场景边界与三维物体框的宽松控制

Shariq Farooq Bhat · Niloy J. Mitra · Peter Wonka
arXiv:2312.03079v1,2023 年 12 月;SIGGRAPH 2024 Conference Papers,DOI 10.1145/3641519.3657525

Figure 1:场景边界控制、三维物体框控制及两种编辑方式的论文原图
Figure 1 · 论文总览。上方给出宽松的深度条件,中间展示普通深度 ControlNet,下面展示 LooseControl;右侧展示三维物体框编辑与属性编辑。阅读时应区分“物体位置受控”和“所有物体细节受控”。
01 / Research motivation

研究问题:用户很难先画出目标图像的完整深度

普通深度 ControlNet 从图像提取精细深度图作为训练条件,因此在生成时也倾向于逐像素遵守输入。用户只给房间墙面和地板的深度,常得到空房间;只给家具立方体深度,常得到方盒外观。用户真正需要表达的是场景边界和物体大致位置,纹理、轮廓与次要物体应由生成模型补足。[论文 §1]

Research gap

条件的语义与训练数据不匹配

精细深度条件来自真实图像的像素深度;草图式几何条件只给可编辑的空间约束。直接把稀疏条件喂给旧模型,模型没有学过“在约束内部自由补全”的映射。

Core hypothesis

把宽松几何转换为深度代理

从真实图像自动计算边界深度或物体框深度,把它们与原图组成训练对。这样可以复用已有深度 ControlNet 与 Stable Diffusion 主干,通过少量 LoRA 更新学习新的条件语义。[论文 §4]

术语边界。此处的“宽松”描述几何条件允许多种满足条件的外观。论文给出的是生成条件与定性示例;它没有训练一个显式三维场景生成器,也没有保证输出几何在多视角下一致。
02 / Problem formulation

两个条件函数定义了控制强度

输入包含文本描述 \(T\) 与用户提供的代理深度图 \(D_c\),输出是生成图像 \(I_{\mathrm{gen}}\)。\(f_D\) 是现成的单目深度估计器,\(\phi\) 是判断生成结果是否满足几何意图的条件函数。普通深度控制将预测深度约束为输入深度:

\[f_D(I_{\mathrm{gen}})=D_c.\]

LooseControl 将控制写作 \(\phi(f_D(I_{\mathrm{gen}}),D_c)=\mathrm{true}\)。其中 C1 为边界深度上限;在论文采用的深度方向约定下,图像内容位于场景边界之前:

\[\phi_{\mathrm{C1}}:\quad f_D(I_{\mathrm{gen}})\le D_c.\]

C2 为近似三维物体框。\(B_i\) 是用户提供的第 \(i\) 个有向框,\(O_{\mathrm{gen}}^i\) 是生成物体,\(\operatorname{3DBox}(O_{\mathrm{gen}}^i)\) 表示其估计框;\(\sim\) 表示近似符合,论文没有定义可直接计算的严格距离阈值:

\[\phi_{\mathrm{C2}}:\quad B_i\sim\operatorname{3DBox}(O_{\mathrm{gen}}^i),\qquad \forall i.\]

这些公式规定希望实现的行为。实际训练仍采用由图像、文本、代理深度构成的监督三元组 \((T,D_c,I)\),没有在每次采样时显式计算 \(\phi\) 或投影至可行集合。论文的 C2 训练条件较严格,推理时展示了把物体框近似使用的效果。[论文 §3]

记号具体对象算法中的作用
(I,T)训练图像与 BLIP-2 生成的图像描述提供目标图像与文本条件
\(D_c\)从训练图像推导的边界或框的代理深度;推理时可以由用户构造兼容 ControlNet 的深度输入接口
\(f_D\)ZoeDepth 等单目深度估计器从图像得到深度;表达几何控制目标
(W,A,B,r)原注意力投影、两个低秩矩阵与秩只训练附加的 LoRA 参数
(gamma,lambda)LoRA 更新比例、ControlNet 条件比例分别调节学习到的更新与空间条件强度
\(\Delta\mathbf h,\mathbf x,\mathbf e_i\)ControlNet 瓶颈残差、扩散潜变量输入、语义编辑方向属性编辑的局部分析与残差修改
03 / Algorithm

将场景几何压成可学习的代理深度

准备图文

取 NYU-Depth-v2 图像;使用 BLIP-2 生成文本描述 \(T\)。

生成几何条件

使用 ZoeDepth 取得深度;C1 提取边界,C2 增加 SAM 分割和物体框。

渲染代理深度

在原视角渲染边界平面或立方体模型,得到与图像成对的 \(D_c\)。

微调与采样

冻结 Stable Diffusion;在预训练深度 ControlNet 的注意力模块训练 LoRA;推理接受文本和用户编辑的代理深度。

C1:场景边界

对图像深度反投影为三角网格,再正交投影至水平平面,用多边形包围投影,沿场景高度拉伸多边形的边形成竖直平面;可加入地面和天花板。最后从原相机视角渲染这些边界平面的深度。训练时为提高效率只使用竖直平面;这也意味着边界估计是一种几何近似。[论文 §4.1]

Figure 2:图像到场景边界代理深度的处理流程
Figure 2 · C1 数据构建。深度估计、三维反投影、水平投影、多边形边界、竖直面拉伸和重渲染,最终产生边界深度条件。图中天花板未显示。

C2:三维物体框

ZoeDepth 给深度,SAM 给分割区域。每个分割区域的像素被反投影为三维点云;为该点云拟合最小体积有向三维包围盒,构成立方体模型并从原相机位置渲染深度。模型看到的是框的前表面深度,学习从粗略占位物恢复语义外观。这个处理依赖深度与分割质量,也依赖相机内参近似。[论文 §4.2]

Figure 3 / Figure 5。论文 Figure 3 对照 C1 与 C2 的代理深度,Figure 5 逐步展示 C2 的 ZoeDepth、SAM、三维反投影、有向框和深度渲染。查看原论文图像。

LoRA 如何改变 ControlNet

对 ControlNet 注意力层的线性投影,冻结原矩阵 \(W\in\mathbb R^{M\times N}\),学习 \(B\in\mathbb R^{M\times r}\) 和 \(A\in\mathbb R^{r\times N}\);\(r\ll\min(M,N)\)。更新作用于特征 \(\mathbf x\),推理时增加比例 \(\gamma\):

\[W'\mathbf x=W\mathbf x+\gamma BA\mathbf x.\]

论文设 \(r=8\),默认 \(\gamma=1.2\)。Stable Diffusion U-Net 在微调中保持冻结。作者也测试整个 ControlNet 的直接微调;其生成样例出现颜色过饱和。论文观察到瓶颈残差更影响条件遵守,高分辨率 skip 残差更多影响局部纹理,因此调整 skip 残差可减轻这种现象。该观察来自实验诊断,尚无独立因果隔离证明。[论文 §4.3,Figure 6]

E1:保持风格的物体框编辑

用户移动、旋转、缩放、添加或删除物体框后,再执行扩散采样。单纯固定随机种子仍可能出现整体画面变化。作者在 Stable Diffusion U-Net 最后两个 decoder 层共享原图生成过程中的注意力 key/value,使编辑结果承接原图的构图和风格,同时接受新几何条件;全部层共享会过度固定图像。官方实现通过 cross-frame attention 对源、目标联合采样,默认 20 步。[论文 §4.4] [官方代码]

E2:沿语义方向编辑

固定控制条件,在某一步考察输入潜变量 \(\mathbf x\) 对 ControlNet 瓶颈残差 \(\Delta\mathbf h\) 的 Jacobian:

\[J=\frac{\partial\Delta\mathbf h}{\partial\mathbf x},\qquad J=U\Sigma V^{\mathsf T},\qquad\Delta\mathbf h'=\Delta\mathbf h+\beta\mathbf e_i.\]

取主导奇异方向对应的 \(\mathbf e_i\),以 \(\beta\) 调节残差,可探索家具密度、物体类型或材质外观等变化。论文给出的是局部潜空间方向和定性结果;“与输入深度条件正交”应理解为作者对编辑作用的描述,文中没有提供严格正交约束或全局语义一致性指标。[论文 §4.5]

04 / Reproduction details

训练、推理与公开实现

项目论文或附录中可核查的设置复现含义
基础模型Stable Diffusion v1.5;预训练深度 ControlNet checkpoint lllyasviel/control_v11f1p_sd15_depth需要匹配 SD1.5 潜空间和深度条件编码
训练图像NYU-Depth-v2;BLIP-2 自动生成图像描述论文未给出本任务最终使用的明确样本数量与随机划分
几何预处理ZoeDepth;C2 额外使用 SAM,min_mask_area=1e4;三维处理使用 PyTorch3D、Open3D分割过滤会影响被建框的物体范围
相机假设反投影视场角在 43°–57° 随机抽取影响框和边界代理深度
训练LoRA rank 8;200 steps;Adam;学习率 \(10^{-4}\);batch size 12只训练 LoRA,SD U-Net 冻结
推理默认ControlNet conditioning scale 1.0;LoRA scale \(\gamma=1.2\)论文附录默认值;公开 Python API 默认图像 512×512,UniPCMultistepScheduler,20 步
未报告参数精确训练样本数、训练随机种子、优化器其他超参数、显卡与耗时、完整提示词集合这些信息无法从论文与已公开的推理代码精确复原

上表训练细节见附录 §7。公开仓库提供 3D Box Control 权重、推理包装与 Gradio 界面;仓库根目录未提供用于从原始数据重新训练的完整脚本或自动构建代理深度的数据处理脚本。此处的复现描述据此区分“运行已公开权重”与“完整重新训练”。[官方仓库]

推理调用接收文本、负面提示词和代理深度图。官方示例负面提示词为 blurry, text, caption, lowquality, lowresolution, low res, grainy, ugly,编辑默认随机种子为 42。论文没有系统比较不同 prompt、采样步数或负面提示词的作用。C1 权重和属性编辑的完整公共接口,在所查阅的仓库根目录未与 C2 Python API 一起完整提供。[官方推理文件]

05 / Evidence & experiments

实验支持了哪些结论

Figure 7 将相同代理深度输入给原始深度 ControlNet 和 LooseControl,包含室内、室外与多个物体的定性对照。原始模型容易生成空场景或立方体外观;新模型能在边界之内补家具,并将物体框变成更自然的物体。Figure 9 改变条件比例 \(\lambda\) 后,作者观察到原始模型在多种比例下仍倾向于空房间。[论文 §5]

Figure 7:相同控制条件下普通 ControlNet 与 LooseControl 的定性比较
Figure 7 · 主要定性比较。每组从上到下依次为输入控制条件、ControlNet 输出、LooseControl 输出。前半组是边界控制,后半组是三维物体框控制。
评估 / 消融论文报告能支持的结论与边界
用户偏好41 名参与者;每人 10 道随机题,其中 C1、C2 各 5 道;匿名两选一;超过 95% 的回答选择 LooseControl支持给定提示词和条件下的相对主观偏好;没有提供细化到每种场景的置信区间或独立几何精度测量
直接微调 vs LoRAFigure 6:直接微调例子颜色过饱和;调节 skip 残差可缓解;LoRA 结果较稳定支持所展示样例中的方法选择;没有给出大样本量化消融表
条件比例 \(\lambda\)Figure 9 对比不同控制强度说明降低原始 ControlNet 条件权重不足以稳定解决空房间现象;没有系统数值曲线
编辑效果Figure 8 展示物体位置、尺寸、形状和局部属性变化展示编辑能力;未报告逐像素变化量、物体位置误差或跨种子稳定性

用户研究采用匿名网页表单与随机选项顺序;问卷比较的是作者方法与 ControlNet,选择题没有独立盲法的训练基线。论文声称“超过 95%”,不提供用于计算精确比例的完整题级数据。图像质量、文本遵守程度和几何遵守程度也未分别量化。[附录 §9]

Figure 10 · 附录的反例。Matterport3D 的房间布局标注可能跨越房间边界或含歧义,直接转换成“边界深度上限”会违反 C1 条件。因此作者选择由单目深度和几何外推构造训练条件。[附录 §8]
06 / Reviewer commentary

事实、解释与研究机会

Paper verified

论文实验证实

两种代理深度条件可以通过短步数 LoRA 微调进入已有深度 ControlNet;Figure 7、Figure 8 和用户研究展示了宽松条件下的图像生成与编辑。论文报告 41 名参与者、超过 95% 的两选一偏好。

Author claim

需要谨慎阅读

“广泛泛化”“编辑方向与深度条件正交”“瓶颈残差负责主要条件遵守”等说法主要依赖定性示例与局部诊断,缺少覆盖多个数据域、几何误差与因果隔离的系统实验证据。

Our interpretation

方法为何有效

训练代理深度把用户可操作的“容器”和“占位盒”映射到真实图像。模型从同一个粗略约束看到不同纹理与细节,于是减弱了对精确像素深度的依赖。这个解释符合构造与结果,但论文未做专门的分布宽度测量。

Research extension

值得继续检验

可定义可计算的三维框满足率、边界违反率与文本一致性指标;对多个相机视角使用同一三维条件,并将渲染反馈加入采样;探索程序化场景或材质图与宽松几何条件联合控制。这些属于后续设想。

优势

研究问题来自真实创作过程:用户能够快速摆放边界与物体框,却难以提供完整场景深度。代理条件复用预训练 ControlNet 的输入形式,200 步 LoRA 微调的训练设置也相对明确。C1、C2、E1、E2 共同形成可操作的创作流程。

局限

训练图像主要是 NYU-Depth-v2,论文没有完整交代本任务图像划分与跨域定量协议。对次要物体的控制较弱;约束过多时输出多样性下降。单目深度、分割、相机视场角及有向框拟合的错误会进入代理条件。作者公开的推理代码只覆盖部分论文流程;完整训练、数据生成、属性编辑的复现实用资料不足。[论文 §6]

07 / References

原始资料

  1. Bhat, Mitra, Wonka. LooseControl: Lifting ControlNet for Generalized Depth Conditioning,原始 PDF v1;arXiv HTML,含补充材料。
  2. ACM SIGGRAPH 2024 正式论文 DOI:10.1145/3641519.3657525;SIGGRAPH 2024 会议条目。
  3. 作者项目页面与示例图。
  4. 作者官方代码仓库;重点核查 推理包装与编辑调用。

图像均引用论文官方 arXiv HTML 的原图。页面可独立打开;公式与图片加载需要网络连接。