训练网络像是买彩票?神经网络剪枝最新进展之彩票假设解读澎湃号·湃客澎湃新闻

神经网络剪枝技术可以极大的减少网络的参数,并降低存储要求,和提高推理的计算性能。而且目前这方面最好的方法通常能保持很高的准确性。因此对通过修剪产生的稀疏架构的研究是一个很重要的方向。本选题的思路是对以下两篇论文做深度解读,一探当今最好的剪枝方法的究竟。

现在有了性能更强的GPU,计算一个更深的神经网络、参数更多的神经网络根本不成问题。但事实上并不是每个人都是人手几张卡的,对于具有更多层和节点的神经网络,减少其存储和计算成本变得至关重要。并且,随着移动设备和可穿戴设备的普及,如何让这些模型在计算能力并不强的移动端也能很好地应用,也成为亟待解决的问题。因此越来越多的研究者开始研究神经网络模型压缩。

综合现有的模型压缩方法,它们主要分为四类:参数剪枝和共享(parameterpruningandsharing)、低秩分解(low-rankfactorization)、转移和紧凑卷积核(transferred/compactconvolutionalfilters)、知识蒸馏(knowledgedistillation)[1]。

论文1:TheLotteryTicketHypothesis:FindingSparse,TrainableNeuralNetworks

介绍

机器学习的训练过程,是数据科学家在理论与现实之间面临的妥协之一。通常情况下,对于特定问题而言,由于训练成本的限制,理想的神经网络架构不能完全实现。一般而言,神经网络最初的训练需要大量的数据集和昂贵的计算成本,其结果得到一个隐藏层之间充满复杂连接的庞大的神经网络结构。这种结构往往需要经过优化技术,移除某些连接来调整模型的大小。数十年来困扰研究者的一个问题是我们是否真的需要这样的庞大的神经网络结构。很明显,如果我们连接网络中的每个神经元,可以解决特定的问题,但可能因为高昂的成本而被迫止步。难道我们不能从更小、更精简的网络开始训练吗?这就是彩票假设的本质。

以博彩来类比,训练机器学习模型就相当于通过购买每一张可能的彩票来获得中奖彩票。但是如果我们知道中奖彩票长什么样子,那么是不是就可以更聪明地来选择彩票?在机器学习模型中,训练过程得到的巨大的神经网络结构相当于一大袋彩票。在初始训练之后,模型需要进行优化,例如剪枝,删除网络中不必要的权重,从而在不牺牲性能的情况下减小模型的大小。这就相当于在袋子中寻找中奖彩票,然后扔掉剩下的彩票。通常情况下,经过剪枝后的网络结构要比原始的小90%左右。那么问题来了,如果网络结构可以缩小,那么为了提高训练效率为什么不一开始就训练这个更小的网络呢?然而,已经有许多实验证明了,如果从头开始训练剪枝后的网络,得到的准确率要比原始网络低很多。

MIT的彩票假设背后的思想是,一个大型的神经网络包含一个小的子网络,如果一开始就训练,那么会得到和原始网络相似的准确率。

彩票假设

我们将一个复杂网络的所有参数当做奖池,上述一组子参数对应的子网络就是中奖彩票。

如何找到中奖彩票

如果彩票假设是正确的,那么下一个问题就是如何设计一种策略来找出中奖彩票。作者提出一种通过迭代找到中奖彩票的方法:

1.随机初始化一个复杂神经网络

2.训练这个网络j次直到收敛

3.剪掉部分权重参数

4.将剩下的子网络用第1步的权重进行初始化,创建中奖彩票

5.为了评估第4步得到的子网络是否是中奖彩票,训练子网络,比较准确率

上述过程可以进行一次或者多次,在只有一次剪枝时,网络训练一次,p%的权重被剪掉。论文中迭代进行n次剪枝,每一次剪掉p^(1/n)%的权重。

实验分析

作者分别在针对MNIST的全连接神经网络和针对CIFAR10的卷积神经网络上做了大量实验。这里以MNIST实验为例:

