Infogain Loss Layer

• Layer type: InfogainLoss
• Doxygen Documentation
• Header: ./include/caffe/layers/infogain_loss_layer.hpp
• CPU implementation: ./src/caffe/layers/infogain_loss_layer.cpp

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.

Parameters

• Parameters (Parameter infogain_param)
• From ./src/caffe/proto/caffe.proto:

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