Deep learning framework by BAIR

Created by
Yangqing Jia
Lead Developer
Evan Shelhamer

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