国际数学家大会丨鄂维南院士作一小时大会报告:从数学角度,理解机器学习的“黑魔法”,并应用于更广泛的科学问题...

64da919667b309586a5daabd1860bcd9.jpeg

来源:大数据分析与应用国家工程实验室

北京时间2022年7月8日晚上22:30,鄂维南院士在2022年的国际数学家大会上作一小时大会报告(plenary talk)。今天我们带来鄂老师演讲内容的分享。鄂老师首先分享了他对机器学习数学本质的理解(函数逼近、概率分布的逼近与采样、Bellman方程的求解);然后介绍了机器学习模型的逼近误差、泛化性质以及训练等方面的数学理论;最后介绍如何利用机器学习来求解困难的科学计算和科学问题,即AI for science。

46edf0fa66970eb17aa1f6764807f12a.jpeg

机器学习问题的数学本质

众所周知,机器学习的发展,已经彻底改变了人们对人工智能的认识。机器学习有很多令人叹为观止的成就,例如:

· 比人类更准确地识别图片:利用一组有标记的图片,机器学习算法可以准确地识别图片的类别:

9d570fdda32a6fdb75a43ad41b77b2a1.jpeg

Cifar-10 问题:把图片分成十个类别

来源:https://www.cs.toronto.edu/~kriz/cifar.html

· Alphago下围棋打败人类:完全由机器学习实现下围棋的算法:

694d0caeb90917ae87cec678247fadf8.jpeg

参考:https://www.bbc.com/news/technology-35761246

· 产生人脸图片,达到以假乱真的效果:

b0740688d6cfebf23572f20c3e9182dd.jpeg

参考:https://arxiv.org/pdf/1710.10196v3.pdf

机器学习还有很多其他的应用。在日常生活中,人们甚至常常使用了机器学习所提供的服务而不自知,例如:我们的邮件系统里的垃圾邮件过滤、我们的车和手机里的语音识别、我们手机里的指纹解锁……

所有这些了不起的成就,本质上,却是成功求解了一些经典的数学问题。

对于图像分类问题,我们感兴趣的其实是函数65e6d77901a499d8736adf3bc84ae09e.jpeg

48510ebbba3f61e6456b20e186f2cb57.jpeg: 图像→类别

函数6b9e5bbc4aba428d3ab54a784288876e.png把图像映射到该图像所属的类别。我们知道1c8cecb91a4822053a6cab4751288578.png在训练集上的取值,想由此找到对函数7c55d9ccc4c0afe2bd45bbebada2eed4.png的一个足够好的逼近

一般而言,监督学习(supervised learning)问题,本质都是想基于一个有限的训练集S,给出目标函数的一个高效逼近

对于人脸生成问题,其本质是逼近并采样一个未知的概率分布。在这一问题中,“人脸”是随机变量,而我们不知道它的概率分布。然而,我们有“人脸”的样本:数量巨大的人脸照片。我们便利用这些样本,近似得到“人脸”的概率分布,并由此产生新的样本(即生成人脸)。

一般而言,无监督学习本质就是利用有限样本逼近并采样问题背后未知的概率分布

对于下围棋的Alphago来说,如果给定了对手的策略,围棋的动力学是一个动态规划问题的解。其最优策略满足Bellman方程。因而Alphago的本质便是求解Bellman方程。

一般而言,强化学习本质上就是求解马尔可夫过程的最优策略

然而,这些问题都是计算数学领域的经典问题!!毕竟,函数逼近、概率分布的逼近与采样,以及微分方程和差分方程的数值求解,都是计算数学领域极其经典的问题。那么,这些问题在机器学习的语境下,到底和在经典的计算数学里有什么区别呢?答案便是:

维度(dimensionality)

例如,在图像识别问题中,输入的维度为9fa66e4a3807006336968fa4d56d10a1.jpeg。而对于经典的数值逼近方法,对于9c82935b1576dce792a2996ea2963aa7.jpeg维问题,含72a649eae7c54a7a5b79ba58c6d57a14.png个参数的模型的逼近误差038b6c59b80ecc9628fd757533a64c68.jpeg. 换言之,如果想将误差缩小10倍,参数个数需要增加f6dd99f045d4b0ae73a2973ba5f9bab7.jpeg. 当维数380efb06244b7a88c5b7047a57afb877.jpeg增加时,计算代价呈指数级增长。这种现象通常被称为:

维度灾难(curse of dimensionality)

