什么损耗层可用于中位频率平衡？怎么样？

Question

我编写了一个python脚本，用于在逐像素分割期间计算每个类的中值频率平衡权重。然后，我在Python Layer模型定义中添加了caffe，该定义将权重发送到损失函数。根据{{​​3}}，用户提到caffe中的SoftmaxWithLoss图层对应于TensorFlow softmax_cross_entropy_with_logits。 我的问题是如何将权重发送到 SoftmaxWithLoss 层？ median frequency balancing可以使用哪些其他损耗图层？我使用InfoGainLoss，但它没有收敛。非常感谢您的帮助。

Answer 1

如果您想根据课程加权"SoftmaxLoss"（即交叉熵损失），那么您需要"InfogainLoss"。
请注意，几个月前，infogain层在caffe中进行了升级：它现在将"Softmax"集成到损失计算中，以进行稳健的梯度估计。