神经网络优化（初始化权重）

最新推荐文章于 2024-07-06 16:29:35 发布

原创

最新推荐文章于 2024-07-06 16:29:35 发布 · 2.5w 阅读

29 ·

CC 4.0 BY-SA版权

文章标签：

#神经网络优化初始化 #caffe

本文介绍了神经网络中权重初始化的重要性，以及Caffe框架中支持的几种初始化方法，包括constant、uniform、Gaussian、positive_unitball、XavierFiller、MSRAFiller和BilinearFiller。每种方法的特点和适用场景被详细阐述，其中Xavier和MSRAFiller更适合ReLU激活函数，有助于避免梯度消失和爆炸问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

因为传统的初始化权重问题是用标准正态分布（均值为0，方差为1）随机初始化的，这其实是存在不合理的部分。

标准正态分布：

可以看出真实数据的分布其实是在靠近坡峰的部分，符合正态分布的。

以下为转载

点击打开链接

caffe中初始化权重

首先说明：在caffe/include/caffe中的 filer.hpp文件中有它的源文件，如果想看，可以看看哦，反正我是不想看，代码细节吧，现在不想知道太多，有个宏观的idea就可以啦，如果想看代码的具体的话，可以看：http://blog.youkuaiyun.com/xizero00/article/details/50921692，写的还是很不错的（不过有的地方的备注不对，不知道改过来了没）。

文件 filler.hpp提供了7种权值初始化的方法，分别为：常量初始化（constant）、高斯分布初始化（gaussian）、positive_unitball初始化、均匀分布初始化（uniform）、xavier初始化、msra初始化、双线性初始化（bilinear）。

275 Filler<Dtype>* GetFiller(const FillerParameter& param) {
276   const std::string& type = param.type();
277   if (type == "constant") {
278     return new ConstantFiller<Dtype>(param);
279   } else if (type == "gaussian") {
280     return new GaussianFiller<Dtype>(param);
281   } else if (type == "positive_unitball") {
282     return new PositiveUnitballFiller<Dtype>(param);
283   } else if (type == "uniform") {
284     return new UniformFiller<Dtype>(param);
285   } else if (type == "xavier") {
286     return new XavierFiller<Dtype>(param);
287   } else if (type == "msra") {
288     return new MSRAFiller<Dtype>(param);
289   } else if (type == "bilinear") {
290     return new BilinearFiller<Dtype>(param);
291   } else {
292     CHECK(false) << "Unknown filler name: " << param.type();
293   }
294   return (Filler<Dtype>*