残差神经网络

残差神经网络Residual Neural Network,简称ResNet[1]属于深度学习模型的一种,其核心在于让网络的每一层不直接学习预期输出,而是学习与输入之间的残差关系。这种网络通过添加“跳跃连接”,即跳过某些网络层的连接来实现身份映射,再与网络层的输出相加合并。其运作机制与高速神经网络英语Highway network类似,通过极大的正偏置权重来打开“门控”。[2] 这一设计使得拥有几十上百层的深度学习模型可以更易于训练,增加模型深度时还能保持甚至提高准确度。所谓的“残差连接”即“直连跳过”,这一概念也被应用于1997年的长短期记忆模型LSTM、[3] Transformer模型(比如BERTGPT系列,ChatGPT等)、AlphaGo ZeroAlphaStar英语AlphaStar (software)以及AlphaFold等。

在残差神经网络中的一个残差块里,残差连接跳过了两个网络层。

残差神经网络由何恺明、张祥雨、任少卿和孙剑开发,这一成果在2015年的ImageNet大规模视觉识别挑战赛中夺冠。[4][5]

基本原理

背景介绍

2012年,针对ImageNet竞赛开发的AlexNet模型是一个包含8层的卷积神经网络。到了2014年,牛津大学的视觉几何组(VGG)通过叠加3x3卷积层将网络深度增加到了19层。[6] 但是,层级的增加却导致训练精度的迅速下降,[7] 这种现象被称为“性能退化”问题。[1]

理论上,如果一个更深的网络仅仅是通过在一个较浅网络的基础上增加额外层来构建的,那么这个更深的网络不应该比其较浅的网络有更高的训练损失。[1] 如果这些额外层具有身份映射的能力,那么更深的网络应该能够实现与其较浅网络相同的功能。但这里存在一个假设,即优化器不能有效地将这些参数化的网络层调整为身份映射。

残差学习

在多层神经网络模型里,设想一个包含若干层的子网络。这个子网络的函数用 来表示,其中 是子网络的输入。残差学习是通过重新设定这个子网络的参数,让参数层表达一个残差函数 。因此,这个子网络的输出 可以表示为:

 

这一原理同样适用于1997年提出的长短期记忆LSTM单元,[3]随时间反向传播英语Backpropagation through time里计算 ,简化为 

函数 常通过矩阵乘法实现,并结合激活函数以及规范化操作(如批量规范化英语Batch normalization或层规范化)。

这类子网络被称作“残差块”。[1] 通过叠加这样的残差块,形成深度残差网络。

在" "公式中的" "操作是通过一个相当于恒等映射的跳跃连接来完成,它将残差块的输入直接与输出连接。在随后的研究中,这种连接常被称作“残差连接”。[8]

信号传递

身份映射的引入有利于信号在前向传播路径和反向传播路径中的传递。[9]

向前传播

如果第 个残差块的输出是第 个残差块的输入(这里假设块与块之间没有激活函数),可以得到:[9]

 

若递归应用此公式,例如, ,可以推导出:

 

这里 表示任意后续残差块的索引(比如处于最末尾的块), 代表任意靠前的块对应的索引。该公式说明了总有一个信号能够直接从浅层块 传递到深层块 

反向传播

残差学习的公式还在一定程度上缓解了梯度消失问题。然而,梯度消失并不是导致性能退化问题的根源,因为通过引入规范化层(如批量规范化)可在一定程度上解决此问题。根据上面的前向传播过程,对 进行求导,可以得到:[9]

 

这里  是最小化损失函数。以上表明,浅层的梯度计算 总会直接加上一个项 。因此,由于额外项 的存在,即使 的梯度很小,总梯度 也不会消失。

残差块

 
两种类型的卷积残差块。左侧是基本块,它由两个3x3卷积层组成。右侧是瓶颈块,该块先通过一个1x1卷积层进行降维,接着是一个3x3卷积层,最后再通过一个1x1卷积层恢复原来的维度。

基本残差块