Pm代表网络还剩下多少的参数。从图3可以观察到,不同剪枝率的子网络的性能不一样,当Pm>21.2%时,Pm越小,即剪枝的参数越多,准确率越高,当Pm<21.1%时,Pm越小,准确率会下降。中奖彩票要比原始网络收敛的更快,同时具有更高的准确率和泛化能力。

由图4可以观察到,迭代剪枝要比oneshot剪枝更快找到中奖彩票,而且在子网络规模较小的情况下依然可以达到较高的准确率。为了衡量中奖彩票中初始化的重要性,作者保留了中奖彩票的结构然后使用随机初始化重新训练。与中奖彩票不同的是,重新初始化的网络学习速度比原来的网络越来越慢,并且在进行少量剪枝之后就会失去测试精度。

总结

本文中,作者提出了彩票假设并给出一种寻找中奖彩票的方法,通过迭代非结构化剪枝的方式可以找到一个子网络,用原始网络的初始化参数来初始化,可以在性能不下降的情况下更快的训练这个子网络,但是如果用随机初始化方法却达不到同样的性能。

论文2:DeconstructingLotteryTickets:Zeros,Signs,andtheSupermask

彩票假设回顾

Frankle和Carbin在彩票假设(LT)论文中提出一种模型剪枝方法:对网络训练后,对所有小于某个阈值的权重置0(即剪枝),然后将剩下的权重重置成原始网络初始的权重,最后重新训练网络。基于这种方法,得到了两个有趣的结果。

一方面经过大量剪枝的网络(删掉85%-95%的权重)与原网络相比性能并没有明显的下降,而且,如果仅仅剪掉50%-90%的权重后的网络性能往往还会高于原网络。另一方面,对于训练好的普通网络,如果重新随机初始化权重然后再训练,得到的结果与之前的相当。而对于彩票假设的网络并没有这个特点,只有当网络使用和原网络一样的初始化权重,才能很好地训练,如果重新初始化会导致结果变差。剪枝掩模(如果删掉权重置0,否则为1)和权重的特定组合构成了中奖彩票。

存在的问题

虽然上篇论文里证明了彩票假设是有效的,然而许多潜在的机制尚未得到很好的理解。例如:LT网络如何使他们表现出更好的性能?为什么掩模和初始权重集如此紧密地耦合在一起,以至于重新初始化网络会降低它的可训练性?为什么简单地选择大的权重构成了选择掩模的有效标准?其他选择掩模的标准也会起作用吗?本篇论文提出了对这些机制的解释,揭示了这些子网络的特殊模式,引入了与彩票算法相抗衡的变体,并获得了意外发现的衍生品:超级掩模(supermask)。

掩模准则

作者将每个权重的掩模值设为初始权值和训练后的权值的函数M(w_i,w_f),可以将这个函数可视化为二维空间中的一组决策边界,如图1所示。不同的掩码标准可以认为是将二维(wi=初始权值,wf=最终权值)空间分割成掩码值为1vs0的区域。

如图所示的掩码准则由两条水平线标识,这两条水平线将整个区域划分为掩码=1(蓝色)区域和掩码=0(灰色)区域,对应于上篇论文中使用的掩模准则:保留最终较大的权重,并剪掉接近于零的权重。作者将这种称为large_finalmask,M(w_i,w_f)=|w_f|。作者还提出了另外8种掩模准则,对应的公式都在下图表示出来了,保留椭圆中彩色部分的权重,将灰色部分的权重剪掉。

作者对这些掩模准则做了一系列对比试验,对于全连接和Conv4网络结果如下图所示。可以发现,magnitudeincrease和large_final相比不相上下,在Conv4网络中还要表现的更好一些。

以随机掩模为基线,我们可以发现那些倾向于保留具有较大最终值的权重的准则能更好的发现子网络,而保留小权值的效果较差。

正负号的重要性

