computer-vision - 三重态损失反向传播梯度公式是什么？

Question

我正在尝试使用 caffe 来实现 Schroff, Kalenichenko and Philbin "FaceNet: A Unified Embedding for Face Recognition and Clustering", 2015 中描述的三元组损失.

我是新手，所以如何计算反向传播的梯度？

Answer 1

我假设您将损失层定义为

layer {
  name: "tripletLoss"
  type: "TripletLoss"
  bottom: "anchor"
  bottom: "positive"
  bottom: "negative"
  ...
}

现在您需要计算每个“底部”的梯度。

损失由下式给出:


梯度 w.r.t “ anchor ”输入( fa ):


梯度 w.r.t “正”输入( fp ):


梯度 w.r.t “负”输入( fn ):


原来的计算(我出于感情原因离开这里......)



请查看 comment更正最后一项。