【问题标题】:how to implement L2 Regularization in caffe or DIGITS?如何在 caffe 或 DIGITS 中实现 L2 正则化?
【发布时间】:2017-07-14 05:00:07
【问题描述】:

我正在使用 Caffe 和 NVIDIA DIGITS。我想使用在 ImageNet 上预训练的 AlexNet,并想根据我的医疗数据对其进行微调。我有近 1000 张图像,其中 80% 用于训练,我通过数据增强(使用裁剪和旋​​转)生成了 40,000 张图像。但是我面临严重的过度拟合。我试图通过添加多个 dropout 层来克服这个问题。结果从:

到:

但我的准确性并没有提高。

我的网络规格:

AlexNet pre-trained on ImageNet

base learning rate: 0.001

learning rate multiplier: 0.1 for convolution layers and 1 for fully connected layers and xavier weight initialisation.

dropout: 0.5

现在我想添加 L2 正则化。我在 Caffe 中没有找到这样的层,我应该自己做。

first question:我的问题你有什么解决办法吗? (我尝试了其他方法,例如更改步长,将学习率从 1 更改为 10^(-5),我发现 0.001 更好,权衡衰减变化,添加各种 dropout 层(如您所见))

second question:你能帮我如何实现L2正则化吗?

【问题讨论】:

    标签: machine-learning neural-network deep-learning caffe


    【解决方案1】:

    在 caffe 中默认有 L2 正则化。
    请参阅this thread 了解更多信息。

    【讨论】:

      猜你喜欢
      • 2017-05-25
      • 2012-03-11
      • 1970-01-01
      • 1970-01-01
      • 2021-05-05
      • 1970-01-01
      • 2016-07-22
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多