Какой слой потерь можно использовать для балансировки средней частоты? Как?

Question

Я написал скрипт на python, который вычисляет средние веса балансировки частоты для каждого класса во время пиксельной сегментации. Затем я добавил Python Layer к caffe определение модели, которое отправляет веса в функцию потерь. Основываясь на этой ссылке, пользователь упоминает, что SoftmaxWithLoss слой в кафе соответствует TensorFlow softmax_cross_entropy_with_logits, У меня вопрос, как я могу отправить веса на слой SoftmaxWithLoss? С какими другими слоями потерь можно использовать median frequency balancing? я использовал InfoGainLoss, но это не сходится. Ваша помощь очень ценится.

Answer 1

Если вы хотите взвесить "SoftmaxLoss" (то есть, перекрестная потеря энтропии) в соответствии с классами, то "InfogainLoss"это то, что вам нужно.
Обратите внимание, что информационный слой был обновлен в кафе несколько месяцев назад: теперь он интегрируется "Softmax" в расчет потерь для надежной оценки градиента.