[论文笔记] Pai-megatron 细节解读之self.jitter_noise参数 (防止过拟合)

本文详细解析了Pai-megatron中self.jitter_noise参数的作用,该参数通过在训练过程中为隐藏状态添加随机噪声来减轻大模型的过拟合问题。类似的策略包括hidden_dropout和attention_dropout。self.jitter_noise通过在隐藏状态上进行逐元素乘法,引入介于0和1之间的随机值,以此提高模型的泛化能力。
摘要由CSDN通过智能技术生成
if self.training and self.jitter_noise > 0:
            hidden_states *= torch.empty_like(hidden_states).uniform_(1.0 - self.jitter_noise, 1.0 + self.jitter_noise) 请你讲一下这段代码

        对 hidden_states 添加的 self.jitter_noise 抖动,是一种减轻大模型过拟合策略。类似的减轻大模型过拟合策略的方法,还有hidden_dropout、attention_dropout。其中hidden_dropout可能带来的影响更大。 

        此代码段是MoE(Mixture of Experts)模型中的一个特定部分,用于在训练过程中为隐藏状态引入随机性,这通常被称为抖动(jitter)或噪声。以下是具体行为的解释:

if self.training and self.jitter_noise > 0:

        这行检查模型是否处于训练模式(self.training是一个标志,如果模型在训练中为

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

心心喵

喵喵(*^▽^*)

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值