基本残差块是原始ResNet研究中最简单的部分。[1] 它包括两个串行的3x3卷积层以及一个残差连接。这两层的输入输出尺寸保持一致。

瓶颈残差块

瓶颈残差块包含三个串联的卷积层和一个残差连接。[1] 该块的第一层是1x1卷积,用于降维,比如降至输入维度的1/4;第二层是3x3卷积;最后一层是另一个1x1卷积,用于恢复维度。ResNet-50、ResNet-101和ResNet-152模型都基于瓶颈块构建。[1]

预激活残差块

预激活残差块[9]在应用残差函数 之前,先使用激活函数,如非线性和规范化的处理。预激活残差块的计算可以表述为:

 

这里的 可以是如线性整流函数等任意非线性激活或归一化操作。这种设计减少了残差块间非恒等映射的数量,被用于训练200层到1000多层的模型。[9]

GPT-2开始,Transformer块常被用于预激活块,这在Transformer模型的相关文献中被称为“预规范化”。[10]

Transformer块

 
原始GPT模型采用的Transformer架构是由两种类型的残差块构成:一个是多头注意力块,另一个是前馈的多层感知器(MLP)块。这种设计通过结合两个功能强大的残差块,使得Transformer能够高效地处理数据并学习复杂的特征,其中每个残差块都利用残差连接来促使信号在网络深层之间的流动以及更有效的进行梯度回传,克服了深度模型训练过程中遇到的梯度消失等问题。

Transformer块是由两个残差块组成,每个残差块都设有一个残差连接。

第一个残差块为多头注意力块,使用了自注意力运算,随后连接一个线性映射层。第二个残差块是一个前馈式的多层感知器(MLP)块,这个块在某种程度上像是一个“反向”的瓶颈块,它通过一个线性映射层(在卷积神经网络中相当于1x1的卷积)来扩大维度,然后通过另一个线性映射层来减少维度。

一个Transformer块包含了四层线性映射。GPT-3模型拥有96个这样的Transformer块(在Transformer领域的文献中,通常将一个Transformer块称作一个“Transformer层”)。因此,该模型包含了大约400层的映射层,包括Transformer块内的96x4层,以及一些额外的层用于输入嵌入和输出预测。

若没有残差连接,训练网络深度极高的Transformer模型将无法获取成功。[11]

相关研究

1961年,在弗兰克·罗森布拉特出版的书籍中介绍了一个含有跳跃连接的三层多层感知器(MLP)模型(详见第15章,第313页[12])。这种模型被称作“交叉耦合系统”,其跳跃连接实际上是交叉耦合连接的一种形式。[12]

在1994年[13]和1996年[14]出版的书籍中提出了在前馈MLP模型中使用的“跳层”连接:“MLP一般允许存在多个隐藏层,并且支持从输入直接到输出的‘跳层’连接”(详见第261页[13],第144页[14]),“...这使得非线性单元能够调整线性函数形式”(详见第262页[13])。这种说法其实已经暗示了非线性MLP的表现就像是在一个线性函数上加上了一个残差函数。

塞普·霍赫赖特英语Sepp Hochreiter在1991年分析了梯度消失问题,并认为这是深度学习效果不佳的原因。[15] 为解决这一问题,长短期记忆(LSTM)网络[3]在每个LSTM单元中引入了权重为1.0的跳跃或残差连接,计算公式为 。在随时间反向传播英语Backpropagation through time中,这个公式转化为前馈神经网络中的残差公式 ,从而实现了深层循环神经网络的有效训练。2000年[16]发布的一种LSTM版本通过“遗忘门”来调节连接,这些门的权重不再固定为1.0,而是可以学习的。在实际实验中,遗忘门通过正偏置权重进行初始化,从而解决梯度消失问题。[16]

