Can I define a custom loss function using intermediate features?
显示 更早的评论
I'm trying to customize my regression layer.
And I want to use intermediate features to my loss function.
Is there any way of utilizing features to customized loss function?
Looking forward to your answers.
Thank you for reading.
采纳的回答
Katja Mogalle
2021-7-2
1 个投票
Hi Sangmin,
I understand you want to define a loss function that requires ground truth data, intermediate network activations, and the final network output as input.
To implement such a loss function, you need to use the custom training loop approach instead of using trainNetwork. You can see an example here: Train Network Using Custom Training Loop.
If you look at the modelGradients function in that example, you'll be able to extract intermediate activations from the dlnetwork in the call to forward using the 'Outputs' name-value pair. And then you can formulate your own loss function without needing a regression layer.
There are already several typical loss functions available that you can make use of in your custom code, if applicable.
I hope this helps.
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Install Products 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!
Translated by 
