深度学习模型训练问题记录之梯度爆炸

weixin_42559117

712人浏览 · 2024-09-29 11:12:56

weixin_42559117 · 2024-09-29 11:12:56 发布

问题描述

模型训练的过程中损失值很大，且反复震荡不下降

解决方法

给模型都加参数初始化，权重参数使用 kaiming_uniform_，偏置参数使用0初始化
调大dropout率
给模型中间层增加BatchNorm/LayerNorm

结果

在改变dropout值和增加归一化层之后，问题没有得到有效解决，但是加入模型参数初始化之后，问题得到了有效解决

结论

最有用的可能是权重参数初始化，也足以看到初始化的重要性

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

小参数・大码力・易部署 | Qwen3.6-27B上线魔乐社区，基于昇腾的部署教程来了

继一周前模型开源发布后，千问再度开源Qwen3.6-27B —— 一个拥有270亿参数的稠密多模态模型，也是社区呼声最高的模型规格。Qwen3.6-27B 依然支持多模态思考与非思考模式，在智能体编程方面达到了旗舰级表现，全面超越前代开源旗舰 Qwen3.5-397B-A17B（总参数397B / 激活参数17B的MoE模型）。作为稠密架构，它无需MoE路由即可部署，是开发者在实用、可广泛部署规模