我正在尝试使用caffe来实现Schroff, Kalenichenko and Philbin "FaceNet: A Unified Embedding for Face Recognition and Clustering", 2015中描述的三元组丢失。
我是新手,所以如何计算反向传播的梯度?
答案 0 :(得分:15)
我假设你将损失层定义为
layer {
name: "tripletLoss"
type: "TripletLoss"
bottom: "anchor"
bottom: "positive"
bottom: "negative"
...
}
现在你需要计算一个渐变w.r.t每个“底部”。
原始计算(我出于感情原因离开这里......)
请参阅comment更正上一学期。