清华唐杰团队造了个“中文ai设计师”,效果超Dall·E

新闻 人工智能
最近清华大学唐杰团队打造了一个“中文版Dall·E”——CogView,它可以将中文文字转图像。

[[402579]]

本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。

要说2021年OpenAI最热最有创意的产品,那么非Dall·E莫属了,这是一个可以从“AI设计师”,只要给它一段文字,就能按要求生成图像。但可惜的是Dall·E并不支持中文。

现在好了,最近清华大学唐杰团队打造了一个“中文版Dall·E”——CogView,它可以将中文文字转图像。

CogView可以生成现实中真实存在场景,如“一条小溪在山涧流淌”:

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

也可以制造不存在的虚拟事物,如“猫猪”:

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

有时候还有点黑色幽默,如“一个心酸的博士生”:

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

CogView现在还提供了试玩网页,你可以在那里输入任何文字去转成图形,不像OpenAI的Dall·E只提供几个关键词修改选项。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

能指定画风,能设计服装

CogView的能力可不仅仅是从文字输入图像,它还能处理不同微调策略的下游任务,例如风格学习、超分辨率、文本图像排名和时装设计。

在使用CogView的时候,可以加入不同风格限定,从而生成不同的绘画效果。在微调期间,图像对应的文本也是“XX风格的图像”。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

CogView设计的服装也像模像样,看起来就像电商展示页,没有虚假痕迹。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

原理

CogView是一个带有VQ-VAE分词器40亿参数的Transfomer,它的总体结构如下:

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

CogView使用GPT模型处理离散字典上的token序列。然后将学习过程分为两个阶段:编码器和解码器学习最小化重建损失,单个GPT通过串联文本优化两个负对数似然 (NLL) 损失。

结果是,第一阶段退化为纯离散自动编码器,作为图像tokenizer将图像转换为标记序列;第二阶段的GPT承担了大部分建模任务。

图像tokenizer的训练非常重要,方法有最近邻映射、Gumbel采样、softmax逼近三种,Dall·E使用的是第三种,而对于CogView来说三者差别不大。

CogView的主干是一个单向Transformer,共有48层、40个注意力头、40亿参数,隐藏层的大小为2560。

在训练中,作者发现CogView有两种不稳定性:溢出(以NaN损失为特征)和下溢(以发散损失为特征),然后他们提出了用PB-Relax、Sandwich-LN来解决它们。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

最后,CogView在MS COCO实现了最低的FID,其性能优于以前基于GAN的模型和以及类似的Dall·E。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

而在人工评估的测试中,CogView被选为最好的概率为37.02%,远远超过其他基于GAN的模型,已经可以与Ground Truth(59.53%)竞争。

清华唐杰团队造了个“中文AI设计师”,效果超Dall·E,可在线试玩

另外作者已经放出了GitHub项目页,不过目前还没有代码,感兴趣的朋友可以关注一下等代码放出。

论文地址:
https://arxiv.org/abs/2105.13290

试用Demo:
https://lab.aminer.cn/cogview/index.html

GitHub页:
https://github.com/THUDM/CogView

 

 

责任编辑:张燕妮 来源: 量子位
相关推荐

2021-09-09 09:29:03

AI 数据模型

2023-09-21 10:31:06

人工智能模型

2023-12-29 13:47:00

AI奥特曼

2013-01-18 17:00:20

设计师创业团队

2023-04-19 15:15:02

办公AI

2022-05-24 14:47:55

AI技术

2022-08-31 08:54:57

AIDALL-E 2OpenAI

2024-01-09 08:00:00

人工智能工具设计师

2022-09-13 15:56:48

AI创造艺术谷歌

2022-08-17 15:52:24

AI

2019-12-10 16:50:01

人工智能设计Philipe Sta

2015-07-27 15:26:47

设计

2023-10-10 13:42:56

训练数据

2018-04-16 08:30:07

AI设计师苏宁

2023-10-23 09:25:08

模型AI

2023-10-11 19:00:59

AIAdobe图像

2013-09-09 15:29:50

设计师图标集

2022-12-25 14:08:10

AI模型

2018-06-07 16:56:29

AI设计师艺术

2018-10-16 09:54:59

代码开发AI
点赞
收藏

51CTO技术栈公众号