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

Question

I have a pretrained caffe model with no loss layers. I want to do the following steps:

1. Compute the cost/grad of some layer in the net.
2. Backpropagate to compute the gradient with respect to the input layer.
3. Perform gradient descent repeating 1 and 2 to optimize input.

I can not figure out how to add a loss layer to a pretrained model to do this. In other NN frameworks you can call a backward() function and pass a cost function. Is there any way to do this in caffe?

Answer 1

You can create a custom layer in caffe for your cost function. Make a call to this cost function in the .prototxt file. You can finetune a pretrained model using your new cost function.

Finetuning is done using the below format of coomandline code:

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

More on caffe finetuning can be found here .