所有的经典算法,例如多项式逼近、小波逼近,都饱受维度灾难之害。很明显,机器学习的成功告诉我们,在高维问题中,深度神经网络的表现比经典算法好很多。然而,这种“成功”是怎么做到的呢?为什么在高维问题中,其他方法都不行,但深度神经网络取得了前所未有的成功呢?

从数学出发,理解机器学习的“黑魔法”:监督学习的数学理论

2.1 记号与设定

神经网络是一类特殊的函数。比如,两层神经网络是:

105767881abb99c5f51b15cd1996c99f.jpeg

其中有两组参数,b81c1f0c9c3f011df300dcaa8feb794c.jpegca491350cb2d68391ae7abb376449205.jpegb34dc2534e85f2dc744d798f22566d41.png是激活函数,可以是:

· a81df6ec6c4787e6e54d8ae9a298fce7.jpeg,ReLU函数;

· 6ab2b79cf2681a12d3455527b707ab8b.jpeg,Sigmoid函数。

而神经网络的基本组成部分即为:线性变换与一维非线性变换。深度神经网络,一般就是如下结构的复合:

f87c912425486b47828878909ee783e4.jpeg

53d43a1472caf8d4897db8b7f3d7ec61.jpeg

为了简便,我们在此省略掉所有的bias项f8e869cdb0f159e1f12727f88f733c21.png0d8cac105028c1a2acffb297b2a491d6.png是权重矩阵,激活函数d0855018491e6fd5f7a94c442d904edc.png作用在每一个分量上。

我们将要在训练集S上逼近目标函数6a9b7c08bb881f0fc3015c598e7f034b.png

77cacd6ac285ed4ebed1cdee01eabe0f.jpeg

不妨假设f98874175e213a753a583b94fb78b1c2.png的定义域为a574dfd8f9d77b3098214f02022d6ff9.jpeg。令28ffd585f476ed58a23449982eea5a35.jpeg0ab053c2b13344461639ce4b8cf7d823.jpeg的分布。那么我们的目标便是:最小化测试误差988c6819513919d795f8cc47b229ae46.jpeg(testing error,也称为population risk或generalization error):

e38bb16d4c16aacc4b25d83df77f5a29.jpeg

2.2 监督学习的误差

监督学习一般有如下的步骤:

第一步:选取一个假设空间(测试函数的一个集合)5e0c00e2913af9b712ac461e4c111328.jpeg(m正比于测试空间的维数);

第二步:选取一个损失函数进行优化。通常,我们会选择经验误差(empirical risk)来拟合数据:

80c959dd29c125bd1f895015dce087f4.jpeg

有时,我们还会加上其他的惩罚项。

第三步:求解优化问题,如:

· 梯度下降:

fe426f0c8f09ac5c5d58b9c46460363a.jpeg

· 随机梯度下降:

83dd37d708e2f14fe5f4b1bdcbdc8178.jpeg

d6e5ef1b37c8437bfdea890f054f203d.jpeg是从1,…n中随机选取的。

如果把机器学习输出的结果记11c10e7ae36b22903a29e6d9ae55d216.jpeg,那么总误差便是2055d7e2fb37ac4b3decd13af9bdce13.jpeg。我们再定义:

e1e120845723ae55d24a61f5f3d310ed.jpeg是在假设空间里最好的逼近;

bd891f412f0e17670e8926ab3b5c923d.jpeg是在假设空间里,基于数据集S最好的逼近。

由此,我们便可以把误差分解成三部分:

24ab7cedfc26d5446e3a0d452847d822.jpeg

4d0aa58550846c42f148cf48f132019c.jpeg是逼近误差(approximation error):完全由假设空间的选取所决定;

199b92426911ba15afe31fb3d705be24.jpeg是估计误差(estimation error):由于数据集大小有限而带来的额外的误差;

6440ad78dbbeaf76fa553507168f0a6a.jpeg是优化误差(optimization error):由训练(优化)带来的额外的误差。

2.3 逼近误差

我们下面集中讨论逼近误差(approximation error)。

我们先用传统方法傅立叶变换做一个对比:

8f084e42052c51afa69aa34598748e43.jpeg

如果我们用离散的傅立叶变换来逼近:

86e6562cf92235f3fcdff0472490bc5f.jpeg

其误差5b76cfc4456ba98a19dc1232d0ce790b.jpeg便是正比于0861aad0d28167e3bd07e1d533ac7c0f.jpeg,毫无疑问地受到维度灾难的影响。

而如果一个函数可以表示成期望的形式:

224cfd329a012806d6978f50943616be.jpeg

f66fe076a39f2b3577303415836c99f0.png是测度5f03c747cb2e3029e2ad35574d478671.png的独立同分布样本,我们有:

