InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Nets

InfoGAN:通过信息最大化的生成对抗网络进行的可解释表示的学习

摘要:
这篇论文描述了InfoGAN,一种对于对抗生成网络的信息理论上的扩展,它能够以完全无监督的方式学习分离的表达。InfoGAN是一个对抗生成网络,它也会最大化隐藏变量的一个小的子集和观察数据之间的互信息。我们推出了可以被高效优化的互信息目标函数的下界。特别地说,InfoGAN成功了从MNIST数据集的数字形状中分离出了书写风格,从3D渲染图片的光照中分离出了姿态,以及SVHN数据集的中央数字中分离出了背景数字。它也从CelebA人脸数据集中发现了一些包络发型,是否戴眼镜和表情等视觉概念。实验表明,InfoGAN学习到了可解释的表达,这些表达比现有的监督方法学习到的表达更有竞争力。

本文提出的GAN的架构如下图所示,生成器G的输入不仅仅是噪声z,而是增加了一个隐含变量的c,这个隐含变量在无监督学习中,并不明确其具体指定的含义,但是就是需要分离度语义信息。辨别器D其实存在两个,一个依旧是分别数据真伪结果,一个给出的是条件概率分布Q(c|x),但是这两个辨别器(D/Q)共用前面所有的卷积层,只是最后分别用不同的全连接层得到最后的输出结果。


这样的网络设计,在原有的GAN训练的loss函数中,加入了一个互信息项的loss,以鼓励生成器G在生成数据的时候,不仅仅使用噪声z,同时也利用隐藏变量c。设计过程中,是要求c与生成的数据的互信息形成新的loss项,但是需要计算条件分布存在困难,于是进行的数学换算过程,然后加入Q的辅助分布辅助这一计算,因此加入的loss项如下:


在LI(G,Q)中,将H(c)看做常数项,那么优化过程中,只要优化公式5中的前面一部分。对于离散变量c,文中指出其概率是将Q(c|x)通过softmax计算获得的结果(softmax一般用于概率的计算,然后Q(c|x)输出被认为本身就带有其概率,因而用softmax做概率的归一处理);对于连续的变量c,文中指出采用高斯分布来计算其概率,也就是说直接使用正态分布的概率密度函数直接计算某数值的概率即可。最终整个模型的训练的loss如下:

这篇文章中采用的互信息的内容,来保证输入的noise和产生的图片之间的联系,并且使用由此设计的Loss来强约束两者之间的关系,最终确保了模型学到了从noise中学到对应的语义信息来生成图片。从某种意义上而言,除了互信息的约束,可以尝试其他的约束来绑定noise和生成的图片之间的联系,从而设计新的loss,可以得到新的GAN的模型用来生成与noise息息相关的图片。

代码有两个版本的实现:

openAI的工业化程度比较高的版本,个人觉得这个代码比较晦涩难懂,同时有些地方可能有点问题:

https://github.com/openai/InfoGAN

比较易懂的代码,Tensorflow版本实现的,推荐先看这个版本,再去看上面那个版本会轻松很多:

https://github.com/JonathanRaiman/tensorflow-infogan

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,905评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,140评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,791评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,483评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,476评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,516评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,905评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,560评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,778评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,557评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,635评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,338评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,925评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,898评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,142评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,818评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,347评论 2 342

推荐阅读更多精彩内容