MATLAB Answers

custom loss function for DNN training

205 views (last 30 days)
Pratheek on 16 May 2019
how can i write a custom loss fucntion for DNN training. I want to try reconstruction loss


Sign in to comment.

Answers (2)

Shounak Mitra
Shounak Mitra on 17 May 2019
You can create custom layers and define custom loss functions for output layers.
The output layer uses two functions to compute the loss and the derivatives: forwardLoss and backwardLoss. The forwardLoss function computes the loss L. The backwardLoss function computes the derivatives of the loss with respect to the predictions.
For eg., to write a weighted cross entropy classification loss, try running this in the MATLAB command window
>> edit(fullfile(matlabroot,'examples','deeplearning_shared','main','weightedClassificationLayer.m'))
Hope this helps

  1 Comment

ghali ahmed
ghali ahmed on 17 Oct 2019
is there more details for a real implementation :)

Sign in to comment.

Translated by