2015年,高速神经网络英语Highway network[2][17]将上述原理应用于前馈神经网络,被媒体报道称为“首个具有数百层深度的前馈神经网络”。[18] 原有的高速神经网络论文[2]不仅提出了非常深的前馈网络的基本架构,还展示了20层、50层和100层网络的实验结果,并提及正在进行的深达900层的实验。50层或100层的网络相较于它们常用的神经网络有更低的训练误差,但与20层的训练结果相比,误差并无降低(详见MNIST数据集中的图1[2])。在超过19层的网络上,并未有训练精度的提高[2]。然而,ResNet的研究证明了多于20层以上的训练结果同样有效。[9] 它指出跳跃连接中的调节可能导致前向和反向传播中信号的消失(详见第3节[9])。这也解释了为什么2000年的LSTM[16]的遗忘门要通过正偏置权重初始化为开启状态:只要门处于开启状态,它就如同1997年的LSTM。同样,高速神经网络如果通过强正偏置权重保持门的开启状态,就表现得如同ResNet一样。现在的神经网络(如Transformer)中使用的跳跃连接主要就是身份映射。

2016年出现的DenseNets[19]被设计为一种深层的神经网络,旨在将每一层与其他所有层连接。DenseNets通过使用身份映射作为跳跃连接来实现这一目标。与ResNet不同,DenseNets通过拼接而非加法将层输出与跳跃连接合并。

利用残差网络架构,实现了具有随机深度的神经网络。[20] 通过随机丢弃一部分网络层,让信号通过跳跃连接进行传播。这种做法也被称为“路径丢弃”。这是训练大型深层模型,如视觉Transformer英语Vision transformer(ViT)的一种十分有效的正则化方法。

与生物学的联络

虽然最初的残差网络研究并未受生物学启发,但后来的研究却发现残差网络与生物学有关。[21][22]

2023年《科学》杂志上发表的一项研究展示了果蝇幼虫大脑的完整神经连接组[23] 这项研究发现了类似于人工神经网络中如ResNet一样的跳跃连接。

