Weighting Classes in a Binary Classification Neural Network

1 次查看（过去 30 天）

Arjun Desai 2018-5-25

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/402588-weighting-classes-in-a-binary-classification-neural-network

评论： Eugene Alexander 2019-5-28

I am building a binary classification neural network. The last 3 layers of my CNN architecture are the following:

fullyConnectedLayer(2, 'Name', 'fc1');
softmaxLayer
classificationLayer

Currently, the classificationLayer uses a crossentropyex loss function, but this loss function weights the binary classes (0, 1) the same. Unfortunately, in my total data is have substantially less information about the 0 class than about the 1 class.

As a result, I want to weight the loss function to penalize misclassifying the 0 class more, with classWeights proportional to 1/(class frequency).

I noted that there is a way to weight classes in the pixelClassificationLayer but not the general classificationLayer, which I would be using as I am working on a classification problem.

How can I add class weights to my loss function for training?

3 个评论
显示 1更早的评论隐藏 1更早的评论

Arjun Desai 2019-3-24

Hi Samreen,

Unfortunately i could not find a good work around. might be something to pitch for future development.

Arjun

Eugene Alexander 2019-5-28

Please take a look at Define Custom Weighted Classification Layer and the example on Speech Command Recognition using Deep Learning. I am trying it right now on a binary classification problem.

请先登录，再进行评论。

请先登录，再回答此问题。