最会造假的GAN模型,雪地里都能有骆驼!Facebook发布IC-GAN,迁移能力史上最强

最会造假的GAN模型,雪地里都能有骆驼!Facebook发布IC-GAN,迁移能力史上最强
2021年09月26日 12:58 新智元

生成对抗网络 (GAN) 在图像生成领域可以说是最强大的 AI 模型,无论是逼真的图片、抽象的拼贴画、风格迁移都不在话下。

但GAN 也有神经网络模型所共有的致命缺点,就是具有局限性,通常只能生成与训练数据集密切相关的物体或场景的图像。

例如,在汽车图像上训练的 GAN 在生成汽车相关图像时可以做到特别逼真,但可能让它生成鲜花、动物之类的模型就会一眼假,因为生成的图像可能会违反物理常识等。

Facebook AI Research 为了解决这个问题,提出了一个新模型Instance-Conditioned GAN (IC-GAN) ,可以生成逼真的、没有见过的图像组合。

https://arxiv.org/abs/2109.05070

例如雪和骆驼这种照片或者在城市中的斑马,可以无缝衔接起来。

目前代码已经开源。

研究人员从核密度估计(kernel density estimation, KDE)技术中得到启发,引入了一种非参数化方法来建模复杂数据集的分布。KDE是一种非参数密度估计器,以参数化核的混合形式对每个训练数据点周围的密度进行建模。

IC-GAN可以看作是一种混合密度估计器,其中每个分量都是通过对训练实例进行条件化得到的。

然而与KDE不同的是,IC-GAN 没有显式地对数据概率进行建模,而是采用了一种对抗性的方法,其中我们使用一个神经网络隐式地对局部密度进行建模,该神经网络将条件实例和噪声向量作为输入。

因此,IC-GAN中的内核不再独立于我们所处理的数据点,我们通过选择实例的邻域大小来控制平滑度,而不是内核带宽参数,我们从中采样真实样本以馈送到鉴别器。

IC-GAN 将数据流形划分为由数据点及其最近邻描述的重叠邻域的混合物,IC-GAN模型能够学习每个数据点周围的分布。通过在条件实例周围选择一个足够大的邻域,可以避免将数据过度划分为小的聚类簇。

当给定一个具有M个数据样本的未标记数据集的嵌入函数f,首先使用无监督或自我监督训练得到f来提取实例特征(instance features)。

然后使用余弦相似度为每个数据样本定义k个最近邻的集合。

使用生成器隐式地模拟条件分布p(x | hi) 时,生成器从单位高斯先验z∼ N(0, 1)变换样本从条件分布中抽取样本x,其中hi是从训练数据中抽取的实例xi的特征向量。

在IC-GAN中,采用对抗式方法来训练生成器,因此生成器与判别器可以联合训练,判别器用来区分hi的真实相邻节点和生成的相邻点。对于每个hi,真实邻居都从Ai中均匀采样。

生成器 G和判别器 D都参与了一个两人最小-最大博弈,在博弈中,二者试图找到目标的纳什均衡的等式。

在训练IC-GAN时,使用所有可用的训练数据点来微调模型。在推理时,与KDE等非参数密度估计方法一样,IC-GAN的生成器也需要实例特征,这些特征可能来自于训练分布或不同的分布。

并且这种方法可以扩展到具有类条件(class condition)的生成上。通过在类标签y上添加一个额外的生成器和判别器,可以让IC-GAN 用于有类条件的生成。IC-GAN 通过向生成器和判别器提供实例的表示作为额外的输入,并通过使用实例的邻居作为鉴别器的真实样本,学习对数据点(也称为实例)的邻域的分布建模。

与对离散簇索引进行条件处理不同,对实例表示进行条件处理自然会导致生成器为相似实例生成相似样本。并且一旦训练完成,IC-GAN可以通过在推理时简单地交换条件实例,轻松地迁移到训练期间未看到的其他数据集。

实验部分研究人员使用了ImageNet和COCO Stuff数据集,实验结果表明,与无条件模型和无监督数据分割基线相比,IC-GAN显著提高了性能。

非选择性的基线模型BigGAN 是通过将训练集中的所有标签设置为零来训练的,IC-GAN在FID和IS分数方面均以64×64和128×128分辨率超过了所有以前的方法,并可以在高分辨率下生成更高的质量的图像。

在进行迁移实验时,首先使用ImageNet上使用BigGAN架构训练IC-GAN,并在测试时使用COCO Stuff实例生成图像,这种数据分割模式都包含未见过的对象组合方式。在ImageNet上训练的IC-GAN在所有分割方面都优于在COCO Stuff上训练的相同模型:在128分辨率下8.5比16.8训练FID。

为了研究ImageNet和COCO Stuff数据分布的接近程度,研究人员以128×128分辨率计算了两个数据集的实际数据序列分割之间的FID 得分为37.2。

因此,IC-GAN的显著迁移能力不能用数据集的相似性来解释,而可以归因于ImageNet预先训练的特征提取器和生成器的有效性。

将COCO Stuff中的条件实例替换为ImageNet中的条件实例时,可以得到43.5的训练FID分数,强调了通过改变条件实例可以实现的重要分布迁移。

研究人员将IC-GAN扩展到类条件情况,并在ImageNet上显示语义可控生成和可比的量化结果。

类条件IC-GAN在FID和所有分辨率方面都优于BigGAN,除了FID在128×128分辨率下的分数。与BigGAN不同,IC-GAN可以通过固定实例特征和交换类条件,或者通过固定类条件和交换实例特征来控制生成图像的语义。

生成的图像保留了类标签和实例的语义,可以在相似的背景下生成不同的狗品种,或在雪地中生成骆驼,在ImageNet中属于未知场景。

凭借这些新功能,IC-GAN 可用于创建新的视觉示例,以扩充数据集以包含不同的对象和场景;为艺术家和创作者提供更广泛、更有创意的 AI 生成内容;并推进高质量图像生成的研究。

参考资料:

https://ai.Facebook.com/blog/instance-conditioned-gans/

财经自媒体联盟更多自媒体作者

新浪首页 语音播报 相关新闻 返回顶部