参考文献

  1. ^ 1.0 1.1 1.2 1.3 1.4 1.5 1.6 He, Kaiming; Zhang, Xiangyu; Ren, Shaoqing; Sun, Jian. Deep Residual Learning for Image Recognition. 10 Dec 2015. arXiv:1512.03385 . 
  2. ^ 2.0 2.1 2.2 2.3 2.4 Srivastava, Rupesh Kumar; Greff, Klaus; Schmidhuber, Jürgen. Highway Networks. 3 May 2015. arXiv:1505.00387  [cs.LG]. 
  3. ^ 3.0 3.1 3.2 Sepp Hochreiter; Jürgen Schmidhuber. Long short-term memory. Neural Computation. 1997, 9 (8): 1735–1780 [2024-01-27]. PMID 9377276. S2CID 1915014. doi:10.1162/neco.1997.9.8.1735. (原始内容存档于2021-01-22). 
  4. ^ Deng, Jia; Dong, Wei; Socher, Richard; Li, Li-Jia; Li, Kai; Fei-Fei, Li. ImageNet: A large-scale hierarchical image database. CVPR. 2009 [2024-01-27]. (原始内容存档于2019-09-29). 
  5. ^ ILSVRC2015 Results. image-net.org. [2024-01-27]. (原始内容存档于2023-09-29). 
  6. ^ Simonyan, Karen; Zisserman, Andrew. Very Deep Convolutional Networks for Large-Scale Image Recognition. 2014. arXiv:1409.1556  [cs.CV]. 
  7. ^ He, Kaiming; Zhang, Xiangyu; Ren, Shaoqing; Sun, Jian. Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification. 2016. arXiv:1502.01852  [cs.CV]. 
  8. ^ Szegedy, Christian; Ioffe, Sergey; Vanhoucke, Vincent; Alemi, Alex. Inception-v4, Inception-ResNet and the impact of residual connections on learning. 2016. arXiv:1602.07261  [cs.CV]. 
  9. ^ 9.0 9.1 9.2 9.3 9.4 9.5 9.6 He, Kaiming; Zhang, Xiangyu; Ren, Shaoqing; Sun, Jian. Identity Mappings in Deep Residual Networks. 2015. arXiv:1603.05027  [cs.CV]. 
  10. ^ Radford, Alec; Wu, Jeffrey; Child, Rewon; Luan, David; Amodei, Dario; Sutskever, Ilya. Language models are unsupervised multitask learners (PDF). 14 February 2019 [19 December 2020]. (原始内容存档 (PDF)于6 February 2021). 
  11. ^ Dong, Yihe; Cordonnier, Jean-Baptiste; Loukas, Andreas. Attention is not all you need: pure attention loses rank doubly exponentially with depth. 2021. arXiv:2103.03404  [cs.LG]. 
  12. ^ 12.0 12.1 Rosenblatt, Frank. Principles of neurodynamics. perceptrons and the theory of brain mechanisms (PDF). 1961 [2024-01-27]. (原始内容存档 (PDF)于2023-05-04). 
  13. ^ 13.0 13.1 13.2 Venables, W. N.; Ripley, Brain D. Modern Applied Statistics with S-Plus. Springer. 1994 [2024-01-27]. ISBN 9783540943501. (原始内容存档于2023-08-22). 
  14. ^ 14.0 14.1 Ripley, B. D. Pattern Recognition and Neural Networks. Cambridge University Press. 1996 [2024-01-27]. (原始内容存档于2023-12-02). 
  15. ^ Hochreiter, Sepp. Untersuchungen zu dynamischen neuronalen Netzen (PDF) (diploma论文). Technical University Munich, Institute of Computer Science, advisor: J. Schmidhuber. 1991 [2024-01-27]. (原始内容存档 (PDF)于2023-03-20). 
  16. ^ 16.0 16.1 16.2 Felix A. Gers; Jürgen Schmidhuber; Fred Cummins. Learning to Forget: Continual Prediction with LSTM. Neural Computation. 2000, 12 (10): 2451–2471. CiteSeerX 10.1.1.55.5709 . PMID 11032042. S2CID 11598600. doi:10.1162/089976600300015015. 
  17. ^ Srivastava, Rupesh Kumar; Greff, Klaus; Schmidhuber, Jürgen. Training Very Deep Networks. 22 July 2015. arXiv:1507.06228  [cs.LG]. 
  18. ^ Schmidhuber, Jürgen. Microsoft Wins ImageNet 2015 through Highway Net (or Feedforward LSTM) without Gates. 2015 [2024-01-27]. (原始内容存档于2023-11-27). 
  19. ^ Huang, Gao; Liu, Zhuang; van der Maaten, Laurens; Weinberger, Kilian. Densely Connected Convolutional Networks. 2016. arXiv:1608.06993 . 
  20. ^ Huang, Gao; Sun, Yu; Liu, Zhuang; Weinberger, Kilian. Deep Networks with Stochastic Depth. 2016. arXiv:1603.09382 . 
  21. ^ Liao, Qianli; Poggio, Tomaso. Bridging the Gaps Between Residual Learning, Recurrent Neural Networks and Visual Cortex. 2016. arXiv:1604.03640 . 
  22. ^ Xiao, Will; Chen, Honglin; Liao, Qianli; Poggio, Tomaso. Biologically-Plausible Learning Algorithms Can Scale to Large Datasets. 2018. arXiv:1811.03567 . 
  23. ^ Winding, Michael; Pedigo, Benjamin; Barnes, Christopher; Patsolic, Heather; Park, Youngser; Kazimiers, Tom; Fushiki, Akira; Andrade, Ingrid; Khandelwal, Avinash; Valdes-Aleman, Javier; Li, Feng; Randel, Nadine; Barsotti, Elizabeth; Correia, Ana; Fetter, Fetter; Hartenstein, Volker; Priebe, Carey; Vogelstein, Joshua; Cardona, Albert; Zlatic, Marta. The connectome of an insect brain. Science. 10 Mar 2023, 379 (6636): eadd9330. PMC 7614541 . PMID 36893230. S2CID 254070919. bioRxiv 10.1101/2022.11.28.516756v1 . doi:10.1126/science.add9330.