现在已经探索了对哪些权重进行减值效果较好。接下来的问题是该将保留下来的权重重置为何值。作者主要是想研究上篇论文中的一个有趣的结果,当重置为原网络初值的时候效果很好,但当随机初始化时,效果会变差。为什么重新初始化效果会变差以及初始化的哪些条件最重要?为了找到问题的答案,作者做了一系列初始化的实验。

Reinit:基于原始的初始化分布来初始化保留的权重

Reshuffle:基于保留权重的原始分布进行初始化

Constant:将保留的权重设为正或负的常数,即每层原初始值的标准差

可以发现保留权重的初始值并没有保留正负号这么重要。如果使用其他的初始化方法,但是忽略正负号,那么效果很差,和随机初始化差不多(图中虚线)。而如果和原来的权重保持一样的正负号,三种方法和LT网络的初始化效果相差无几(图中实线)。只要保持正负号一致,即使将剩下的权重都设为常量也不会影响网络的表现。

超级掩模

在开头提到了超级掩模的概念,它是一种二值掩模,当作用到随机初始化的网络上时,即使不重新训练,也可以得到更高的准确率。下面介绍如何找到最佳的超级掩模。

基于上述对初始符号重要性的洞察以及让权重更接近最终值的考虑,作者引入了一种新的掩模准则,选择较大的权重,而且这些权重在训练后也保持相同的正负号,作者将其称为large_final,samesign。并且用large_final,diffsign作为对照,两者的区别如下图所示。

参考文献

1.Cheng,Yu,etal."Asurveyofmodelcompressionandaccelerationfordeepneuralnetworks."arXivpreprintarXiv:1710.09282(2017).

2.Frankle,Jonathan,andMichaelCarbin."Thelotterytickethypothesis:Findingsparse,trainableneuralnetworks."ICLR(2019).

3.Zhou,Hattie,etal."Deconstructinglotterytickets:Zeros,signs,andthesupermask."arXivpreprintarXiv:1905.01067(2019).

