这篇博士论文在 Tishby 的指导下完成,汇集了师徒二人及其他合著者在深度学习 + 信息论领域的研究成果,非常值得一读。
深度学习
的黑箱属性一直为人所诟病,很多研究者都在努力解决这一问题。其中,用
信息论
来提高
深度
神经网络
的可解释性是一个非常有趣的方向。
在这个方向上,「信息瓶颈」提出者、希伯来大学计算机科学教授 Naftali Tishby 及其学生的论文属于必读文献。
2015 年,Tishby 和他的学生 Noga Zaslavsky 发表了一篇论文,假设
深度学习
是一个信息瓶颈程序,尽可能地压缩数据噪声,保留数据想表达的信息。也就是说,
神经网络
就像把信息挤进瓶颈一样,只留下与一般概念最为相关的特征,去掉大量无关的
噪音
数据。
2017 年,Tishby 和他的另一个学生 Ravid Shwartz-Ziv 联合进行了一场引入注目的实验,揭示了发生在
深度学习
之中的挤压过程,表明
深度
神经网络
在压缩状态中提高泛化能力,从而更加擅长标记测试数据。这篇论文就是大名鼎鼎的《 Opening the black box of Deep Neural Networks via
Infor
mation 》。
深度学习
先驱 Geoffrey Hinton 在听了 Tishby 的报告之后给他发了邮件:「信息瓶颈极其有趣,我要再听一万遍才能真正理解它,当今能听到如此原创的想法非常难得,或许它就是解开谜题的那把钥匙。」纽约大学心理学和
数据科学
助理教授 Brenden Lake 也认为,Tishby 的研究成果是「打开
神经网络
黑箱的重要一步」。
在 Tishby 去世的这年,Ravid Shwartz-Ziv 完成了他的博士论文 ——《
Infor
mation Flow in Deep Neural Networks 》。
这篇论文在 Tishby 的指导下完成,汇集了师徒二人及其他合著者在
深度学习
+
信息论
领域的研究成果,非常值得一读。
最近,Ravid Shwartz-Ziv 表示,他已经将该论文上传到了 arXiv。
论文链接:https://arxiv.org/pdf/2202.06749.pdf
尽管
深度
神经网络
已经取得了巨大的成功,但我们还没有一个全面的理论来解释这些网络如何工作或如何构造。深度网络通常被视为黑盒,我们无法清楚地解释它们的预测结果或可靠性。如今,了解
深度
神经网络
的突破性性能是科学界面临的最大挑战之一。为了更有效地使用这些算法并改进它们,我们需要了解它们的动态行为(dynamic behavior)以及它们学习新表示的能力。
在这篇博士论文中,作者应用了
信息论
中的原理和技术来解决上述问题,以提高我们的理论理解,并运用这一理解来设计更好的算法。
第二章和第三章介绍了作者针对深度学习
模型提出的信息论
方法。
作为对
深度学习
系统的解释,作者提出使用信息瓶颈(IB)理论。这种分析网络的新范式揭示了它们的分层结构、泛化能力和学习动态。基于这一分析,作者发现深度网络优化了每一层关于输入和输出变量的互信息,导致每一层都要在压缩和预测之间做出权衡。作者对这些网络的分析和数值研究表明,
随机梯度下降
算法遵循 IB 权衡原则,分两个阶段工作:快速经验误差最小化阶段和慢速表示压缩阶段。这些阶段通过每层不同的信噪比(SNR)来区分。
此外,他们证明了 SGD 由于压缩阶段而达到了这个最优界限,并在表示压缩上导出了一个新的高斯界限,同时将其与压缩时间联系起来。他们的结果还表明,网络的层
收敛
到 IB 理论界限,导致编码器和解码器分布之间的自洽关系。
第四章讨论了将 IB 应用于深度神经网络
时最困难的问题之一 —— 估计高维空间中的互信息。
尽管互信息在
数据科学
中是一个重要的量,但它在计算方面一直很有挑战性。互信息的计算只适用于离散变量或已知
概率分布
的有限数量的问题。为了更好地估计
信息论
数量(information-theoretic quantities)并研究泛化信号,作者研究了几个框架,并利用了最近的理论研究成果,如神经正切核(NTK)框架。
在研究中,他们发现,对于无限宽的
神经网络
的无限集合,他们可以获得许多
信息论
数量及其边界的易处理的计算。许多量可以通过网络的核(kernel)以封闭形式的解来描述。通过分析这些内容,我们可以了解网络的重要
信息论
数量,以及压缩、泛化和样本大小之间的关系。
第五章介绍了一个新的信息论
框架 —— 双重信息瓶颈(dualIB)。
尽管 IB 框架有很多优点,但它也有几个缺点:IB 是完全非
参数
化的,并且只在概率空间上运行。此外,IB 公式不涉及预测未见模式的任务,并假定能够完全获取联合概率。因此,作者开发了 dualIB,它仅通过在失真函数中的项之间进行切换来解决 IB 的一些缺点。dualIB 可以解释数据的已知特征,并使用它们对未知的例子做出更好的预测。作者提供了 dualIB 自洽方程,该方程允许他们获得解析解。局部稳定性分析揭示了解的临界点的潜在结构,产生了最佳模式表示的完整分叉图。
他们发现了 dualIB 目标的几个有趣的性质。首先,当以
参数
化形式表达时,dualIB 保留其结构。它还优化了平均预测误差指数,从而提高了关于样本大小的预测精度。除了 dualIB 的解析解,他们还提供了一个变分 dualIB 框架,该框架使用
深度
神经网络
来优化泛函。该框架实现了真实世界数据集的 dualIB 的实现。通过它,作者对其动态变化进行了实验评估,并验证了现代
深度
神经网络
的理论预测。
总之,该论文提出了一个新的
信息论
视角来研究
深度
神经网络
,它利用了
深度学习
和 IB 框架之间的对应关系。这一独特的视角可以提供许多好处,比如对
深度
神经网络
有更深入的理解,解释它们的行为,并改善它们的性能。同时,他们的研究也开拓了新的理论和实践研究问题。
Ravid Shwartz-Ziv 是希伯来大学计算机科学与
神经科学
博士,读博期间专注于
深度
神经网络
与
信息论
之间的联系,试图基于
信息论
对 DNN 进行更深入的理解,并将其应用于大规模的问题。2018 年,他拿到了谷歌博士生奖研金。
2021 年,Shwartz-Ziv 博士毕业并进入纽约大学
数据科学
中心担任 CDS Faculty Fellow,与 Andrew Gordon Wilson、Yann Lecun 等人一起研究贝叶斯深度网络、
信息论
和
自监督学习
。
Google Scholar 主页:https://scholar.google.co.il/citations?user=SqsLFwMAAAAJ&hl=en
理论
博士论文
Naftali Tishby
信息瓶颈
信息论
深度学习