be88637c1f9d1a4e69b165e4f57fcc1e.jpeg

那么此时的误差是:

fe0afe4f89ef6f77de8ad97d8db94972.jpeg

可以看到,这是与维数无关的!

如果让激活函数为bdde4df8008d8273eb06e605ef076533.jpeg,那么b8875ba69db22d9a615fb906f393e8e3.jpeg就是以7005a87b93e60e7c8d39acd47396875e.jpeg为激活函数的两层神经网络。此结果意味着:这一类(可以表示成期望)的函数,都可以由两层神经网络逼近,且近误差的速率与维数无关

对于一般的双层神经网络,我们可以得到一系列类似的逼近结果。其中关键的问题是:到底什么样的函数可以被双层神经网络逼近?为此,我们引入Barron空间的定义:

afc2637d5c93caf090276c6bb6f0af83.jpeg

Barron空间的定义

参考:E, Chao Ma, Lei Wu (2019)

对于任意的Barron函数,存在一个两层神经网络e087701f06fda8d9b43863a9042ef3c7.png,其逼近误差满足:

d41aff7bef8b4eab1a06b9049a391518.jpeg

可以看到这一逼近误差与维数无关!(关于这部分理论的细节,可以参考: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 泛化性:训练误差与测试误差的差别

人们一般会期待,训练误差与测试误差的差别会正比于ea4a12ce9741a038b3e4e6d8367b7b36.png(n是样本数量)。然而,我们训练好的机器学习模型和训练数据是强相关的,这导致这样子的Monte-Carlo速率不一定成立。为此,我们给出了如下的泛化性理论:

da6c282525410b807d3d4405cb07a916.jpeg

简言之,我们用Rademacher复杂度来刻画一个空间在数据集上拟合随机噪声的能力。Rademacher复杂度的定义为:

6de9fa54fb3f2d07335c5c902faf68f8.jpeg

其中0bdc5217cc6e601591f655a6d21286e1.png是取值为1或-1的独立同分布的随机变量。

951264c3c70969a701cacf817a1694b5.png是李朴西斯空间中的单位球时,其Rademacher复杂度正比于6591f97efbfa7c9b8ef9c2201419127b.jpeg

当d增加时,可以看到拟合需要的样本大小指数上升。这其实是另一种形式的维度灾难。

2.5 训练过程的数学理解

关于神经网络的训练,有两个基本的问题:

梯度下降方法到底能不能快速收敛?

训练得到的结果,是否有比较好的泛化性?

对于第一个问题,答案恐怕是悲观的。Shamir(2018)中的引理告诉我们,基于梯度的训练方法,其收敛速率也受维度灾难的影响。而前文提到的Barron space,虽然是建立逼近理论的好手段,但对于理解神经网络的训练却是一个过大的空间。

特别地,这样子的负面结果可以在高度超参数(highly over-parameterized regime)的情形(即m>>n)下得到具体刻画。在此情形下,参数的动力学出现了尺度分离的现象:对于如下的两层神经网络:

e9ed01f85acc577f01c42d15d71814e9.jpeg

在训练过程中,dce5c9bed7a599394681eff23d0e0888.png的动力学分别为:

8bb5afd61e901d51bfc359ff68522528.jpeg

由此可以看到尺度分离的现象:当m很大的时候,7b98162f83d69c5d8e2bf6e7290e26bc.jpeg的动力学几乎被冻结住。

这种情形下,好消息是我们有了指数收敛(Du et al, 2018);坏消息却是这时候,神经网络表现得并不比从random feature model模型好。

我们也可以从平均场的角度理解梯度下降方法。令:896080f8d4f884bf0e768863d578a595.jpeg,并令:

2af5a530a24b9f8864226a0855efc4da.jpeg

c9da056c13ef82fde3886f01b50c4778.jpeg

54f0869ad4afea41914c6d3fa5113911.png是下列梯度下降问题的解:

766950cc9db3950fe4c0c8b2182d5598.jpeg

当且仅当9190c775e0a85c1ac7152ab16ba4c9a7.png是下面方程的解(参考:Chizat and Bach (2018), Mei, Montanari and Nguyen (2018), Rotsko  and Vanden-Eijnden (2018), Sirignano and Spiliopoulos (2018)):

bf2e79b145bd8ae1d4449032a0738db6.jpeg

这一平均场动力学,实际上是在Wassenstein度量意义下的梯度动力学。人们证明了:如果其初始值91af9a44b05e597b8d576e3d6e892114.png的支集为全空间,且梯度下降的确收敛,那么其收敛结果必然是全局最优(参考:Chizat and Bach (2018,2020), Wojtowytsch (2020))。

机器学习的应用

3.1 解决高维科学计算问题

既然机器学习是处理高维问题的有效工具,我们便可运用机器学习解决传统计算数学方法难以处理的问题。

第一个例子便是随机控制问题。传统方法求解随机控制问题需要求解一个极其高维的Bellman方程。运用机器学习方法,可以有效求解随机控制问题。其思路与残差神经网络颇为类似(参考Jiequn Han and E (2016)):

de1e4cf34866fad50548cc85412c48ce.jpeg

第二个例子便是求解非线性抛物方程。非线性抛物方程可以被改写成一个随机控制问题,其极小点是唯一的,对应着非线性抛物方程的解。

75c6995a89e79644c63b63ee0e1adb78.jpeg

3.2 AI for science

利用机器学习处理高维问题的能力,我们可以解决更多科学上的难题。这里我们举两个例子。第一个例子是Alphafold。

19896ad391e513b64eccf129ec5bc0da.jpeg

参考:J. Jumper et al. (2021)

第二个例子,便是我们自己的工作:深度势能分子动力学(DeePMD)。这是能达到从头计算精度的分子动力学。我们所使用的新的模拟“范式”便是:

利用量子力学第一性原理计算提供数据

利用神经网络,给出势能面准确的拟合(参考:Behler and Parrinello (2007), Jiequn Han et al (2017), Linfeng Zhang et al (2018))。

运用DeePMD,我们能够模拟一系列材料和分子,可以达到第一性层面的计算精度

02d80599c7eced77978bac7c4ba0c4bf.jpeg

我们还实现了一亿原子的第一性原理精度的模拟,获得了2020年的戈登贝尔奖

8007058f15d1b3534cf3fe4531947a94.jpeg

参考:Weile Jia, et al, SC20, 2020 ACM Gordon Bell Prize

我们给出了水的相图

1f70b7a3e2c10488fb1c2afee12b7f6b.jpeg

参考:Linfeng Zhang, Han Wang, et al. (2021)

而事实上,物理建模横跨多个尺度:宏观、介观、微观,而机器学习恰好提供了跨尺度建模的工具

754e35f39efa6e648c4ca18a73365102.jpeg

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);

