Pytorch基本使用—参数初始化

news/2024/11/6 13:59:30/

深度学习模型参数初始化是指在训练深度神经网络时,对网络的权重和偏置进行初始化的过程。合适的参数初始化可以加速模型的收敛,并提高模型的性能。

✨ 1 基本介绍

在深度学习中,常用的参数初始化方法有以下几种:

  1. 零初始化(Zero Initialization):将所有权重和偏置初始化为0。然而,这种方法会导致所有神经元具有相同的输出,无法破坏对称性,因此不常用。
  2. 随机初始化(Random Initialization):将权重和偏置随机初始化为较小的随机值。这种方法可以打破对称性,但并不能保证初始化的权重和偏置能够适应网络的输入和输出分布。
  3. Xavier初始化(Xavier Initialization):根据每一层的输入维度和输出维度的大小来进行初始化。Xavier初始化方法根据激活函数的导数和输入输出的维度来调整初始化的范围,使得每一层的激活值保持在一个合适的范围内。
  4. He初始化(He Initialization):类似于Xavier初始化,但在计算权重的标准差时,将输入维度除以2。这是由于ReLU等非线性激活函数的性质导致的。
  5. 预训练初始化(Pretraining Initialization):在某些情况下,可以使用预训练的模型参数来初始化新的模型。例如,利用在大规模数据集上预训练的模型参数来初始化新任务的模型,可以加快模型的收敛速度。

需要注意的是,不同的参数初始化方法适用于不同的网络架构和激活函数。在实际应用中,需要根据具体情况选择适当的参数初始化方法。此外,还可以通过调整学习率和正则化等技巧来进一步优化训练过程。

✨ 2 零初始化(不常用)

🎈 2.1 理论

这里主要分析一下神经网络为什么不能将参数全部初始化为0
假设我们有下面的网络(为了简单,全部以线性函数计算):

第一层计算为:
在这里插入图片描述
第二层计算为:
在这里插入图片描述
以参数W11和W12的反向传播为例,梯度为:
在这里插入图片描述
因为都是0,则梯度为0,则参数更新停止。

✨ 3 Xavier初始化

第二节我们简单总结了为什么神经网络参数不能输出化为0,接下来我们讨论Xavier初始化。

🎃 3.1 介绍

在神经网络中,每个神经元的输入是由上一层的神经元输出和权重参数决定的。如果权重参数初始化过大,会导致输入值变得很大,从而使得激活函数的导数趋近于0,造成梯度消失问题。相反,如果权重参数初始化过小,会导致输入值变得很小,从而使得激活函数的导数趋近于1,造成梯度爆炸问题。

Xavier初始化通过根据网络层的输入和输出维度来合理地初始化权重参数,使得权重参数的方差保持在一个相对稳定的范围内。这样可以避免梯度消失和梯度爆炸问题,有助于提高网络的训练效果。

⛱️ 3.2 推导

这里我们以下列网络为例:
在这里插入图片描述

首先看前向传播
在这里插入图片描述
方差为(这里应用概率论相关计算公式,需要注意的是这里Xi经过归一化,E(Xi)=0)
在这里插入图片描述
如果Xi和Wi独立同分布,那么D(a1)的最终公式为
在这里插入图片描述
这里在这里插入图片描述代表着输入维度
而我们的目标是在这里插入图片描述,因此在这里插入图片描述


与上述计算方式一样,反向传播最终结果是在这里插入图片描述。只是这里,在这里插入图片描述是输出的维度大小。


但是在这里插入图片描述在这里插入图片描述一般情况下是不同的,因此,这里采取一种折中的方式
在这里插入图片描述,我们让在这里插入图片描述在区间[a, b]上均匀采样(均匀分布)
结合均匀分布方差公式在这里插入图片描述,解出Xavier初始化采样范围为在这里插入图片描述

🎈 3.3 构造

torch.nn.init.xavier_uniform_(tensor, a=0, b=1)
  1. tensor:需要填充的张量
  2. a:均匀分布的下界
  3. b:均匀分布的上界

☃️ 3.4 例子

w = torch.empty(3, 5)
nn.init.uniform_(w)

result:

tensor([[0.2116, 0.3085, 0.5448, 0.6113, 0.7697],[0.8300, 0.2938, 0.4597, 0.4698, 0.0624],[0.5034, 0.1166, 0.3133, 0.3615, 0.3757]])

http://www.ppmy.cn/news/884540.html

相关文章

【HCIA】10.VLAN间通信

VLAN间通信的解决方法 使用路由器的物理接口 路由器三层接口作为网关,转发本网段前往其它网段的流量。路由器三层接口无法处理携带VLAN Tag的数据帧,因此交换机上联路由器的接口需配置为Access。路由器的一个物理接口作为一个VLAN的网关,因此…

宿舍台灯什么光对眼睛好?推荐适合学生宿舍用的台灯

现在也进入了开学季,很多学生初来乍到,不知道需要带些什么去学校,以我上学的经验来看,宿舍最不稳定的就是光照时间和亮度了,有时候宿舍早早关灯了,基本上人手一个台灯,而用什么光的台灯对眼睛好…

学生护眼台灯哪种好?2022值得入手的学生智能护眼台灯

在我国的儿童近视率是在逐渐增加的,在儿童以及青少年的年龄区间占比是最大的,所以不得不重视孩子的视力问题了,在学生时代的晶状体是比较清晰的,在学习时用护眼台灯也是可以有效的缓解眼部疲劳和预防近视,分享2022值得…

台灯AAA和AA有什么区别?国AA台灯有必要买吗

护眼灯AAA和AA的区别还蛮大的,一个是国家标准AA,用于衡量台灯的照度和均匀度的高低好坏,另一个是国际照明委员会认证的健康照明等级,用于衡量台灯的光线是否健康。所以这两个指标是衡量不同的东西,不能混淆比较和使用&…

初中生护眼台灯哪个牌子好?2022开学寝室台灯推荐

迎来开学季,学生们都在准备入学的用品,很多家长都会给孩子考虑非常周全,不管是在校住还是在家住,都能用得上的物品,随着年龄增大压力越大,读书也是如此的,升学就是课业压力越来越大的&#xff0…

台灯怎么选对眼睛好?盘点护眼台灯品牌排行

台灯是许多家庭使用小家电,而真正有用的是护眼台灯,能最大限度的保护眼睛,光线太弱的情况写字、阅读对眼睛伤害最大,选择一看好的护眼台灯很有必要的。 怎么选择护眼台灯? (1) RG0无蓝光等级,现在的电子产…

从大力教育到大力智能:字节用台灯切入在线教育靠谱吗?

最近,在线教育企业们的日子有些不好过。 2020年火爆一时的在线教育,在2021年反而销声匿迹。时间进入七月份,本来是各大平台大张旗鼓招生的好时机,如今却没有一家在线教育企业打出招生广告。相比去年铺天盖地的广告洗脑&#xff0…

2021-2027年中国智能台灯行业市场研究及前瞻分析报告

【报告类型】产业研究 【报告价格】4500起 【出版时间】即时更新(交付时间约3个工作日) 【发布机构】智研瞻产业研究院 【报告格式】PDF版 本报告介绍了中国智能台灯行业市场行业相关概述、中国智能台灯行业市场行业运行环境、分析了中国智能台灯行…