THE END
1.一种彩票的中奖率是1%,买100张这种彩票,就()中奖.刷刷题APP(shuashuati.com)是专业的大学生刷题搜题拍题答疑工具,刷刷题提供一种彩票的中奖率是1%,买100张这种彩票,就()中奖.A.一定B.一定不会C.不可能D.有可能的答案解析,刷刷题为用户提供专业的考试题库练习。一分钟将考试题Word文档/Excel文档/PDF文档转化为在线题https://www.shuashuati.com/ti/be1307dfae5347ef9a522eeeb2f9150f.html
2.一种彩票的中奖率是1%.买100张这种彩票.中奖情况如何?()A.一定B解答:解:一种彩票的中奖率是1%,买100张这种彩票,买这种可能中奖; 故选:C. 点评:1%是中奖的概率,只是说明有中奖的可能,无论买多少彩票都只是有可能中奖,不要被数字迷惑. 练习册系列答案 综合复习与测试系列答案 课时总动员系列答案 期末赢家系列答案 http://www.1010jiajiao.com/xxsx/shiti_id_a34aa2efdc3737ce113301b06001f310
3.一种彩票的中奖率是1%,那么买100张彩票是否会中奖?()A. 可能会中奖B. 一定会中奖C. 一定不会中奖 扫码下载作业帮搜索答疑一搜即得 答案解析 查看更多优质解析 解答一 举报 由分析知:一种彩票的中奖率是1%,那么买100张彩票,可能会中奖;故选:A. 一种彩票的中奖率是1%,属于不确定事件,可能中奖,也可能不中奖,进而得出结论. 本题考点:事件发生的可能性大小语言描述https://qb.zuoyebang.com/xfe-question/question/7f942d72d6ced601064dc3eecd69cbc7.html
4.1%中奖率的彩票买100张一定能中奖吗?(一) 1%中奖率的彩票买100张一定能中奖吗?答案是:不一定。 这个比较容易理解,毕竟彩票的总量不是100张,而是若干个100张。如果还是觉得有点模糊,那么换个例子就更清楚些,50%中奖率的彩票买两张一定能中奖吗?50%中奖率,抽其中两张,没有中奖的概率为50%*50%=25%,中奖的概率为75%。而抽奖次数逐渐增大,结果会https://www.jianshu.com/p/390b59b5920c
5.刮刮乐怎么买才能中奖率高刮刮乐中奖技巧和口诀→MAIGOO知识2、尽量不要玩一本里出过大奖的票。 例如一本票里面出了500以上的奖金,不是尽量,是一定不能玩,刮一张可能你会回本钱,连续刮只会越亏越多。 刮刮乐怎么买才能中奖率高 1、可以购买4-6种,然后分散刮,隔几张刮一张,这是考虑到刮刮乐彩票的设奖分散的特点。 这样跳着刮,能增加中奖的几率。刮刮乐有很多票种https://www.maigoo.com/goomai/297672.html
6.生活中的概率(精选十篇)我们在日常生活中的好多事情都多多少少牵扯到了统计或者概率计算的问题,例如人口普查,粮食生产状况的研究,交通状况的研究,体育项目成绩的研究;天气预报中的降水概率,买彩票的中奖概率,患有某种遗传病的概率等。生活中的概率问题往往让我们意想不到,学会怎样运用概率,可以让我们简单的解决生活中遇到的一些问题,有时候还https://www.360wenmi.com/f/cnkeybrueeyc.html
7.买彩票的中奖率有多高?理财频道如果非要把买彩票比作投资/投机,刚刚点出了两个问题:一买彩票中奖概率低,二买彩票中奖后到手金额不一定高。 保险是保障,硬要从投资的角度来说的话,我不是精算师算不出理赔率,但很明显的是,现在疾病发生率有年轻化的趋势,就个人而言,罹患重疾或慢性病的概率比中大奖高。 https://money.hexun.com/2019-01-17/195916242.html
8.一辈子买彩票,中大奖的几率有多大?大数据告诉你双色球以80岁为一个限定标准,因为购买彩票必须是要年满18周岁,也就是说只剩下了62年时间来买彩票,这期间就需要花费20000块钱。20000万块钱对于一个普通人来说是可以接受的,毕竟时间已经有长达几十年,而且这20000块钱还是一个最基本的购买标准。 其次,了解到了时间以及花费的费用之后 ,就要算到中奖的概率,一等奖的中https://www.163.com/dy/article/GQETIMOE0552O6T4.html
9.6种人容易中彩票什么面相的人买彩票中奖率高6种人容易中彩票,是哪6种人呢?我们经常能在新闻中,看见有人中彩票之后一夜爆发,但是这种几率降临在自己的身上少之又少,那么什么面相的人买彩票中奖率高呢? 问:6种人容易中彩票 答: 1、高频彩民:经常购买彩票的人,特别是每次都按照相同的数字购彩的人,中奖的概率会相应地增加。 https://www.qiyuange.com/xiangshu/95697.html
10.贝叶斯居然用事件概率,就轻松预知了未来?普莱斯在向皇家学会提交贝叶斯的研究结果时,他能够确定,如果你买了一张彩票并中奖了,那么至少有一半的彩票都能中奖的概率是75%。 但是,考虑概率的概率问题会让人有点儿头晕。更重要的是,如果有人在催促我们:“好吧,但是你认为彩票的中奖率到底是多少?”我们仍然不知道该说什么。 https://blog.csdn.net/BULpreZHt1ImlN4N/article/details/81463791
11.彩票神人!魔术师成功预测奖号男子半年中2万注奖据悉,本次开出的彩票大奖奖金高达240万英镑,如果布朗提前买了自己预测出的这注彩票号码,那么他将轻而易举地赢得240万英镑的奖金。然而据内幕人士称,当布朗本周宣布自己将表演这一惊人的“预测彩票大奖”魔术后,他就被禁止购买任何一张彩票。 麻省理工统计天才破解即开彩 中奖率达90% http://www.hkwb.net/nrpd/content/2012-01/28/content_606097_0.htm