连续介质动力学: 2e34b31c77f017967d7bff2df105aef6.jpeg (2020)

在未来五到十年,我们有可能做到:跨越所有物理尺度进行建模和计算。这将彻底改变我们如何解决现实问题:如药物设计、材料、燃烧发动机、催化……

fa55eed3678a449ca79db2cdd1c33a74.jpeg

总结

机器学习根本上是高维中的数学问题。神经网络是高维函数逼近的有效手段;这便为人工智能领域、科学以及技术领域提供了众多新的可能性。

这也开创了数学领域的一个新主题:高维的分析学。简而言之,可以总结如下:

监督学习:高维函数理论;

无监督学习:高维概率分布理论;

强化学习:高维Bellman方程;

时间序列学习:高维动力系统。

a31a51d99bb5c4cfbf5ffc89973ee64a.jpeg

outside_default.png

鄂维南,中国科学院院士,美国数学学会会士,美国工业与应用数学学会会士,英国物理学会会士。北京大学讲席教授、北京大数据研究院院长、大数据分析与应用技术国家工程实验室主任、北京大学国际机器学习研究中心主任、北京科学智能研究院院长。1982年毕业于中国科学技术大学数学系,1985年获中国科学院计算中心硕士学位,1989年获美国加州大学洛杉矶分校博士学位。主要从事计算数学、应用数学及其在力学、物理、化学和工程等领域中的应用等方面的研究。曾获首届美国青年科学家和工程师总统奖,冯康科学计算奖,国际工业与应用数学协会颁发的Collatz奖,美国工业与应用数学学会R. E. Kleinman奖,美国工业与应用数学学会卡门 (Theodore von Karman) 奖,SIAM和ETH Zürich联合授予的Peter Henrici奖及ACM戈登·贝尔奖(ACM Gordon Bell Prize)。

未来智能实验室的主要工作包括:建立AI智能系统智商评测体系,开展世界人工智能智商评测;开展互联网(城市)大脑研究计划,构建互联网(城市)大脑技术和企业图谱,为提升企业,行业与城市的智能水平服务。每日推荐范围未来科技发展趋势的学习型文章。目前线上平台已收藏上千篇精华前沿科技文章和报告。

  如果您对实验室的研究感兴趣,欢迎加入未来智能实验室线上平台。扫描以下二维码或点击本文左下角“阅读原文”

outside_default.png

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值