MATLAB Answers

custom loss function for DNN training

99 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

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.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by