Information Bottleneck信息瓶颈理论

可以理解为一个损失函数
信息瓶颈理论把神经网络理解为一个编码器和一个解码器,编码器将输入x编码成Z,解码器将Z解码成输出y
而信息瓶颈理论的目标则是
在这里插入图片描述
R I B ( θ ) R_{IB}(\theta) RIB(θ)就是信息瓶颈, θ \theta θ是网络的参数,也是要优化的东西
I ( Z , Y ; θ ) I(Z,Y;\theta) I(Z,Y;θ)就是输出Y和中间feature Z的互信息
I ( Z , X ; θ ) I(Z,X;\theta) I(Z,X;θ)是输入X和中间feature Z的互信息
互信息参考互信息的定义
在决策树的C4.5算法中又被叫做信息增益
信息瓶颈理论的本质就是:最大化Z和Y的互信息,尽量减少X和Z的互信息

互信息也可以简单理解成互相包含的信息。
按照这种理解,减少X和Z的互信息,同时增大Z和Y的互信息,实际上是希望Z中尽量减少X的与Y不相关的信息,保留X的和Y最相关的那部分信息。

通过公式推导,可以得到 R I B ( θ ) R_{IB}(\theta) RIB(θ)的下界L,即 R I B ( θ ) 》 L R_{IB}(\theta)》L RIB(θ)L,最大化IB相当于最大化L,取L的相反数为 J I B J_{IB} JIB,即最大化IB相当于最小化 J I B J_{IB} JIB,因此,可以把 J I B J_{IB} JIB作为模型的损失函数:
在这里插入图片描述

上式中, q ( y n ∣ z ) q(y_n|z) q(ynz)即为模型的decoder, p ( z ∣ x n ) p(z|x_n) p(zxn)即为模型的encoder, ∣ ∣ || 是和的意思,即求这两者的KL散度

总结:
IB理论把深度学习阶段分为两部分,前一段时间尽量增加中间feature和Y的互信息,后一段时间尽量压缩X和中间feature的互信息,使得中间feature包含X最精华的信息。

  • 7
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值