一张图即出AI视频！谷歌全新扩散模型，让人物动起来

  新智元报道  

编辑：alan

【新智元导读】近日，来自谷歌的研究人员发布了多模态扩散模型VLOGGER，只需一张照片，和一段音频，就能直接生成人物说话的视频！

只需一张照片，和一段音频，就能直接生成人物说话的视频！

近日，来自谷歌的研究人员发布了多模态扩散模型VLOGGER，让我们朝着虚拟数字人又迈进了一步。

论文地址：https://enriccorona.github.io/vlogger/paper.pdf

VLOGGER接收单个输入图像，使用文本或者音频驱动，生成人类说话的视频，包括口型、表情、肢体动作等都非常自然。

我们先来看几个例子：

如果感觉视频使用别人的声音有点违和，小编帮你关掉声音：

可以看出整个生成的效果是非常优雅自然的。

VLOGGER建立在最近生成扩散模型的成功之上，包括一个将人类转成3D运动的模型，以及一个基于扩散的新架构，用于通过时间和空间控制，增强文本生成图像的效果。

VLOGGER可以生成可变长度的高质量视频，并且这些视频可以通过人脸和身体的高级表示轻松控制。

比如我们可以让生成视频中的人闭上嘴：

或者闭上双眼：

与之前的同类模型相比，VLOGGER不需要针对个体进行训练，不依赖于面部检测和裁剪，而且包含了肢体动作、躯干和背景，——构成了可以交流的正常的人类表现。

AI的声音、AI的表情、AI的动作、AI的场景，人类开始的价值是提供数据，再往后可能就没什么价值了？

在数据方面，研究人员收集了一个新的、多样化的数据集MENTOR，比之前的同类数据集大了整整一个数量级，其中训练集包括2200小时、800000个不同个体，测试集为120小时、4000个不同身份的人。

研究人员在三个不同的基准上评估了VLOGGER，表明模型在图像质量、身份保存和时间一致性方面达到了目前的最优。

VLOGGER

VLOGGER的目标是生成一个可变长度的逼真视频，来描绘目标人说话的整个过程，包括头部动作和手势。

如上图所示，给定第1列所示的单个输入图像和一个示例音频输入，右列中展示了一系列合成图像。

包括生成头部运动、凝视、眨眼、嘴唇运动，还有以前模型做不到的一点，生成上半身和手势，这是音频驱动合成的一大进步。

VLOGGER采用了基于随机扩散模型的两阶段管道，用于模拟从语音到视频的一对多映射。

第一个网络将音频波形作为输入，以生成身体运动控制，负责目标视频长度上的凝视、面部表情和姿势。

第二个网络是一个包含时间的图像到图像的平移模型，它扩展了大型图像扩散模型，采用预测的身体控制来生成相应的帧。为了使这个过程符合特定身份，网络获取了目标人的参考图像。

VLOGGER使用基于统计的3D身体模型，来调节视频生成过程。给定输入图像，预测的形状参数对目标标识的几何属性进行编码。

首先，网络M获取输入语音，并生成一系列N帧的3D面部表情和身体姿势。

然后渲染移动3D身体的密集表示，以在视频生成阶段充当2D控件。这些图像与输入图像一起作为时间扩散模型和超分辨率模块的输入。

音频驱动的运动生成

管道的第一个网络旨在根据输入语音预测运动。此外还通过文本转语音模型将输入文本转换为波形，并将生成的音频表示为标准梅尔频谱图（Mel-Spectrograms）。

管道基于Transformer架构，在时间维度上有四个多头注意力层。包括帧数和扩散步长的位置编码，以及用于输入音频和扩散步骤的嵌入MLP。

在每一帧中，使用因果掩码使模型只关注前一帧。模型使用可变长度的视频进行训练（比如TalkingHead-1KH数据集），以生成非常长的序列。

研究人员采用基于统计的3D人体模型的估计参数，来为合成视频生成中间控制表示。

模型同时考虑了面部表情和身体运动，以生成更好的表现力和动态的手势。

此外，以前的面部生成工作通常依赖于扭曲（warped）的图像，但在基于扩散的架构中，这个方法被忽视了。

作者建议使用扭曲的图像来指导生成过程，这促进了网络的任务并有助于保持人物的主体身份。

生成会说话和移动的人类

下一个目标是对一个人的输入图像进行动作处理，使其遵循先前预测的身体和面部运动。

受ControlNet的启发，研究人员冻结了初始训练的模型，并采用输入时间控件，制作了编码层的零初始化可训练副本。

作者在时间域中交错一维卷积层，网络通过获取连续的N帧和控件进行训练，并根据输入控件生成参考人物的动作视频。

模型使用作者构建的MENTOR数据集进行训练，因为在训练过程中，网络会获取一系列连续的帧和任意的参考图像，因此理论上可以将任何视频帧指定为参考。

不过在实践中，作者选择采样离目标剪辑更远的参考，因为较近的示例提供的泛化潜力较小。

网络分两个阶段进行训练，首先在单帧上学习新的控制层，然后通过添加时间分量对视频进行训练。这样就可以在第一阶段使用大批量，并更快地学习头部重演任务。

作者采用的learning rate为5e-5，两个阶段都以400k的步长和128的批量大小训练图像模型。

多样性

下图展示了从一个输入图片生成目标视频的多样化分布。最右边一列显示了从80个生成的视频中获得的像素多样性。

在背景保持固定的情况下，人的头部和身体显著移动（红色意味着像素颜色的多样性更高），并且，尽管存在多样性，但所有视频看起来都很逼真。

视频编辑

模型的应用之一是编辑现有视频。在这种情况下，VLOGGER会拍摄视频，并通过闭上嘴巴或眼睛等方式改变拍摄对象的表情。

在实践中，作者利用扩散模型的灵活性，对应该更改的图像部分进行修复，使视频编辑与原始未更改的像素保持一致。

视频翻译

模型的主要应用之一是视频翻译。在这种情况下，VLOGGER会以特定语言拍摄现有视频，并编辑嘴唇和面部区域以与新音频（例如西班牙语）保持一致。

参考资料：https://enriccorona.github.io/vlogger/