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

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

现在有了性能更强的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.速学!受欢迎的十大彩票推荐笔架山老爷 发布于:重庆市 2024.11.30 21:44 分享到 受欢迎的十大彩票推荐+Aゞ:179 89 25】網: R Y 9 9 . C C 】【域名手动在浏览器打开】【独家团队】【精准计划】【万人聊天室】不论昨日如何,今天都是全新的。勇敢地迈出第一步,因为你的梦想正等着你 推荐视频 已经到底了 热门视频 已经到底了 https://www.sohu.com/a/831997289_122137570
2.小幸运变大惊喜小伙中500万与家人分享有一天,他因为工作特别忙没买,结果当天恰好开的正是那组自选号,就这样他与1000元奖金失之交臂,至今想来仍觉可惜。自那以后,他干脆放弃了自选守号,全部机选。为避免再次错失良机,他还特意要了站主的联系方式,工作忙不过来的时候就请销售员帮忙代打,中奖彩票就是销售员帮他打的。https://www.cwl.gov.cn/c/2021-04-16/483132.shtml
3.6种人容易中彩票什么面相的人买彩票中奖率高6种人容易中彩票,是哪6种人呢?我们经常能在新闻中,看见有人中彩票之后一夜爆发,但是这种几率降临在自己的身上少之又少,那么什么面相的人买彩票中奖率高呢? 问:6种人容易中彩票 答: 1、高频彩民:经常购买彩票的人,特别是每次都按照相同的数字购彩的人,中奖的概率会相应地增加。 https://www.qiyuange.com/xiangshu/95697.html
4.买哪种刮刮乐容易中奖?几元刮刮乐容易中奖其实刮刮乐这中即开彩票多付出就多回报,不过其实哪种都一样的,但是即开型彩票确实有个小窍门,买的时候从一本新的没开封、第二十到第二十九张,如果中了大一些的奖,那么这一叠就不要再买了,再换一叠新的。 买哪种刮刮乐容易中奖 找一个人气比较旺的彩票店,里面一定要有很多人才行,这点很重要,进去之后,如果https://www.vipc.cn/article/58ef4de21d38ec001a9186a0
5.十二种中奖方法十二个最简单的买彩票技巧分享·方法一:购买彩票十二种中奖方法。 十二个最简单的买彩票技巧分享。 ·方法一:购买彩票不在乎多,要坚持以少量的投入获得最大的回报。选号不必刻意苛求,彩票开奖号码的趋势对任何人来说都不可能是明晰的。你非神仙又怎可能获得十分明了的选号?感觉感性一点马虎一点吧,从大气而模糊中不经意的闯出一些灵感来,如大胆选择断档选择多数https://www.360doc.cn/article/29215996_1100894819.html
6.你的大脑在欺骗你……运营:太好啦!不过还是有很多人前仆后继地会去买,因为他们感觉自己可能会中。 在他们「心目中」中奖的概率远远高于「实际」能中奖的概率了,这是一种极度偏离现实的认知偏差。 为什么彩票是选7个数字就行了? 这7个数字选择性很少,而且都是比较熟悉的小数字,人们感觉这应该有机会中吧。 https://36kr.com/p/1721616760833
7.快乐八哪种玩法最容易中奖在快乐8游戏中,哪种玩法最容易中奖,主要取决于中奖概率和中奖金额的综合考量。以下是对快乐8几种主要玩法的中奖概率和中奖金额的简要分析: 一、中奖概率分析 “选一”玩法:中奖概率为25%,是快乐8中中奖概率最高的玩法。但相应地,中奖金额也较低。 “选四”玩法:最高中奖概率可达25.8%,相对较高,且不容易空奖金https://h5.bbs.17500.cn/forum/221/thread/10861040?page=0