深度学习中梯度消失、爆炸的原因以及解决方案

前言:
从神经网络的发展过程中,有一个问题困扰了很多人,也是神经网络发展道路上的一大绊脚石。它就是大家并不陌生的梯度消失、爆炸。本文主要深入介绍梯度消失、爆炸的产生原因和目前的一些解决方案,让大家对于梯度消失、爆炸的现象能够有更深的理解。其中有些知识相对基础,大家可以根据需求进行跳跃阅读。

  1. 反向传播算法
    说到梯度消失和爆炸,就不得不说到反向传播算法。我们都知道,现在的深度学习模型都是通过基于对于loss进行梯度下降法来更新参数达到学习的目的。由于深度学习网络一般有很多层,直接用loss对每个学习参数进行求导在编程上难以实现,这也是反向传播算法被提出的原因。反向传播算法实则是利用了链式法则,从后往前计算loss对于每一层每一个学习参数的梯度。由于在神经网络中的层结构可以分解成简单的加、减、乘、max等简单易求梯度的模块,因此用反向传播算法来求解梯度易于实现且利用到了神经网络的结构,计算复杂度也不高。但是,通过链式法则我们不难发现,在更新参数的过程中,梯度很容易会随着层数的增加,从后往前快速减小或增大。这个现象就是梯度消失、爆炸。
  2. 梯度消失、爆炸
    之前介绍了梯度消失、爆炸现象的来源,但是很多人可能对于为什么梯度可能会随着层数的增加而快速减小或增大。
    首先,假设我们的模型的某个输出为H(x) = wj * f(wi * x + b),其中f是激活函数。我们想要求H(x)对x的偏导,我们可以得知是wj * f(wi * x + b)对于x的偏导数。所以我们可以看出,影响输出H(x)对于输入的梯度大小因素有两个,一个是参数wj的大小,一个是激活函数的输出大小。这样我们就很容易理解梯度消失和爆炸的问题了。我们知道,在深度学习的发展开端,激活函数是由sigmoid函数担当的。如figure 1所示,sigmoid函数在自变量值绝对值较大时,梯度已经接近0了。也就是说,当我们的上一层输出已经进入激活函数饱和区的情况下,梯度会非常小。因此,在反向传播中,随着层数的累积,很可能出现大量饱和的神经元,并且梯度指数级减小向零逼近。而这样的现象导致了我们训练的困难(因为参数基本得不到更新)。而梯度爆炸则相反,即某些参数的梯度值过大,导致在反向传播过程中逐级增大,最终爆炸。我们在来看几张图,更直观的说明深度网络中的梯度问题【1】(figure 2, figure 3)。其中横坐标代表训练的epoch数。从图中可以看出,深度网络中,处于后面的层的参数更新速度要快于位于前面的层,说明梯度从后面层到前面层出现了减小的现象。


    figure 1(来自维基百科)

    figure 2

    figure 3
  3. 梯度消失和爆炸的部分解决方案
    在这边我主要介绍分层预训练后微调、梯度限制和正则、激活函数的变更、BatchNorm以及残差结构这5种方法。
    3.1 分层预训练后微调
    这个方法来自Hinton在2006年发表的一篇论文,Hinton为了解决梯度的问题,提出采取无监督逐层训练方法。他的基本思想是每次训练一层隐节点,训练时将上一层隐节点的输出作为输入,而本层隐节点的输出作为下一层隐节点的输入,此过程就是逐层“预训练”(pre-training);在预训练完成后,再对整个网络进行“微调”(fine-tunning)。Hinton在训练深度信念网络(Deep Belief Networks中,使用了这个方法,在各层预训练完成后,再利用BP算法对整个网络进行训练。此思想相当于是先寻找局部最优,然后整合起来寻找全局最优,此方法有一定的好处,但是目前应用的不是很多了。
    3.2 梯度限制和正则
    梯度限制主要是针对梯度爆炸提出的。当梯度的值超过我们所设的阈值,则按阈值输出梯度。
    而权值的正则一般指的是L1和L2正则。正则的思想主要是限制权值过大造成对于某些输入过分迁就的过拟合现象。正则起到使得权值分布较为均匀且值都比较小。在梯度消失、爆炸的问题这里,当我们的权值大小得到限制,一则不容易出现于激活函数的饱和区的梯度消失,二来不会因为权重过大导致某些参数梯度过大的爆炸。因此,正则化也有助于梯度问题。
    3.3 激活函数的变更
    通过上面我们可以知道,因为sigmoid等早期的激活函数有饱和区的问题导致梯度消失。所以,通过更换其他激活函数可以避免这个问题。
    如relu激活函数,当输出大于0时,其梯度总是1。这也就解决了因为激活函数导致的梯度消失的问题。但是relu也有它的缺点:输出不是以0为中心的以及由于负数部分恒为0,会导致一些神经元无法激活(可通过设置小学习率部分解决)。
    3.4 BatchNorm
    Batchnorm是深度学习发展以来提出的最重要的成果之一了,目前已经被广泛的应用到了各大网络中,具有加速网络收敛速度,提升训练稳定性的效果,Batchnorm本质上是解决反向传播过程中的梯度问题。batchnorm全名是batch normalization,简称BN,即批规范化,通过规范化操作将输出信号x规范化保证网络的稳定性。
    通过我们之前第2点的介绍钟,我们知道在梯度反向传播过程中会有权重项。batchnorm就是通过对每一层的输出进行normalization,使得其输出分布更加均匀,消除了w带来的放大缩小的影响,进而解决梯度消失和爆炸的问题。
    3.5 残差结构
    虽然残差结构的提出主要是为了解决网络在深度加深时的退化问题,但是从figure 4可以看出,由于shortcut的存在,残差网络的输出对于输入求编导时,不管输出对于F(x)是多少总有一个常数梯度1的存在,所以一定程度上解决了反向传播中梯度消失的问题。


    figure 4

参考资料:
【1】《Neural networks and deep learning》
【2】Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun,Deep Residual Learning for Image Recognition

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,402评论 6 499
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,377评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,483评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,165评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,176评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,146评论 1 297
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,032评论 3 417
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,896评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,311评论 1 310
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,536评论 2 332
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,696评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,413评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,008评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,659评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,815评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,698评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,592评论 2 353