Caffe使用自定义成本函数计算相对于输入的梯度

Question

我有一个没有损失层的预训练caffe模型。我想执行以下步骤：

1. 计算网络中某层的成本/等级。
2. 反向传播以计算相对于输入图层的渐变。
3. 执行梯度下降重复1和2以优化输入。

我无法弄清楚如何将损耗图层添加到预训练模型来执行此操作。在其他NN框架中，您可以调用backward（）函数并传递成本函数。有没有办法在caffe中做到这一点？

Answer 1

您可以在caffe中为自己的成本函数创建自定义图层。在.prototxt文件中调用此成本函数。您可以使用新的成本函数微调预训练模型。

使用以下格式的coomandline代码完成Finetuning：

./build/tools/caffe train --solver theAboveMentioned.prototxt --weights thePreTrainedWeightsFile

有关caffe finetuning的更多信息，请here。