谷歌推出贪婪生长方法(Greedy Growing):用来训练大规模、高分辨率的基于像素的图像扩散模型

谷歌发布论文讨论了一个非常有趣的话题:如何通过一种称为“贪婪生长”(Greedy Growing)的方法来训练大规模、高分辨率的基于像素的图像扩散模型,且无需级联超分辨率组件。简单来说,就是科学家们找到了一种新技巧,可以更有效地教会计算机如何根据文本描述生成高清晰度的图像。

贪婪生长方法的关键洞察源自于对核心组件的精心预训练,特别是那些负责文本到图像对齐与高分辨率渲染的组件。我们首先展示了一个无下采样(上采样)编码(解码)器的浅层UNet扩展所带来的益处。扩展其深层核心层被证明能提升对齐效果、物体结构和构成成分。在此核心模型基础上,开发人员提出了一种贪心算法,该算法使架构向高分辨率端到端模型成长,同时保持预训练表示的完整性,稳定训练过程,并减少了对大型高分辨率数据集的需求。这使得单一阶段模型能够生成高分辨率图像,而无需超分辨率级联。

我们的关键成果基于公开数据集,表明我们能够在不采用额外正则化方案的情况下,训练出规模达80亿参数的非级联模型。我们的全管道模型“Vermeer”,使用内部数据集训练以生成1024×1024尺寸的图像,无需级联,在与SDXL的比较中,获得了44.0%的人类评估者的偏好,相比之下SDXL仅为21.4%。

例如,如果你告诉这个模型“生成一幅画,画中有一只站在树枝上的聪明老猫头鹰,它的眼睛在月光下闪烁着黄色的光芒”,模型就会根据你的描述生成一幅详细的图像,而且图像的分辨率非常高,可以清楚地看到猫头鹰的羽毛和眼睛的细节。这就是这项技术的强大之处。

主要功能:

  • 生成高分辨率图像: 该模型能够生成非常详细的图像,分辨率可以达到1024×1024像素,这比之前的方法有显著提升。
  • 无需超分辨率级联: 传统的图像生成模型通常需要多个阶段(级联)来逐步提高图像分辨率,但这个模型能够一步到位,避免了多阶段处理带来的复杂性和潜在的质量问题。

主要特点:

  • 简单高效: 通过“贪婪生长”方法,模型的训练过程更加简单和稳定,不需要复杂的级联结构或额外的正则化技术。
  • 预训练核心组件: 模型首先在大量文本-图像数据集上预训练核心层,这些核心层负责文本到图像的对齐和高分辨率渲染。
  • 一次性端到端模型: 与传统的多阶段模型不同,这个模型是一次性训练完成的,这意味着它可以作为一个整体来生成图像,简化了训练和推理过程。

工作原理:

  1. 预训练核心层: 首先,模型在大量文本-图像对上训练,以学习如何理解和处理文本描述。
  2. 贪婪生长算法: 在核心层预训练完成后,通过一个贪婪算法逐步增加模型的复杂度,同时保持预训练层的稳定性。
  3. 端到端训练: 最终,模型以高分辨率目标进行端到端的训练,生成高清晰度的图像。

具体应用场景:

  • 艺术创作辅助: 艺术家和设计师可以利用这个模型根据他们的文本描述快速生成图像概念。
  • 虚拟现实和游戏开发: 在这些领域中,高分辨率的图像生成可以用于创建逼真的虚拟环境和角色。
  • 自动化内容生成: 对于需要大量图像内容的平台(如社交媒体、广告等),这个模型可以自动化图像的生成过程,提高效率。
  • 教育和培训: 在教育领域,可以根据教学内容自动生成图像,增强学习体验。
0

评论0

没有账号?注册  忘记密码?