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

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

现在有了性能更强的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.双色球买什么中奖率高教你几招双色球买什么中奖率高 教你几招 第一招:结合前人之鉴 该方法乃世界最为经典的双色球制胜法则。先寻找一群彩民,在预测的范围内集中购买、统一分配,中奖几率显著提高,尽管说不能稳重一等奖,但是小奖之类的还是频频出现的,非常给力。 第二招:量力而行https://www.17500.cn/arts/sdetail-11159241.html
2.如何提高彩票中奖几率:13步骤即开型彩票有不同的赔率、样式和图案,但是价格是区分它们的最简单方式。一般根据类型不同和地区不同,一张即开型彩票的价格在5到100元不等。比较便宜的彩票中奖率低,收益少,不同中奖层次间的差价也小。贵一些的彩票底价至少为25元,对应的中奖率高,高回报的分布也较为均匀、中头奖的几率也更高。 换句话说,http://zh.m.wikihow.com/%E6%8F%90%E9%AB%98%E5%BD%A9%E7%A5%A8%E4%B8%AD%E5%A5%96%E5%87%A0%E7%8E%87?amp=1
3.农民朋友怎样买彩票提高中奖率?看这种方法真的很灵验!首先学习宇宙农民朋友怎样买彩票提高中奖率?看这种方法真的很灵验! 首先学习宇宙吸引力法则,也可以不学,按我说的方法做一样。 朗达拜恩的《秘密》4部曲,主要介绍了宇宙吸引力法则,中心思想是——相信什么就来什么;你的愿望就是宇宙的命令……还有3本,分别叫《力量》、《魔力》、《英雄》。 https://www.360doc.cn/article/29215996_1052504187.html
4.刮刮乐怎么买才能中奖率高刮刮乐中奖技巧和口诀→MAIGOO知识刮刮乐怎么买才能中奖率高 1、可以购买4-6种,然后分散刮,隔几张刮一张,这是考虑到刮刮乐彩票的设奖分散的特点。 这样跳着刮,能增加中奖的几率。刮刮乐有很多票种,面值有2元、5元、10元、20元,针对这些票种,我们可以组合出不同的刮奖方法来。 2、忽视与自己有关的幸运数字。吉祥数字、生日、电话号码、门牌号https://www.maigoo.com/goomai/297672.html
5.午间道如何拿下美国强力球15亿美元彩票大奖?今日,美国“强力球”(Powerball)将开出全球彩票史上最大奖:15亿美元!你想成为那个幸运儿吗?身在中国,怎么在美国赢走大奖?机选还是人工选号,网上能不能买、中奖率有多少、巨奖中奖者一定会被诅咒吗…… 关于美国15亿彩票大奖需要知道的10件事。 文|阿白 https://static.nfapp.southcn.com/content/201601/14/c35548.html
6.你对于彩票有一定了解,并且希望在购买彩票时有更高的中奖概率在当今的彩票市场中,高赔率彩票平台是吸引无数彩民的热门选择。如果你对于彩票有一定了解,并且希望在购买彩票时有更高的中奖概率和获得巨额奖金的机会,那么高赔率彩票平台肯定是你不容错过的选择。加入[hg582058.com]足球比赛软件在现代体育产业中扮演着重要的角色。随着足球比赛的日益普及和商业化,越来越多的球迷、俱https://wap.xxsb.com/content/2024-02/19/content_252221.html
7.疆内彩闻——新疆福利彩票网三是返奖高、奖金高。意思是该游戏总返奖率为58%,其中“选十”玩法单注奖金最高达500万元,“选九”玩法固定奖最高奖金30万元。 四是全国联销、天天开奖。 3、“快乐8”游戏是高频游戏吗? 不是,是基诺型彩票游戏。 4、“快乐8”游戏新疆区上市销售时间? http://www.xjflcp.com/article/2020/12/29/c_70475.htm
8.正版大全版本迷失第一季第9集正在播放哪个彩票中奖率高而且奖金多 七乐艺术学校 712直播下载 九球直播体育平台 福彩3d预测最准专家949499 备用情人小说txt笔趣阁免费阅读 大赢家原版 乐博乐博创始人身价 ns9000 金龙彩82950k,CCm最新版本更新内容 日本的经典电影 欢乐斗地主 比赛 1000亿彩票官方正版网站 综合体育活动 2002欧洲冠军杯http://wap.zldoamg.top/project/12021839237494.html
9.日本发售巨奖彩票头奖1亿日元中奖几率高(图)一等奖(10万日元)有4万注,二等奖(1万日元)有8万注。售完即止。 上述彩票每张300日元,销售日期截至6月1日。6月12日在东京摇号。在东京银座的热门彩票点,早上8点半起就有约350人排起长队等待发售。茨城县石市的一名公司职员说:“中1亿日元大奖的几率比较高,我就想这次中个大奖买房子。”https://world.huanqiu.com/article/9CaKrnJvpqu