Infogain Loss Layer

A generalization of MultinomialLogisticLossLayer that takes an “information gain” (infogain) matrix specifying the “value” of all label pairs.

Equivalent to the MultinomialLogisticLossLayer if the infogain matrix is the identity.


message InfogainLossParameter {
  // Specify the infogain matrix source.
  optional string source = 1;
  optional int32 axis = 2 [default = 1]; // axis of prob

Dialogue & Discussion