HCMA: Hierarchical Cross-model Alignment for Grounded Text-to-Image Generation
TL;DR 精炼摘要
本文提出了层次化跨模态对齐(HCMA)框架,旨在解决文本到图像生成中的语义保真度与空间控制的矛盾。HCMA结合全局和局部对齐模块,实现复杂场景的高质量生成。实验显示,该方法在MS-COCO 2014验证集上超越现有技术,提升了FID和CLIP分数,证明了其有效性。
摘要
Text-to-image synthesis has progressed to the point where models can generate visually compelling images from natural language prompts. Yet, existing methods often fail to reconcile high-level semantic fidelity with explicit spatial control, particularly in scenes involving multiple objects, nuanced relations, or complex layouts. To bridge this gap, we propose a Hierarchical Cross-Modal Alignment (HCMA) framework for grounded text-to-image generation. HCMA integrates two alignment modules into each diffusion sampling step: a global module that continuously aligns latent representations with textual descriptions to ensure scene-level coherence, and a local module that employs bounding-box layouts to anchor objects at specified locations, enabling fine-grained spatial control. Extensive experiments on the MS-COCO 2014 validation set show that HCMA surpasses state-of-the-art baselines, achieving a 0.69 improvement in Frechet Inception Distance (FID) and a 0.0295 gain in CLIP Score. These results demonstrate HCMA's effectiveness in faithfully capturing intricate textual semantics while adhering to user-defined spatial constraints, offering a robust solution for semantically grounded image generation. Our code is available at https://github.com/hwang-cs-ime/HCMA.
思维导图
论文精读
中文精读
1. 论文基本信息
1.1. 标题
HCMA: Hierarchical Cross-model Alignment for Grounded Text-to-Image Generation (HCMA:面向基础文本到图像生成的层次化跨模态对齐)
1.2. 作者
Hang Wang (The Hong Kong Polytechnic University), Zhi-Qi Cheng (University of Washington), Chenhao Lin (Xi'an Jiaotong University), Chao Shen (Xi'an Jiaotong University), Lei Zhang (The Hong Kong Polytechnic University)。
1.3. 发表期刊/会议
根据 ACM Reference Format,该论文将被发表在一个 ACM 会议中。
1.4. 发表年份
论文的 arXiv 发布日期为 2025-05-10T05:02:58.000Z。ACM Reference Format 中显示的年份为 2018,这可能是引用格式中的占位符或未来出版的惯例。我们将以 arXiv 的发布日期 2025 年为主要参考。
1.5. 摘要
尽管文本到图像 (Text-to-image) 合成模型在生成视觉上引人注目的图像方面取得了显著进展,但现有方法在处理涉及多对象、细微关系或复杂布局的场景时,往往难以同时兼顾高层次的语义保真度 (semantic fidelity) 和明确的空间控制 (spatial control)。为了弥补这一鸿沟,本文提出了一个用于基础文本到图像生成 (grounded text-to-image generation) 的层次化跨模态对齐 (Hierarchical Cross-Modal Alignment, HCMA) 框架。HCMA 在每个扩散采样 (diffusion sampling) 步骤中集成了两个对齐模块:一个全局模块 (global module) 持续将潜在表示 (latent representations) 与文本描述对齐,以确保场景级的连贯性 (scene-level coherence);一个局部模块 (local module) 则利用边界框布局 (bounding-box layouts) 将对象锚定在指定位置,从而实现细粒度的空间控制 (fine-grained spatial control)。在 MS-COCO 2014 验证集上进行的广泛实验表明,HCMA 超越了最先进的基线方法,在 Fréchet Inception Distance (FID) (Fréchet 起始距离) 上取得了 0.69 的改进,在 CLIP Score (CLIP 分数) 上取得了 0.0295 的提升。这些结果证明了 HCMA 在忠实捕捉复杂文本语义的同时,遵守用户定义空间约束的有效性,为语义基础图像生成提供了一个鲁棒的解决方案。
1.6. 原文链接
https://arxiv.org/abs/2505.06512v3 PDF 链接: https://arxiv.org/pdf/2505.06512v3.pdf
2. 整体概括
2.1. 研究背景与动机
2.1.1. 研究背景
文本到图像 (Text-to-image) 合成技术近年来取得了飞速发展,使得生成模型能够根据自然语言提示 (natural language prompts) 生成视觉上引人入胜的图像。这些技术通常利用扩散模型 (Diffusion Models) 和大规模预训练模型,如 CLIP 和 Latent Diffusion Model (LDM),来桥接文本和图像模态。
2.1.2. 动机与核心问题
尽管当前的模型能生成看似合理且美观的图像,但在处理复杂场景时,它们仍然面临显著挑战。具体来说,当场景涉及:
- 多对象 (multiple objects)
- 细微关系 (nuanced relations)
- 复杂布局 (complex layouts) 现有解决方案常常表现不佳。它们往往难以同时实现高层次的语义保真度(即图像内容与文本描述的整体语义一致性)和显式的空间控制(即对象在图像中精确的位置和大小)。例如,模型可能在整体上捕获了文本描述的场景,但在局部区域,生成的对象可能偏离用户指定的位置或类别要求。这种“语义保真度与空间控制”之间的不一致性是当前文本到图像生成领域的关键挑战。
2.1.3. 创新思路
为了解决上述问题,HCMA 提出了一个层次化跨模态对齐 (Hierarchical Cross-Modal Alignment) 框架。其核心创新在于:
- 分层对齐: 在图像生成过程中,同时进行全局层面的对齐(确保整个场景与文本提示的语义一致性)和局部层面的对齐(确保特定对象在指定边界框内的位置和类别准确性)。
- 持续对齐: 将这两个对齐模块集成到扩散模型的每个采样步骤中,而非仅在最终步骤进行检查或修正,从而在整个生成过程中持续引导图像向正确的方向发展。
- 双重约束: 通过这种双重、持续的对齐机制,模型能够同时满足全局文本语义和局部几何布局的约束,从而生成既符合高层语义又具有精确空间控制的图像。
2.2. 核心贡献/主要发现
HCMA 框架的主要贡献总结如下:
- HCMA 框架 (HCMA Framework): 提出了 HCMA,一个基于层次化跨模态对齐的基础文本到图像生成框架,确保了语义保真度 (semantic fidelity) 和空间可控性 (spatial controllability)。通过解耦全局和局部对齐过程,HCMA 有效地保留了文本语义,同时遵守了边界框约束。
- 双重对齐策略 (Dual-Alignment Strategy): 引入了一种双层对齐机制,包括一个全局级别的字幕到图像 (caption-to-image) 模块和一个局部级别的对象到区域 (object-to-region) 模块。每个组件都通过不同的损失函数进行优化,以在整个扩散采样过程中保持整体提示的连贯性和严格精确的对象放置。
- 广泛的实验验证 (Extensive Experimental Validation): 在 MS-COCO 2014 验证集上,HCMA 超越了领先的基线方法,在 FID (Fréchet Inception Distance) 上取得了 0.69 的改进,在 CLIP Score (CLIP 分数) 上取得了 0.0295 的提升。这些结果证实了该框架在生成宏观层面连贯、微观层面遵守边界框约束的图像方面的有效性。
3. 预备知识与相关工作
3.1. 基础概念
3.1.1. 文本到图像生成 (Text-to-Image Generation)
这是一个旨在根据自然语言描述(即文本提示)合成对应图像的人工智能任务。其目标是生成与文本语义内容准确匹配的图像。
3.1.2. 扩散模型 (Diffusion Models)
扩散模型是一类生成模型,它们通过模拟一个逐步去噪 (denoising) 的过程来生成数据。在训练阶段,模型学习如何从被逐渐添加噪声的图像中恢复原始图像。在推理阶段,模型从随机噪声开始,逐步去除噪声以生成清晰的图像。这种逐步生成的方式使其在图像合成领域表现出色。
3.1.3. 潜在扩散模型 (Latent Diffusion Model, LDM)
LDM 是一种特殊的扩散模型,它将扩散过程从高维的像素空间 (pixel space) 转移到低维的潜在空间 (latent space)。首先,通过一个预训练的变分自编码器 (Variational AutoEncoder, VAE) 将图像编码成一个压缩的潜在表示。扩散模型在这个低维潜在空间中进行去噪操作,从而大大降低了计算成本,提高了生成效率和可扩展性。最后,生成的潜在表示再由 VAE 解码回像素空间得到最终图像。
3.1.4. 稳定扩散 (Stable Diffusion)
Stable Diffusion 是基于 LDM 的一个具体实现,它利用了大规模数据集 LAION-5B 进行训练,并结合了 CLIP 文本编码器,从而在解释复杂提示和生成逼真图像方面取得了显著进步。它是目前最流行和广泛使用的文本到图像生成模型之一。
3.1.5. 交叉注意力 (Cross-Attention)
交叉注意力 是一种注意力机制 (Attention Mechanism),它允许模型在处理一种模态(如图像的潜在表示)时,能够关注另一种模态(如文本嵌入)中的相关信息。在文本到图像生成中,它常用于将文本提示的语义信息融合到图像的潜在表示中,以指导图像的生成过程。
其核心计算公式如下:
- 符号解释:
-
(Query): 查询矩阵,通常来自目标模态(例如,图像的潜在表示)。
-
(Key): 键矩阵,通常来自源模态(例如,文本嵌入)。
-
(Value): 值矩阵,通常也来自源模态。
-
: 键矩阵的转置。
-
: 键向量的维度,用于缩放点积,防止梯度过大。
-
: 归一化函数,将注意力权重转换为概率分布。
-
: 最终的注意力输出,表示查询与源模态信息的加权和。
在本文中,
交叉注意力具体表现为: \begin{array} { r l } & { \mathbf { Q } = \mathbf { Z } _ { t } \cdot \mathbf { W } ^ { Q } , \mathbf { K } = \mathbf { c } \cdot \mathbf { W } ^ { K } , \mathbf { V } = \mathbf { c } \cdot \mathbf { W } ^ { V } , } \\ & { { \mathrm { A t t e n t i o n } } ( \mathbf { Z } _ { t } , \mathbf { c } ) = { \mathrm { S o f t m a x } } \left( \frac { \mathbf { Q } \mathbf { K } ^ { T } } { \sqrt { d } } \right) \cdot \mathbf { V } , \end{array}
-
- 符号解释:
- : 在时间步 的潜在表示 (latent representation)。
- : 文本嵌入 (text embedding)。
- : 可学习的投影矩阵,用于将输入转换为查询、键和值。
- : 通道维度 (channel dimension)。
Attention函数通过计算 和 之间的相似性,然后将文本上下文集成到潜在图像特征中。
3.1.6. 边界框 (Bounding Box)
边界框 是一种在图像中标记对象位置和大小的矩形框。它通常由左上角坐标、宽度和高度(或两个对角线坐标)定义。在基础文本到图像生成 (Grounded Text-to-Image Generation) 任务中,边界框作为显式空间约束,指定了特定对象应该出现在图像中的哪个位置。
3.1.7. CLIP (Contrastive Language-Image Pre-training)
CLIP 是由 OpenAI 开发的一种多模态模型,通过在大规模文本-图像对数据集上进行对比学习 (contrastive learning) 进行预训练。它能够将图像和文本编码到同一个共享的特征空间中,使得图像和其对应文本的嵌入彼此接近,而不相关图像和文本的嵌入则彼此远离。这使得 CLIP 能够用于零样本 (zero-shot) 图像分类、图像检索以及评估文本和图像之间的语义一致性(如 CLIP Score)。
3.2. 前人工作
3.2.1. 文本到图像生成 (Text-to-Image Generation)
早期的文本到图像方法在生成高分辨率图像或保持语义保真度方面面临挑战。Latent Diffusion Model (LDM) [24] 是一个重要的突破,它将生成过程转移到低维潜在空间,显著降低了计算开销。基于 LDM,Stable Diffusion Models v1.4 和 v1.5 通过优化采样策略和改进 VAE 进一步提高了图像质量和用户控制。后续版本如 、 和 SD-XL 通过先进的注意力机制和更高分辨率能力,进一步提升了复杂提示的细节和上下文感知合成能力。
其他工作包括 LAFITE [40] 在零样本和无语言场景中的应用,Gafni et al. [8] 引入人类先验知识以提高生成质量,以及 Kang et al. [10] 通过注意力图引导解决对象计数不准确的问题。然而,这些方法大多侧重于全局文本-图像连贯性,常忽视更精细的空间或以对象为中心的语义对齐需求。
3.2.2. 基础文本到图像生成 (Grounded Text-to-Image Generation)
基础文本到图像生成 是 文本到图像生成 的扩展,它除了文本描述外,还指定对象的位置或空间布局(如边界框)。这要求生成的图像不仅要保持整体语义保真度,还要根据预定义的几何约束准确地排列对象。
-
LoCo [38]:利用填充令牌 (padding tokens) 中嵌入的语义先验 (semantic priors) 来增强图像生成与输入约束之间的对齐。
-
Layout-Guidance [7]:引入了一种无需训练的框架,通过修改交叉注意力层实现前向和后向引导,从而对布局进行细粒度控制。
-
MultiDiffusion [3]:基于预训练的扩散模型,提出了一个数值优化问题来增强可控图像质量。
-
SALT-AG [29]:用空间感知噪声 (spatially aware noise) 替代随机噪声,并采用注意力引导正则化 (attention-guided regularization) 来提高合成输出中的布局遵循性。
-
GLIGEN [14]:通过整合门控自注意力机制 (gated self-attention mechanism) 来增强空间可控性。
-
BoxDiff [32]:利用内外部边界框交叉注意力约束来遵循用户定义的对象布局。
-
Attention-Refocusing [21]:利用
GPT-4[20] 提出对象放置方案,然后优化交叉注意力层和自注意力层。尽管这些方法在空间可控性方面取得了不同程度的成就,但大多数方法并未明确地在生成过程的每个阶段强制执行全局文本-图像连贯性和局部对象级对齐。
3.3. 技术演进
文本到图像生成技术经历了从早期基于 GAN (Generative Adversarial Networks) 的模型到后来 VAE (Variational Autoencoders) 和 Diffusion Models 的演进。LDM 的提出使得高分辨率图像生成在计算上变得可行,并催生了 Stable Diffusion 等一系列强大的生成模型。随着生成质量的提升,研究焦点逐渐转向如何增强对生成内容的可控性 (controllability),这包括对图像风格、姿态以及本论文关注的空间布局的控制。
基础文本到图像生成 是这一演进中的一个关键分支,它通过引入额外的空间约束(如边界框)来满足更精细的用户需求。从最初的修改 交叉注意力 机制,到引入门控模块或优化问题,前人工作不断探索如何将这些空间约束有效地融入到生成过程中。HCMA 的工作则代表了在这一方向上的最新进展,它通过引入分层对齐 (hierarchical alignment) 的概念,旨在同时解决全局语义和局部空间控制的挑战,以实现更鲁棒和精细的生成。
3.4. 差异化分析
HCMA 与现有方法的关键区别和创新点在于其层次化 (hierarchical) 和持续 (continuous) 的对齐策略:
- 分层对齐的全面性: 大多数现有方法,如
GLIGEN和BoxDiff,侧重于在特定机制(如门控自注意力、内外部框约束)下增强局部空间控制或将文本提示条件化到特定区域。HCMA 则明确地提出了全局(字幕到图像)和局部(对象到区域)两个层面的对齐模块,并在每个扩散步骤中同时进行。这意味着 HCMA 不仅关注对象是否在边界框内,还关注整个场景与文本提示的整体语义一致性。 - 持续迭代细化: HCMA 将这两个对齐模块集成到每个扩散采样步骤中。这意味着模型在去噪的每一步都会根据全局和局部损失进行潜在表示的调整,而不仅仅是在最终结果上进行检查。这种“双重细化”机制确保了从生成过程的早期阶段开始就持续引导模型遵守语义和空间约束,从而避免了局部偏差累积导致最终结果不佳的问题。相比之下,其他方法可能仅在模型架构中引入一次性或特定层的约束,而未能在整个扩散过程中持续优化。
- 鲁棒性和连贯性: 通过这种分层、持续的对齐,HCMA 能够更有效地处理多对象、复杂布局和细微关系的场景,因为它同时确保了场景级的连贯性和对象级的精确性,缓解了现有方法中常见的对象错位或分类错误问题。
4. 方法论
本节将详细介绍 HCMA (Hierarchical Cross-Modal Alignment) 框架,其核心思想是在扩散采样过程中交错进行专门的对齐和去噪步骤,以同时解决全局语义(字幕到图像)和局部布局(对象到区域)问题。通过这种分层对齐策略,HCMA 确保生成的图像既能保持整体文本一致性,又能对单个对象提供精确的空间控制。
4.1. 方法原理
HCMA 的核心原理是,在文本到图像生成过程中,通过在每个扩散步中交替执行“对齐更新”和“去噪更新”,来同时强制执行全局和局部两种约束。
-
全局对齐 (Global Alignment): 确保生成的图像在整体上与文本提示的语义内容保持一致。这通过计算潜在图像表示的全局特征与文本提示的
CLIP嵌入之间的相似性来实现,并以此来引导生成。 -
局部对齐 (Local Alignment): 确保图像中的每个对象都准确地位于其指定的边界框内,并且其类别与文本提示中对应对象的描述相符。这通过计算边界框区域内的潜在图像特征与对应对象类别的
CLIP嵌入之间的相似性来实现。这两个对齐步骤是相互补充的:全局对齐提供高层语义方向,防止图像内容偏离整体描述;局部对齐提供细粒度空间控制,确保特定对象精确放置。通过将这些对齐信号注入到扩散模型的去噪过程中,HCMA 能够逐步细化潜在表示,最终生成既忠实于文本语义又遵守精确空间布局的图像。
4.2. 核心方法详解 (逐层深入)
4.2.1. 预备知识 (Preliminaries)
扩散模型 (Diffusion Models) 近年来已成为文本到图像生成中的强大范式。其中,潜在扩散模型 (Latent Diffusion Model, LDM) [24] 引入了在学习到的潜在空间而不是原始像素空间中操作的思想,从而在各种条件输入(如文本、参考图像、空间布局)下实现了更高效和可扩展的图像合成。在 LDM 的基础上,Stable Diffusion 利用 LAION-5B 数据集 [27] 的子集,并采用 CLIP 文本编码器 [22],有效桥接了文本和图像特征,显著提高了对复杂提示的解释和合成图像的真实感。
形式上,给定时间步 的潜在表示 ,Stable Diffusion 的训练目标是学习一个噪声估计函数 ,该函数预测在每个扩散步中添加的高斯噪声 :
- 符号解释:
-
: 噪声估计模型 的可学习参数。
-
: 真实的高斯噪声。
-
: 噪声估计模型在给定潜在表示 、时间步 和文本嵌入 时预测的噪声。
-
: 文本描述 经过
CLIP文本编码器后的嵌入。 -
: 均方误差 (Mean Squared Error, MSE),衡量预测噪声与真实噪声之间的差异。
在推理过程中,一个带有空间变换器 (spatial transformer) 的
U-Net[25] 通过交叉注意力机制融合潜在表示 和文本嵌入 :
-
\begin{array} { r l } & { \mathbf { Q } = \mathbf { Z } _ { t } \cdot \mathbf { W } ^ { Q } , \mathbf { K } = \mathbf { c } \cdot \mathbf { W } ^ { K } , \mathbf { V } = \mathbf { c } \cdot \mathbf { W } ^ { V } , } \\ & { { \mathrm { A t t e n t i o n } } ( \mathbf { Z } _ { t } , \mathbf { c } ) = { \mathrm { S o f t m a x } } \left( \frac { \mathbf { Q } \mathbf { K } ^ { T } } { \sqrt { d } } \right) \cdot \mathbf { V } , \end{array}
- 符号解释:
-
: 查询 (Query)、键 (Key) 和值 (Value) 矩阵,分别通过将潜在表示 和文本嵌入 乘以可学习的投影矩阵 得到。
-
: 可学习的投影矩阵。
-
: 通道维度。
-
: 交叉注意力操作的输出,将文本上下文集成到潜在图像特征中。
重复的
交叉注意力步骤将文本上下文集成到潜在图像特征中,指导去噪过程生成与文本对齐的高质量图像。
-
4.2.2. 问题公式化 (Problem Formulation)
本研究关注基础文本到图像生成,旨在合成与文本描述和边界框约束都对齐的图像。
给定一个噪声图像 、一个文本描述 以及一组边界框 ,其中 是边界框的数量。每个边界框 指定了一个对象的位置,其类别标签 通过 CLIP 编码生成 ,其中 。
一个预训练的 VAE 将噪声图像 转换为潜在表示 ,并且每个边界框 使用 傅里叶嵌入 (Fourier embeddings) [19] 映射到 。
与通常在单次过程中融合文本引导和去噪的现有扩散技术不同,HCMA 将每个扩散步 划分为两个互补的子过程,从而确保与全局文本语义和局部对象约束进行更精确的对齐:
-
层次语义对齐 (Hierarchical semantic alignment): 将潜在表示与文本提示和每个边界框的对象类别对齐。这确保了精炼后的潜在表示 尊重全局(场景级)和局部(区域级)约束。
-
标准去噪 (Standard denoising):
U-Net鲁棒地预测并去除时间步 的噪声,最终从新的对齐潜在表示 产生 。数学上,这两个过程可以表达为:
- 符号解释:
-
: 在时间步 的噪声潜在表示。
-
: 经过对齐模块 处理后的对齐潜在表示。
-
: 对齐模块,输入包括噪声潜在表示 、时间步 、文本描述 、边界框集合 和对象类别 。
-
:
U-Net的去噪过程,输入包括对齐潜在表示 、时间步 、文本嵌入 、边界框特征 、对象类别嵌入 和噪声估计模型 。 -
: 去噪后的潜在表示,用于下一个时间步。
通过迭代这些交替的子过程,模型逐步细化潜在表示,确保最终输出遵守全局文本语义和局部边界框规范。这种迭代细化产生高质量图像,在每个粒度级别上都忠实于所提供的基础条件。
-
4.2.3. HCMA 模型 (HCMA Model)
4.2.3.1. 整体架构 (Overall Architecture)
如下图 (原文 Figure 2) 所示,HCMA 模型接收一个输入图像 、一个文本描述 和一组边界框 。
从随机选择的扩散步 中提取潜在表示 。每个边界框 通过傅里叶嵌入 [19] 转换为 。然后,一个 Vision Transformer (ViT) 编码 ,产生丰富多尺度的视觉特征,这些特征随后通过我们的层次对齐模块,该模块包含:
-
字幕到图像对齐 (Caption-Image Alignment, C2IA): 通过将潜在图像表示与文本提示对齐,在全局层面强制执行语义连贯性。
-
对象到区域对齐 (Object-Region Alignment, O2RA): 在局部层面强制执行空间和类别一致性,将边界框区域与各自的对象类别对齐。
通过将全局和局部对齐信号明确注入到扩散采样过程中,HCMA 在整体文本-图像保真度与对特定视觉对象的细粒度控制之间实现了平衡。
4.2.3.2. 潜在特征提取 (Latent Feature Extraction)
在扩散步 ,潜在表示 被展平并投影到 ,其中 且 。一个 MLP (Multi-Layer Perceptron) 随后将 转换为一组视觉词元 (visual tokens) 。我们在这些词元前添加一个可学习的 [CLS] 词元,并添加位置嵌入,然后将得到的序列输入 ViT。
ViT 的输出自然分解为两部分:
- 全局表示 : 与
[CLS]词元相关联,捕获高层次语义。 - 局部表示 : 包含用于后续对象到区域对齐的块级细节。 全局和局部表示都作为接下来描述的层次对齐机制的基础,确保模型有效整合文本语义和对象级约束。
4.2.3.3. 字幕到图像对齐与全局损失 (Caption-to-Image Alignment & Global-Level Loss)
在文本到图像生成中,实现合成图像与文本描述之间的整体保真度仍然是一个关键且持续的挑战。在缺乏鲁棒的全局约束的情况下,模型可能准确地将某些局部区域与文本片段对齐,但仍未能捕获提示的更广泛主题或上下文元素。
字幕到图像对齐 (C2IA) 模块通过将潜在图像表示和文本嵌入投影到共享语义空间中来解决这一缺陷,从而有效促进整体连贯性。具体来说,我们从 ViT 的 [CLS] 词元中获得全局表示 ,并通过一个三层全连接网络进行投影,生成 。同时,文本提示 由 CLIP 文本编码器编码,得到 。为了量化全局对齐,我们定义:
- 符号解释:
-
: 在时间步 的全局对齐损失。
-
: 潜在图像表示的全局特征向量,由
ViT的[CLS]词元经过MLP投影得到。 -
: 文本提示 经过
CLIP文本编码器后的嵌入向量。 -
: 向量点积。
-
: 向量的 L2 范数(模长)。
-
此公式计算的是
1减去 和 之间的余弦相似度 (cosine similarity)。通过最小化 ,模型逐步细化潜在图像的全局特征,使其更紧密地匹配文本嵌入。因此,每个扩散步都被引导生成不仅反映局部细节,而且反映 所描述的总体主题和上下文的图像。
-
4.2.3.4. 对象到区域对齐与局部损失 (Object-to-Region Alignment & Local-Level Loss)
即使有鲁棒的全局对齐,如果忽视局部约束,仍可能发生对象错位或分类错误。对象到区域对齐 (O2RA) 通过将特定的边界框区域与其对应的对象类别联系起来解决此问题,从而保持对文本指令的局部保真度。
我们将 重塑为 ,并将其与边界框特征 融合。这产生了 。经过均值池化 (mean pooling) 和 MLP 投影后,我们得到 ,其中 对应于第 个边界框。如果 是标签 的 CLIP 嵌入,则局部层损失为:
- 符号解释:
-
: 在时间步 的局部对齐损失。
-
: 边界框的总数量。
-
: 对应于第 个边界框 的潜在图像区域特征向量,由局部表示 融合边界框特征 经过均值池化和
MLP投影得到。 -
: 第 个边界框对应的对象类别标签 经过
CLIP编码器后的嵌入向量。 -
此公式计算的是所有边界框区域特征与其对应类别嵌入之间余弦相似度的平均值的
1减去该平均值。因此,最小化 驱动模型在每个边界框区域生成与指定类别和空间范围精确匹配的对象。
-
4.2.3.5. 潜在空间中的采样更新 (Sampling Update in Latent Space)
在推理过程中,HCMA 在每个扩散步 交错进行对齐和去噪。 设 表示时间步 的潜在表示。我们首先应用一个对齐更新:
- 符号解释:
-
: 经过对齐更新后的潜在表示。
-
: 当前时间步的潜在表示。
-
: 梯度操作,计算损失函数关于 的梯度。
-
: 全局对齐损失。
-
: 局部对齐损失。
-
: 平衡全局和局部损失的超参数。
-
: 对齐学习率。
这一步细化了 ,以更好地满足字幕到图像和对象到区域的约束。接下来,我们将更新后的潜在表示 传递给
U-Net进行去噪更新:
-
- 符号解释:
-
: 下一个时间步的潜在表示。
-
: 步长参数。
-
: 学习到的噪声估计模型,用于预测噪声。
通过从 到 迭代,我们在每个步骤执行“双重细化”——在对齐和去噪之间交替——最终获得最终的潜在表示 。然后,这个精炼后的潜在表示由
VAE解码成合成图像,确保输出在保持高视觉保真度的同时,遵守文本语义和边界框约束。
-
4.3. 训练与推理 (Training and Inference)
HCMA 的整体过程如原文的 Algorithm 1 所示。
-
训练 (Training):
- 在每个训练步 :
- 使用
CLIP文本编码器编码文本提示 得到 。 - 使用
CLIP编码器编码对象类别 得到 。 - 使用傅里叶编码计算边界框嵌入 。
- 通过
ViT从 中提取视觉词元,得到全局表示 和局部表示 。 - 通过
MLP投影 和 得到 和 。 - 计算全局对齐损失
\Omega_t^G = 1 - \cos(\mathbf{f}_t^G, \mathbf{f}_T(c))。 - 计算局部对齐损失
\Omega_t^L = \frac{1}{M} \sum_{i=1}^M \left( 1 - \cos(\mathbf{f}_{t,b_i}^L, \mathbf{f}_T(y_i)) \right)。 - 计算对齐引导的潜在表示 。
- 使用噪声损失 更新噪声预测模型 。
- 使用
- 经过足够的训练周期后,HCMA 将能够熟练地在潜在空间中整合文本语义和边界框约束。
- 在每个训练步 :
-
推理 (Inference / Sampling):
-
随机初始化潜在表示 。
-
从 到
1迭代:- 像训练时一样计算 和 。
- 计算对齐后的潜在表示 。
- 预测噪声 。
- 更新潜在表示 。
-
最终返回潜在状态 ,它将被
VAE解码成图像,该图像展示了对文本的语义保真度和按规定放置对象的准确性。通过将层次化跨模态对齐集成到基于扩散的采样框架中,HCMA 为在复杂文本描述和边界框布局下生成图像提供了一个灵活且鲁棒的解决方案。它强制执行字幕到图像和对象到区域的约束,缓解了对象错误分类或场景不匹配等问题。此外,通过交替对齐和去噪步骤,模型逐步整合文本和布局引导,从而实现可靠的多对象或多布局生成而不牺牲图像质量。
该图像是一个示意图,展示了HCMA框架中的不同模块,包括CLIP文本编码器、变换编码器和U-Net结构。图中描述了图像生成过程中,标签与图像的对齐模块以及对象区域对齐模块如何协同工作,涉及公式 。
-
原文 Figure 2: HCMA框架,该框架全面统一了文本描述与明确定义的空间约束。
5. 实验设置
5.1. 数据集
实验主要使用了 MS-COCO 2014 数据集 [16]。
- 规模: 该数据集包含 82,783 张图像用于训练,40,504 张图像用于验证。
- 特点: 每张图像都附带五个文本描述其语义内容的字幕,以及其中存在的对象的边界框和类别标签。这种丰富的标注方案为评估基础文本到图像生成中的全局语义对齐和局部对象一致性提供了强大的平台。
- 实验划分: 对于 HCMA 方法,MS-COCO 2014 训练集被划分为 80% 用于训练和 20% 用于验证,以与文本到图像研究的标准实践保持一致。官方的 MS-COCO 2014 验证集则被用作测试集。为了与现有方法进行直接比较并确保对框架进行全面评估,从测试集中采样了 30,000 张图像来报告最终的定量结果。
5.2. 评估指标
在所有实验中,采用 Fréchet Inception Distance (FID) 和 CLIP Score 来衡量性能。
5.2.1. Fréchet Inception Distance (FID)
- 概念定义 (Conceptual Definition):
FID[9] 是一种量化生成图像分布与真实图像分布之间相似性的度量。它通过比较两组图像(真实图像和生成图像)在预训练Inception-V3模型中提取的特征向量的均值和协方差来评估它们的相似性。FID值越低,表示生成图像的视觉保真度和多样性越高,与真实图像的分布越接近。 - 数学公式 (Mathematical Formula):
- 符号解释 (Symbol Explanation):
- : 真实图像特征的均值向量。
- : 生成图像特征的均值向量。
- : 两个均值向量之间的欧几里得距离的平方,衡量了分布中心位置的差异。
- : 真实图像特征的协方差矩阵。
- : 生成图像特征的协方差矩阵。
- : 矩阵的迹 (trace) 运算(对角线元素之和)。
- : 这部分衡量了两个分布的协方差结构差异,反映了图像多样性和模式匹配能力。
5.2.2. CLIP Score (CLIP 分数)
- 概念定义 (Conceptual Definition):
CLIP Score[22] 用于评估生成图像与其对应文本提示之间的语义对齐程度。它利用预训练的CLIP模型将图像和文本嵌入到同一个共享的特征空间中,然后计算它们嵌入向量之间的余弦相似度。CLIP Score越高,表示生成图像与文本描述的语义一致性越强。 - 数学公式 (Mathematical Formula):
- 符号解释 (Symbol Explanation):
- : 生成图像的总数量。
- : 第 个生成的图像。
- : 第 个图像对应的文本提示。
- :
CLIP图像编码器,将图像编码为特征向量。 - :
CLIP文本编码器,将文本编码为特征向量。 - : 计算两个向量 和 之间的余弦相似度。
5.3. 对比基线
HCMA 与七个最先进的模型进行了比较,这些模型均遵循其在 MS-COCO 2014 数据集上的官方训练和验证设置,并在其官方验证集上进行评估。这些基线包括:
- 仅依赖文本提示的文本到图像生成方法:
SD-v1.4[24]SD-v1.5[24]Attend-and-Excite[6]
- 接受文本提示和边界框布局的基础文本到图像方法:
-
BoxDiff[32] -
Layout-Guidance[7] -
GLIGEN (LDM)[14] -
GLIGEN (SD-v1.4)[14]为了确保公平性和可复现性,所有对比基线都采用了其官方的超参数设置。
-
5.4. 实现细节
- 基础模型: HCMA 构建在
Stable Diffusion v1.4之上作为主要的扩散骨干网络。 - 训练环境: 在 HPC 环境下使用 1024 的批处理大小 (batch size),实现高效并行化。
- Vision Transformer (ViT) 配置: 包含 8 个注意力头、6 个 Transformer 块,隐藏维度 ,输入分辨率为 。
- 训练过程: 训练进行 60 个 epoch,使用
Adam优化器,学习率为 ,权重衰减为 。 - 超参数: 全局和局部损失的平衡参数 和 均设置为 1。
- 采样过程: 采用
PLMS(Pseudo Linear Multi-step) 采样器,共 50 个步骤。在每个扩散步中,交替进行对齐优化和去噪,确保每个部分更新都彻底保留语义完整性。 - 可复现性: 在整个实验过程中保持固定的随机种子,并定期监控性能以选择稳定的检查点。最终模型在对齐精度和计算效率之间取得平衡,同时在不同场景和对象排列中保持高视觉保真度。
6. 实验结果与分析
6.1. 核心结果分析
以下是原文 Table 1 的结果,它将 HCMA 框架与七个最先进的基线方法进行了比较。
| Methods | Image-level | Region-level | ||
|---|---|---|---|---|
| FID (↓) | CLIP score (↑) | FID (↓) | CLIP score (↑) | |
| SD-v1.4 [24] | 12.12 | 0.2917 | 13.66 | 0.2238 |
| SD-v1.5 [24] | 9.43 | 0.2907 | 13.76 | 0.2237 |
| Attend-and-Excite [6] | 14.87 | 0.2854 | 14.57 | 0.2224 |
| BoxDiff [32] | 20.26 | 0.2833 | 14.27 | 0.2248 |
| Layout-Guidance [7] | 21.82 | 0.2869 | 21.50 | 0.2262 |
| GLIGEN (LDM) [14] | 13.05 | 0.2891 | 18.53 | 0.2381 |
| GLIGEN (SD-v1.4) [14] | 12.07 | 0.2936 | 17.25 | 0.2356 |
| HCMA(SD-v1.4) | 8.74 | 0.3231 | 12.48 | 0.2392 |
从结果可以看出,HCMA 在 FID 和 CLIP Score 两项指标上均取得了最高性能,这表明其在图像级和区域级都表现出卓越的有效性。
6.1.1. 图像级性能 (Image-Level Performance)
- 与
SD-v1.5相比,HCMA 的FID提高了 0.69(从 9.43 降至 8.74),CLIP Score提高了 0.0324(从 0.2907 升至 0.3231)。这表明 HCMA 在生成的场景中具有增强的语义一致性和更高的视觉保真度。 - 与最强的基础生成方法
GLIGEN (SD-v1.4)相比,HCMA 在FID上取得了 3.33 的提升(从 12.07 降至 8.74),在CLIP Score上取得了 0.0295 的提升(从 0.2936 升至 0.3231)。这些显著的提升反映了 HCMA 处理更精细的对象交互、复杂场景构成和详细文本指令的能力,而这些是其他方法常难以处理的。这些改进可能源于全局语义对齐和局部边界框遵循之间的协同作用,每个扩散步骤都受益于高层连贯性和精确的空间约束。
6.1.2. 区域级性能 (Region-Level Performance)
-
在局部质量方面,HCMA 在
FID上超越SD-v1.5达 1.28(从 13.76 降至 12.48),在CLIP Score上超越 0.0155(从 0.2237 升至 0.2392),这突显了其在保持对象完整性和对齐局部语义方面的卓越能力。 -
与
GLIGEN (SD-v1.4)相比,HCMA 在FID上进一步提升了 4.77(从 17.25 降至 12.48),在CLIP Score上提升了 0.0036(从 0.2356 升至 0.2392),这强调了层次化跨模态对齐在强制严格空间约束而不牺牲真实感方面的有效性。总体而言,这些发现证实,统一全局(图像范围)和局部(区域级)对齐使得 HCMA 能够以连贯的方式忠实捕捉多对象提示并满足复杂的布局要求。
6.2. 消融实验/参数分析
为了探究字幕到图像对齐 (C2IA) 和对象到区域对齐 (O2RA) 模块在 HCMA 框架中的作用,作者通过系统地禁用一个或两个模块进行了消融实验。以下是原文 Table 2 的结果:
| C2IA | O2RA | Image-level | Region-level | ||
|---|---|---|---|---|---|
| FID (↓) | CLIP score (↑) | FID (↓) | CLIP score (↑) | ||
| 12.07 | 0.2936 | 17.25 | 0.2356 | ||
| ✓ | 9.62 | 0.3225 | 11.65 | 0.2389 | |
| ✓ | 9.14 | 0.3229 | 11.75 | 0.2386 | |
| ✓ | ✓ | 8.74 | 0.3231 | 12.48 | 0.2392 |
6.2.1. C2IA 模块的有效性 (Effectiveness of the C2IA Module)
移除 C2IA (即 HCMA (w/o C2IA),对应表格中第二行 ✓ 和 ``) 意味着将所有对齐都集中在局部边界框的遵循上,只留下 O2RA 作为语义一致性的唯一机制。在这种情况下,模型在每个扩散步只对齐对象类别和边界框,但缺乏确保生成图像与文本提示之间全局语义连贯性的明确策略。
从 Table 2 可以看出,没有 C2IA 时,图像级的 FID 性能下降了 0.88 (9.62 vs 8.74),CLIP Score 下降了 0.0006 (0.3225 vs 0.3231);区域级的 CLIP Score 也下降了 0.0003 (0.2389 vs 0.2392)。尽管有所下降,HCMA (w/o C2IA) 仍然比同时缺少两个模块的模型(表格第一行)取得了显著提升。这表明即使是单独的局部对齐也能提供有意义的改进。然而,这些结果强调了持续将潜在表示与文本叙述对齐的重要性。通过提供全局上下文,C2IA 有助于协调对象排列、颜色和整体场景构成。它的缺失会导致某些场景范围的语义减弱,从而导致分数降低。
6.2.2. O2RA 模块的有效性 (Effectiveness of the O2RA Module)
类似地,省略 O2RA (即 HCMA (w/o O2RA),对应表格中第三行 `` 和 ✓) 意味着只有文本提示与整个图像对齐,从而消除了任何显式的边界框对齐。
Table 2 显示,这种省略导致图像级 FID 和 CLIP Score 分别下降了 0.4 (9.14 vs 8.74) 和 0.0002 (0.3229 vs 0.3231),同时区域级 CLIP Score 损失了 0.0006 (0.2386 vs 0.2392)。尽管这些下降与移除 C2IA 相比可能显得不大,但它们表明局部约束在引导精确对象放置方面仍然发挥着关键作用。事实上,HCMA (w/o O2RA) 仍然优于缺乏两个模块的模型,这突出表明单独的全局语义可以提供部分增益。然而,O2RA 的缺失允许对象偏离其指定位置和属性,限制了该方法忠实表示复杂或受严格约束提示的能力。
6.2.3. 组合分析 (Combined Analysis)
总的来说,这些消融实验结果表明 C2IA 和 O2RA 在 HCMA 中扮演着互补的角色。C2IA 通过持续将图像表示与文本提示匹配来维护场景范围的语义叙述,而 O2RA 则细化单个边界框以反映其目标内容。尽管每个模块都能独立提升性能,但当两者结合时,协同效应最为明显:完整的 HCMA 模型在图像和区域级别上都取得了最高的 FID 和 CLIP Score。值得注意的是,出现了一个轻微的权衡,即全局对齐可能间接影响局部生成,潜在地影响区域级保真度,但总体效果仍然是压倒性的积极。这种相互作用凸显了 HCMA 在捕捉提示的整体含义和强制精确局部约束之间取得的微妙平衡,最终产生了更连贯和空间准确的生成。
6.3. 定性分析
下图(原文 Figure 4)展示了 HCMA 框架与四个基线模型在五个不同复杂度和对象排列的样本上的视觉比较。
该图像是一个插图,展示了多种自然场景中不同对象的文本描述与对应的图像,包括熊、摩托车手、冲浪者和大象家庭。每个图像旁边都有相应的文本描述和目标框,为理解图像与文本之间的关系提供了清晰的对照。
原文 Figure 4: MS-COCO 2014 验证集上不同方法的视觉比较。
6.3.1. 单一对象 (Single Object)
在第一列中,针对一个更简单的单一对象场景,Attend-and-Excite 引入了伪影 (artifacts),而 SD-v1.5 生成的熊具有不真实的特征。尽管 BoxDiff 是一种基础文本到图像方法,但它忽略了指定的布局约束。GLIGEN (SD-v1.4) 将熊放置在正确的位置,但未能遵守文本提示和边界框指示的精确数量。相比之下,HCMA 正确地将一只熊放置在预定义的地点,保持了逼真的外观(例如,投射出可信的阴影)并匹配了所需的物体数量。
6.3.2. 对象对关系 (Object Pair Relationships)
第二列和第三列描绘了一对相互作用或根据介词关系排列的对象。在这里,HCMA 方法根据边界框约束和文本指令精确地定位了每个不同的对象,生成了视觉上一致的场景,这些场景表现出最少的伪影并保持了逼真的质量。这一结果强调了 HCMA 处理比简单放置更复杂的空间关系的能力,揭示了其全局和局部对齐模块的协同作用。
6.3.3. 计数和分类 (Counting and Classification)
在最后两列中,HCMA 不仅生成了正确数量的对象——例如第四列中的多头大象和第五列中的甜甜圈——而且还区分了对象类别(例如,区分成年象和幼象)。相比之下,GLIGEN (SD-v1.4) 在第四列中错误地合成了三头成年象,并在第五列中错误分类了烘焙食品(生成了四个甜甜圈外加一个额外的面包片)。这些结果强调了 HCMA 在对齐文本指令与计数准确性和细粒度分类方面的熟练程度,同时保持了高视觉保真度。
7. 总结与思考
7.1. 结论总结
本文提出了层次化跨模态对齐 (Hierarchical Cross-Modal Alignment, HCMA) 框架,该框架将全局文本-图像连贯性与局部边界框遵循性相结合,用于基础文本到图像生成。通过在每个扩散步骤中交替进行字幕到图像对齐 (C2IA) 和对象到区域对齐 (O2RA),HCMA 保持了高层次的语义保真度,同时强制执行精确的区域特定约束。在 MS-COCO 2014 验证集上的实验结果表明,HCMA 在 FID 和 CLIP Score 上相对于最先进的基线方法取得了显著提升,这表明这种分层对齐方法有效地解决了仅侧重于全局语义或局部放置的方法的缺点。HCMA 提供了一个鲁棒、灵活的基础,适用于需要整体文本保真度和精确空间配置的任务,为更先进和可控的生成模型铺平了道路。
7.2. 局限性与未来工作
论文作者指出了 HCMA 的一些潜在扩展方向:
- 多模态整合: HCMA 可以通过结合额外的模态(例如,深度图、参考图像)来丰富布局控制。这将允许模型在生成图像时考虑更复杂的几何和视觉上下文。
- 自适应权重: 可以通过在全局和局部模块之间使用自适应加权机制来处理不同的提示复杂性和领域特定需求。例如,对于需要极高空间精度的提示,可以增加局部对齐的权重;对于更强调场景氛围的提示,则可以增加全局对齐的权重。
- 人机交互: 未来的工作可能包括引入人机循环反馈 (human-in-the-loop feedback)、针对特定数据集的领域适应 (domain adaptation) 或实时交互式调整。这将进一步增强 HCMA 在虚拟原型设计、创意媒体设计和沉浸式故事讲述等场景中的适用性。
7.3. 个人启发与批判
7.3.1. 个人启发
HCMA 框架的核心思想——在生成过程中同时且持续地进行全局和局部对齐——提供了一个非常直观且强大的范式。它启发我思考,在许多复杂的生成任务中,分层约束和迭代细化可能比单一的、端到端的模型或仅在最终阶段进行后处理的方法更有效。这种策略不仅适用于图像生成,也可能迁移到其他需要同时满足宏观和微观约束的生成任务中,例如:
-
代码生成: 全局上符合用户意图和项目架构,局部上确保代码语法正确和逻辑清晰。
-
3D 模型生成: 全局上符合整体造型和风格,局部上确保细节(如纹理、连接处)的精确性。
-
故事生成: 全局上保持叙事连贯性和主题统一,局部上确保句子语法和角色行为的合理性。
此外,HCMA 的成功也再次强调了
CLIP等预训练多模态编码器在语义对齐方面的强大能力。将其嵌入到生成模型的每个去噪步骤中,有效地利用了其强大的语义理解能力来指导生成,这对于未来开发更具可控性的 AIGC (AI-Generated Content) 工具具有指导意义。
7.3.2. 批判与潜在改进
尽管 HCMA 取得了显著的进步,但仍存在一些潜在的问题或可改进之处:
- 计算成本: 在每个扩散步骤中都进行
ViT特征提取、两次MLP投影以及两次梯度计算(用于全局和局部损失),这无疑增加了计算负担和推理时间。虽然论文没有详细讨论这方面的开销,但对于需要实时或高吞吐量生成(如交互式应用)的场景,这可能是一个限制。未来的工作可以探索更轻量级的对齐机制,或者自适应地调整对齐的频率和强度。 - 超参数敏感性: 损失权重 和 的设置对模型性能至关重要。论文中简单地将它们都设为 1。然而,在不同场景或用户需求下,全局语义和局部空间约束的重要性可能有所不同。例如,当用户特别强调对象之间的复杂关系时,可能需要更高的 ;当需要极高空间精度时,可能需要更高的 。缺乏自适应或可学习的权重调整机制可能限制了模型的灵活性。
- 边界框的局限性: 论文主要关注边界框作为空间约束。虽然边界框在许多场景中非常有用,但它们可能不足以表达更复杂的空间信息,例如对象形状、姿态、部分关系或遮挡。论文也提到了未来可以引入深度图等额外模态,这将是解决这一局限性的有效途径。
- 关系建模: 尽管 HCMA 能够更好地处理“对象对关系”,但其对关系本身的显式建模可能仍有提升空间。例如,如果文本提示是“一只猫坐在狗的上面”,HCMA 可能会确保猫和狗都在各自的边界框内,并且大致遵循了上下关系,但对于更复杂的交互或姿态,可能还需要更精细的关系感知对齐机制。
- 泛化性评估: 实验主要在 MS-COCO 2014 数据集上进行。虽然这是一个标准且丰富的基准,但评估 HCMA 在其他领域(如艺术风格、专业领域图像)或更具挑战性的文本提示(如抽象概念、罕见对象组合)上的泛化能力,将有助于更全面地理解其鲁棒性。
- 用户控制粒度: 论文中提到的“用户定义空间约束”主要是通过边界框和类别标签实现的。未来的方向可以探索更细粒度的用户交互,例如通过草图、遮罩 (mask) 或参考图像来提供更丰富的空间和语义指导,从而将用户的创造力更直接地融入生成过程。
相似论文推荐
基于向量语义检索推荐的相关论文。