机器学习数学本质的理解
来源:计算机视觉与机器学习
近日,国际数学家大会丨鄂维南院士作一小时大会报告:从数学角度,理解机器学习的“黑魔法”,并应用于更广泛的科学问题。
鄂维南院士在2022年的国际数学家大会上作一小时大会报告(plenary talk)。
今天我们带来鄂老师演讲内容的分享。
鄂老师首先分享了他对机器学习数学本质的理解(函数逼近、概率分布的逼近与采样、Bellman方程的求解);
然后介绍了机器学习模型的逼近误差、泛化性质以及训练等方面的数学理论;
最后介绍如何利用机器学习来求解困难的科学计算和科学问题,即AI for science。
机器学习问题的数学本质
众所周知,机器学习的发展,已经彻底改变了人们对人工智能的认识。机器学习有很多令人叹为观止的成就,例如:
比人类更准确地识别图片:利用一组有标记的图片,机器学习算法可以准确地识别图片的类别:
Cifar-10 问题:把图片分成十个类别
来源:https://www.cs.toronto.edu/~kriz/cifar.html
Alphago下围棋打败人类:完全由机器学习实现下围棋的算法:
参考:https://www.bbc.com/news/technology-35761246
产生人脸图片,达到以假乱真的效果:
参考:https://arxiv.org/pdf/1710.10196v3.pdf
机器学习还有很多其他的应用。在日常生活中,人们甚至常常使用了机器学习所提供的服务而不自知,例如:我们的邮件系统里的垃圾邮件过滤、我们的车和手机里的语音识别、我们手机里的指纹解锁……
所有这些了不起的成就,本质上,却是成功求解了一些经典的数学问题。
对于图像分类问题,我们感兴趣的其实是函数:
: 图像→类别
函数把图像映射到该图像所属的类别。我们知道在训练集上的取值,想由此找到对函数的一个足够好的逼近。
一般而言,监督学习(supervised learning)问题,本质都是想基于一个有限的训练集S,给出目标函数的一个高效逼近。
对于人脸生成问题,其本质是逼近并采样一个未知的概率分布。在这一问题中,“人脸”是随机变量,而我们不知道它的概率分布。然而,我们有“人脸”的样本:数量巨大的人脸照片。我们便利用这些样本,近似得到“人脸”的概率分布,并由此产生新的样本(即生成人脸)。
一般而言,无监督学习本质就是利用有限样本,逼近并采样问题背后未知的概率分布。
对于下围棋的Alphago来说,如果给定了对手的策略,围棋的动力学是一个动态规划问题的解。其最优策略满足Bellman方程。因而Alphago的本质便是求解Bellman方程。
一般而言,强化学习本质上就是求解马尔可夫过程的最优策略。
然而,这些问题都是计算数学领域的经典问题!!毕竟,函数逼近、概率分布的逼近与采样,以及微分方程和差分方程的数值求解,都是计算数学领域极其经典的问题。那么,这些问题在机器学习的语境下,到底和在经典的计算数学里有什么区别呢?答案便是:维度(dimensionality)。
例如,在图像识别问题中,输入的维度为。而对于经典的数值逼近方法,对于维问题,含个参数的模型的逼近误差. 换言之,如果想将误差缩小10倍,参数个数需要增加. 当维数增加时,计算代价呈指数级增长。这种现象通常被称为:维度灾难(curse of dimensionality)。
所有的经典算法,例如多项式逼近、小波逼近,都饱受维度灾难之害。很明显,机器学习的成功告诉我们,在高维问题中,深度神经网络的表现比经典算法好很多。然而,这种“成功”是怎么做到的呢?为什么在高维问题中,其他方法都不行,但深度神经网络取得了前所未有的成功呢?
从数学出发,理解机器学习的“黑魔法”:监督学习的数学理论
2.1 记号与设定
神经网络是一类特殊的函数。比如,两层神经网络是
其中有两组参数,和。是激活函数,可以是: ,ReLU函数;Sigmoid函数。而神经网络的基本组成部分即为:线性变换与一维非线性变换。深度神经网络,一般就是如下结构的复合:
为了简便,我们在此省略掉所有的bias项。是权重矩阵,激活函数作用在每一个分量上。
我们将要在训练集S上逼近目标函数不妨假设X的定义域为。令为x的分布。那么我们的目标便是:最小化测试误差(testing error,也称为population risk或generalization error)
2.2 监督学习的误差
监督学习一般有如下的步骤:
第一步:选取一个假设空间(测试函数的一个集合)(m正比于测试空间的维数);
第二步:选取一个损失函数进行优化。通常,我们会选择经验误差(empirical risk)来拟合数据:
有时,我们还会加上其他的惩罚项。
第三步:求解优化问题,如:
梯度下降:
随机梯度下降:
是从1,…n中随机选取的。
如果把机器学习输出的结果记,那么总误差便是。我们再定义:
是在假设空间里最好的逼近;
是在假设空间里,基于数据集S最好的逼近。
由此,我们便可以把误差分解成三部分:
是逼近误差(approximation error):完全由假设空间的选取所决定;
是估计误差(estimation error):由于数据集大小有限而带来的额外的误差;
是优化误差(optimization error):由训练(优化)带来的额外的误差。
2.3 逼近误差
我们下面集中讨论逼近误差(approximation error)。
我们先用传统方法傅立叶变换做一个对比:
如果我们用离散的傅立叶变换来逼近:其误差便是正比于,毫无疑问地受到维度灾难的影响。而如果一个函数可以表示成期望的形式:令是测度的独立同分布样本,我们有:那么此时的误差是:可以看到,这是与维数无关的!
如果让激活函数为,那么就是以为激活函数的两层神经网络。此结果意味着:这一类(可以表示成期望)的函数,都可以由两层神经网络逼近,且逼近误差的速率与维数无关!
对于一般的双层神经网络,我们可以得到一系列类似的逼近结果。其中关键的问题是:到底什么样的函数可以被双层神经网络逼近?为此,我们引入Barron空间的定义:
Barron空间的定义参考:E, Chao Ma, Lei Wu (2019)
对于任意的Barron函数,存在一个两层神经网络,其逼近误差满足:
可以看到这一逼近误差与维数无关!(关于这部分理论的细节,可以参考:E, Ma and Wu (2018, 2019), E and Wojtowytsch (2020)。其他的关于Barron space的分类理论,可以参考Kurkova (2001), Bach (2017),Siegel and Xu (2021))
类似的理论可以推广到残差神经网络(residual neural network)。在残差神经网络中,我们可以用流-诱导函数空间(flow-induced function space)替代Barron空间。
2.4 泛化性:训练误差与测试误差的差别
人们一般会期待,训练误差与测试误差的差别会正比于(n是样本数量)。然而,我们训练好的机器学习模型和训练数据是强相关的,这导致这样子的Monte-Carlo速率不一定成立。为此,我们给出了如下的泛化性理论:
简言之,我们用Rademacher复杂度来刻画一个空间在数据集上拟合随机噪声的能力。
Rademacher复杂度的定义为:其中是取值为1或-1的独立同分布的随机变量。
当d增加时,可以看到拟合需要的样本大小指数上升。这其实是另一种形式的维度灾难。
2.5 训练过程的数学理解
梯度下降方法到底能不能快速收敛?
训练得到的结果,是否有比较好的泛化性?
对于第一个问题,答案恐怕是悲观的。Shamir(2018)中的引理告诉我们,基于梯度的训练方法,其收敛速率也受维度灾难的影响。而前文提到的Barron space,虽然是建立逼近理论的好手段,但对于理解神经网络的训练却是一个过大的空间。
特别地,这样子的负面结果可以在高度超参数(highly over-parameterized regime)的情形(即m>>n)下得到具体刻画。在此情形下,参数的动力学出现了尺度分离的现象:对于如下的两层神经网络:在训练过程中,的动力学分别为:。
我们也可以从平均场的角度理解梯度下降方法。令:,并令:则是下列梯度下降问题的解:当且仅当是下面方程的解(参考:Chizat and Bach (2018), Mei, Montanari and Nguyen (2018), Rotsko and Vanden-Eijnden (2018), Sirignano and Spiliopoulos (2018)):这一平均场动力学,实际上是在Wassenstein度量意义下的梯度动力学。人们证明了:如果其初始值的支集为全空间,且梯度下降的确收敛,那么其收敛结果必然是全局最优(参考:Chizat and Bach (2018,2020), Wojtowytsch (2020))。
机器学习的应用
第一个例子便是随机控制问题。传统方法求解随机控制问题需要求解一个极其高维的Bellman方程。运用机器学习方法,可以有效求解随机控制问题。其思路与残差神经网络颇为类似(参考Jiequn Han and E (2016)):
利用量子力学第一性原理计算提供数据;
利用神经网络,给出势能面准确的拟合(参考:Behler and Parrinello (2007), Jiequn Han et al (2017), Linfeng Zhang et al (2018))。
运用DeePMD,我们能够模拟一系列材料和分子,可以达到第一性层面的计算精度:
我们还实现了一亿原子的第一性原理精度的模拟,获得了2020年的戈登贝尔奖:
我们给出了水的相图:
参考:Linfeng Zhang, Han Wang, et al. (2021)
而事实上,物理建模横跨多个尺度:宏观、介观、微观,而机器学习恰好提供了跨尺度建模的工具。
AI for science,即用机器学习解决科学问题,已经有了一系列重要的突破,如:
量子多体问题:RBM (2017), DeePWF (2018), FermiNet (2019),PauliNet (2019),…;
密度泛函理论: DeePKS (2020), NeuralXC (2020), DM21 (2021), …;
分子动力学: DeePMD (2018), DeePCG (2019), …;
动理学方程: 机器学习矩封闭 (Han et al. 2019);
连续介质动力学: (2020)
在未来五到十年,我们有可能做到:跨越所有物理尺度进行建模和计算。这将彻底改变我们如何解决现实问题:如药物设计、材料、燃烧发动机、催化……
总结
机器学习根本上是高维中的数学问题。神经网络是高维函数逼近的有效手段;这便为人工智能领域、科学以及技术领域提供了众多新的可能性。
这也开创了数学领域的一个新主题:高维的分析学。简而言之,可以总结如下:
监督学习:高维函数理论;
无监督学习:高维概率分布理论;
强化学习:高维Bellman方程;
时间序列学习:高维动力系统。
加入知识星球【我们谈论数据科学】
500+小伙伴一起学习!