麻省理工、Meta开源:无需人工标注,创新文生图模型
文生图领域一直面临着一个核心难题,就是有条件图像生成的效果,远超无条件的图像生成。有条件图像生成是指模型在生成图像的过程中,会额外使用类别、文本等辅助信息进行指导,这样可以更好的理解用户的文本意图,生成的图像质量也更高。
而无条件图像生成完全基于模型自己学习的数据分布,难以形成有效的指导,所以,我们经常会有的AI平台生成的图像驴唇不对马嘴,或者图像有重大缺陷。
为了解决这个问题,麻省理工学院和Meta的AI研究人员开发了RCG模型(Representation-Conditioned imageGeneration)。
RCG是一种创新性的“基于表示条件的图像生成”模型。所谓“表示条件”,就是让模型自己从数据中提取表示来进行指导,而无需任何人工数据标注。所以,RCG能以更小的消耗,生成与有指导模型相媲美的图片。
开源地址:https://github.com/LTH14/rcg
论文地址:https://arxiv.org/abs/2312.03701
从RCG的论文来看,其图片生成流程采用了拟人化思路。先规划出要生成图片的大体轮廓,然后再根据这一构思去绘制图片的细节。
就像我们人类画画一样,先在大脑中构思画作的整体思路,例如,想画一匹马,先想象马的形体;想画一个苹果,先想象个大红色的圆球,然后根据整体“表示”画出图像的细节。
图像编码器
图像编码器的作用是从图像中提取有意义的表示。这些表示在保留图像的语义信息的同时,也要比较简单、易于建模。
目前流行的自监督学习算法主要可以分为两大类:1)基于预训练任务,这类方法会人为设计一个预训练任务,让模型去预测一些伪标签。比如旋转预测任务,让模型预测图像被旋转的角度等。
2)基于对比学习,这类方法会构造正样本和负样本。让模型学会区分和拉远它们在表示空间的距离,代表性算法包括MoCo、SimCLR等。
实验证明,基于对比学习方法,可以获得更加有效的图像表示,因此RCG选择了这种方式。并使用了MoCo v3进行预训练,这是图像分类任务上效果最优的自监督对比学习算法之一。
为了使表示维度可控,RCG使用了带有预测头的Transformer模型。预测头会规划成256维的向量,这样就获得了编码器输出的表示。该表示同时还进行了归一化处理,以符合高斯分布。
表示生成模块
在提取到图像的表示之后,下一步就是对这些表示进行采样。通常,我们希望模型可以捕捉表示空间的分布,具有生成各种新颖表示的能力,从而指导下游的图像生成。
为此,RCG提出了表示扩散模型RDM。它使用全连接网络作为主干,包含数个残差块。每个块里面包括激活函数、线性层等。
RDM模块训练时,采用了DDIM算法。即先将真实图像表示加入高斯噪声,让RDM去除噪并重构原始表示;生成时,从完全噪声表示出发,逐步采样清晰的表示。
RDM模块的参数和计算量很小,所以即使增加块数和宽度,也不会带来太大的额外负担。这样我们可以充分发掘表示空间的丰富信息,指导下游生成。
像素生成器
提取表示并对其建模之后,最后一步就是根据表示来生成图像像素。RCG中的像素生成器可以使用任意的条件图像生成模型,只需要将原有的条件(比如类别标签)替换为RDM生成的表示即可。
训练时,MAGE接收带有遮蔽(Mask)的图像作为输入,其中带X的是被遮蔽的,同时也输入对应图像的表示,然后学习重构被遮蔽的内容。
为了验证RCG模型的效果,研究团队基于ImageNet测试平台中的256×256数据集进行了实验。
结果显示,RCG的FID分数为3.56和IS的253.4,超过了之前最好的无条件生成模型MAGE的结果,与当前状态最好的有条件模型CDM差不多。
每周AI大事件 | 必应聊天将支持第三方浏览器、钉钉斜杠“/”启动邀请测试
欢迎来到站长之家的[每周AI大事件],这里记录了过去一周值得关注的AI领域热点内容,帮助大家更好地了解人工智能领域的动态和发展风向。Part1动态[国内要闻]小冰首批网红明星AI克隆人提前上线站长网2023-06-03 09:19:430000摩根大通:越南处于新兴东南亚人工智能发展的“前沿”
划重点:1.📈摩根大通分析:越南对人工智能有更长远的展望。2.🇲🇾马来西亚和🇵🇭菲律宾相对越南AI发展处于较低水平。3.🌐东南亚地区的AI发展前景备受关注。站长网2023-10-30 16:22:150000iPhone15机型基本敲定 入门款机型有望搭载4800万主摄
iPhone15要到9月才会发布,但是现在已经开始引起人们的兴趣了。苹果没有正式公布其下一代旗舰手机的存在,但是有很多可信度高的iPhone15传闻让我们可以期待一下。站长网2023-05-15 09:31:210000Stack Overflow 调查显示开发人员已准备好使用 AI 工具辅助编程
AI工具在编码问答网站StackOverflow的开发者调查中被发现正变得普遍,尽管编码人员对其准确性持怀疑态度。这项调查在该网站正面临处理AI生成内容引发的版主罢工时进行,对其来说正值一个有趣的时刻。调查发现,77%的受访者对在工作流程中使用AI持积极态度,70%的人已经使用或计划在今年使用AI编码工具。站长网2023-06-16 14:38:590000