原文:
www.kdnuggets.com/2019/05/lottery-ticket-hypothesis-neural-networks.html
来源:Pixabay
训练机器学习模型是现实世界数据科学解决方案中最具挑战性和计算开销的方面之一。几十年来,人工智能(AI)社区已经开发了数百种技术来改进机器学习模型的训练,基于一个单一的公理假设,即训练应该覆盖整个模型。最近,来自麻省理工学院(MIT)的 AI 研究人员发表了一篇挑战这一假设的论文,提出了一种更聪明、更简单的方法,通过关注模型的子集来训练神经网络。在 AI 社区中,MIT 的论文以引人注目的名字彩票票假设而闻名。
1. Google 网络安全证书 - 快速入门网络安全职业生涯。
2. Google 数据分析专业证书 - 提升你的数据分析技能
3. Google IT 支持专业证书 - 支持你的组织的 IT 工作
训练机器学习模型的过程是数据科学家在理论和现实世界解决方案的约束之间常常面临妥协的领域之一。往往,一个看起来非常适合特定问题的神经网络结构无法完全实现,因为训练成本可能过高。通常,神经网络的初始训练需要大量的数据集和几天的昂贵计算使用。结果是非常大的神经网络结构,神经元和隐藏层之间有大量连接。这种结构通常需要经过优化技术来删除一些连接并调整模型的大小。
多年来困扰人工智能研究人员的一个问题是,我们是否真的需要那些大型神经网络结构。显然,如果我们连接几乎每一个神经元,可能会实现一个能够完成初始任务的模型,但成本可能会很高。我们不能从更小、更简洁的神经网络结构开始吗?这就是彩票票假设的核心。
彩票票假设
使用赌博世界的类比,机器学习模型的训练常常被比作通过购买每一张可能的彩票来赢得彩票。但如果我们知道中奖的彩票是什么样的,我们难道不能更聪明地选择票据吗?
在机器学习模型中,训练过程会产生大型神经网络结构,这些结构相当于一大包彩票。经过初步训练后,模型需要经过优化技术,如剪枝,以去除网络中不必要的权重,从而减少模型的大小而不牺牲性能。这相当于在袋子中寻找中奖票据并丢弃其余部分。很常见的是,剪枝技术最终会产生比原始网络小 90%的神经网络结构。显然的问题是:如果一个网络可以缩小尺寸,为什么我们不直接训练这个更小的结构,以便提高训练效率呢?矛盾的是,实际机器学习解决方案中的经验表明,剪枝所揭示的架构从一开始就更难训练,准确性低于原始网络。所以你可以买一大包彩票并逐步找到中奖号码,但相反的过程却太难了。或者我们曾经这样认为????
麻省理工学院的彩票票据假设的主要思想是,一致地,大型神经网络将包含一个较小的子网络,如果从一开始进行训练,将会实现与较大结构相似的准确度。具体而言,研究论文将假设概述如下:
- ***彩票票据假设:**一个随机初始化的密集神经网络包含一个子网络,该子网络在隔离训练时可以匹配原始网络的测试准确率,经过最多相同数量的迭代训练。
在论文的背景下,小型子网络通常被称为中奖票据。
考虑一个形式为 f(t, a, p) 的神经网络,其中 t=训练时间,a=准确率和 p=参数。现在考虑 s 为从原始结构中生成的所有可训练神经网络的子集,经过剪枝过程。彩票票据假设告诉我们,存在 a f’(t’, a’, p’) ∈ s,使得 t’ <= t, a’ >= a 和 p’ <= p。简单来说,传统的剪枝技术揭示的神经网络结构比原始结构更小且更容易训练。
如果彩票票据假设成立,那么下一个显而易见的问题是找到识别中奖票据的策略。这个过程涉及到一个智能训练和剪枝的迭代过程,可以总结为以下五个步骤:
-
随机初始化一个神经网络。
-
训练网络直到其收敛。
-
剪枝网络的一部分。
-
为了提取“中奖票”,请将网络剩余部分的权重重置为(1)中的值——即训练开始之前它们接收到的初始化值。
-
为了评估步骤(4)中得到的网络是否确实是一个“中奖票”,请训练剪枝后的未训练网络,并检查其收敛行为和准确性。
这个过程可以运行一次或多次。在一次剪枝方法中,网络训练一次,剪枝 p% 的权重,并重置剩余的权重。尽管一次剪枝肯定有效,但彩票票假设论文显示,当该过程在 n 轮中迭代应用时,效果最佳;每轮剪枝 p1/n%的权重,而这些权重在前一轮中幸存下来。然而,一次剪枝通常产生非常稳健的结果,而无需计算上昂贵的训练。
麻省理工学院团队在一组神经网络架构中测试了彩票票假设,结果显示剪枝方法不仅能够优化架构,还能找到“中奖票”。查看下图:
在这些结果中观察两个主要方面。“中奖票”在没有宽网络的剩余冗余的情况下,比宽网络训练得更快。实际上,它们越瘦,训练得越快(在合理范围内)。然而,如果你随机重新初始化网络的权重(控制),结果网络的训练速度现在比完整网络更慢。因此,剪枝不仅仅是找到正确的架构,还包括识别“中奖票”,这是网络中特别幸运初始化的子组件。
根据实验结果,麻省理工学院团队扩展了他们最初的假设,提出了所谓的彩票系统猜想,表达了以下内容:
- **彩票票猜想:**回到我们的动机问题,我们将假设扩展到一个未经测试的猜想,即 SGD 寻找并训练一个子集的良好初始化权重。密集的、随机初始化的网络比剪枝后的稀疏网络更容易训练,因为可能的子网络更多,从中训练可能恢复一个“中奖票”。
这个猜想在概念上似乎是有意义的。剪枝子网络的池子越大,找到“中奖票”的机会就越大。
彩票票假设可能成为近年来最重要的机器学习研究论文之一,因为它挑战了神经网络训练中的传统智慧。尽管剪枝通常通过训练原始网络、移除连接并进一步微调来进行,但彩票票假设告诉我们,最佳的神经网络结构可以从一开始就学习到。
原文。经许可转载。
简介:耶稣·罗德里格斯是 Invector Labs 的管理合伙人、加密货币和天使投资人、演讲者、作家、CIO.com 专栏作家以及人工智能爱好者。
资源:
相关内容: