Tutorial: Triplet Loss Layer Design for CNN

 

Tutorial:  Triplet Loss Layer Design for CNN

Xiao Wang  2016.05.02

 

　　Triplet Loss Layer could be a trick for further improving the accuracy of CNN. Today, I will introduce the whole process, and display the code for you. This tutorial mainly from the blog: 

　　http://blog.csdn.net/tangwei2014/article/details/46812153 

　　http://blog.csdn.net/tangwei2014/article/details/46788025

　　and the paper: <FaceNet: A Unified Embedding for Face Recognition and Clustering>.

 

　　First, Let's talk about how to add the layer into caffe and make test this layer to check whether it works or not. And then, we will discuss the paper and introduce the process of how the triplet loss come from. In the new version of caffe framework, it mainly consists of these steps for add a new layer i.e. 

　　step 1. add the paprameter message in the corresponding layer, which located in ./src/caffe/proto/caffe.proto ;

　　step 2. add the declaration information of the layer in ./include/caffe/***layers.hpp ;

　　step 3. add the corresponding .cpp and .cu files in ./src/caffe/layers/, realize the function of the new added layer;

　　step 4. add test code of new added layers in ./src/caffe/gtest/, test its foreward and back propagation and its computation speed. 

 

　　Let's do it step by step. 

　　First, we add triplet loss layer in caffe.proto file:

　　we could found that in line 101, it said: SolverParameter next available ID: 40 (last added: momentum2), thus we add the ID: 40 as the new added information :　　

　　　　　　　　message RankParameter {
　　　　　　　　optional uint32 neg_num = 1 [default = 1];
　　　　　　　　optional uint32 pair_size = 2 [default = 1];
　　　　　　　　optional float hard_ratio = 3;
　　　　　　　　optional float rand_ratio = 4;
　　　　　　　　optional float margin = 5 [default = 0.5];
　　　　　　　　}

 

　　　　

　　　　

　　Second, we add the declearation information about triplet loss layer in ./include/caffe/TripletLoss_layers.hpp 

　　

 

　　Third, We compile the triplet loss layer of .cpp and .cu file 

　　First of all is the .cpp file 

#include <vector>

#include <algorithm>
#include <cmath>
#include <cfloat>

#include "caffe/layer.hpp"
#include "caffe/util/io.hpp"
#include "caffe/util/math_functions.hpp"
#include "caffe/vision_layers.hpp"

using std::max;
using namespace std;
using namespace cv;

namespace caffe {

int myrandom (int i) { return caffe_rng_rand()%i;}


template <typename Dtype>
void RankHardLossLayer<Dtype>::Reshape(
  const vector<Blob<Dtype>*>& bottom, const vector<Blob<Dtype>*>& top) {
  LossLayer<Dtype>::Reshape(bottom, top);

  diff_.ReshapeLike(*bottom[0]);
  dis_.Reshape(bottom[0]->num(), bottom[0]->num(), 1, 1);
  mask_.Reshape(bottom[0]->num(), bottom[0]->num(), 1, 1);
}


template <typename Dtype>
void RankHardLossLayer<Dtype>::set_mask(const vector<Blob<Dtype>*>& bottom)
{

    RankParameter rank_param = this->layer_param_.rank_param();
    int neg_num = rank_param.neg_num();
    int pair_size = rank_param.pair_size();
    float hard_ratio = rank_param.hard_ratio();
    float rand_ratio = rank_param.rand_ratio();
    float margin = rank_param.margin();

    int hard_num = neg_num * hard_ratio;
    int rand_num = neg_num * rand_ratio;

    const Dtype* bottom_data = bottom[0]->cpu_data();
    const Dtype* label = bottom[1]->cpu_data();
    int count = bottom[0]->count();
    int num = bottom[0]->num();
    int dim = bottom[0]->count() / bottom[0]->num();
    Dtype* dis_data = dis_.mutable_cpu_data();
    Dtype* mask_data = mask_.mutable_cpu_data();

    for(int i = 0; i < num * num; i ++)
    {
        dis_data[i] = 0;
        mask_data[i] = 0;
    }

    // calculate distance
    for(int i = 0; i < num; i ++)
    {
        for(int j = i + 1; j < num; j ++)
        {
            const Dtype* fea1 = bottom_data + i * dim;
            const Dtype* fea2 = bottom_data + j * dim;
            Dtype ts = 0;
            for(int k = 0; k < dim; k ++)
            {
              ts += (fea1[k] * fea2[k]) ;   
            }                                
            dis_data[i * num + j] = -ts;    
            dis_data[j * num + i] = -ts;    
        }
    }

    //select samples

    vector<pair<float, int> >negpairs;
    vector<int> sid1;
    vector<int> sid2;


    for(int i = 0; i < num; i += pair_size)
    {
        negpairs.clear();
        sid1.clear();
        sid2.clear();
        for(int j = 0; j < num; j ++)
        {
            if(label[j] == label[i])
                continue;
            Dtype tloss = max(Dtype(0), dis_data[i * num + i + 1] - dis_data[i * num + j] + Dtype(margin));
            if(tloss == 0) continue;

            negpairs.push_back(make_pair(dis_data[i * num + j], j));
        }
        if(negpairs.size() <= neg_num)
        {
            for(int j = 0; j < negpairs.size(); j ++)
            {
                int id = negpairs[j].second;
                mask_data[i * num + id] = 1;
            }
            continue;
        }
        sort(negpairs.begin(), negpairs.end());

        for(int j = 0; j < neg_num; j ++)
        {
            sid1.push_back(negpairs[j].second);
        }
        for(int j = neg_num; j < negpairs.size(); j ++)
        {
            sid2.push_back(negpairs[j].second);
        }
        std::random_shuffle(sid1.begin(), sid1.end(), myrandom);
        for(int j = 0; j < min(hard_num, (int)(sid1.size()) ); j ++)
        {
            mask_data[i * num + sid1[j]] = 1;
        }
        for(int j = hard_num; j < sid1.size(); j++)
        {
            sid2.push_back(sid1[j]);
        }
        std::random_shuffle(sid2.begin(), sid2.end(), myrandom);
        for(int j = 0; j < min( rand_num, (int)(sid2.size()) ); j ++)
        {
            mask_data[i * num + sid2[j]] = 1;
        }

    }


}




template <typename Dtype>
void RankHardLossLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
    const vector<Blob<Dtype>*>& top) {

    const Dtype* bottom_data = bottom[0]->cpu_data();
    const Dtype* label = bottom[1]->cpu_data();
    int count = bottom[0]->count();
    int num = bottom[0]->num();
    int dim = bottom[0]->count() / bottom[0]->num();


    RankParameter rank_param = this->layer_param_.rank_param();
    int neg_num = rank_param.neg_num();      // 4
    int pair_size = rank_param.pair_size();  // 5
    float hard_ratio = rank_param.hard_ratio();
    float rand_ratio = rank_param.rand_ratio();
    float margin = rank_param.margin();
    Dtype* dis_data = dis_.mutable_cpu_data();
    Dtype* mask_data = mask_.mutable_cpu_data();

    set_mask(bottom);
    Dtype loss = 0;
    int cnt = neg_num * num / pair_size * 2;

    for(int i = 0; i < num; i += pair_size)
    {
        for(int j = 0; j < num; j++)
        {
            if(mask_data[i * num + j] == 0) 
                continue;
            Dtype tloss1 = max(Dtype(0), dis_data[i * num + i + 1] - dis_data[i * num + j] + Dtype(margin));
            Dtype tloss2 = max(Dtype(0), dis_data[i * num + i + 1] - dis_data[(i + 1) * num + j] + Dtype(margin));
            loss += tloss1 + tloss2;
        }
    }

    loss = loss / cnt;
    top[0]->mutable_cpu_data()[0] = loss;
}




template <typename Dtype>
void RankHardLossLayer<Dtype>::Backward_cpu(const vector<Blob<Dtype>*>& top,
    const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {


    const Dtype* bottom_data = bottom[0]->cpu_data();
    const Dtype* label = bottom[1]->cpu_data();
    Dtype* bottom_diff = bottom[0]->mutable_cpu_diff();
    int count = bottom[0]->count();
    int num = bottom[0]->num();
    int dim = bottom[0]->count() / bottom[0]->num();

    RankParameter rank_param = this->layer_param_.rank_param();
    int neg_num = rank_param.neg_num();
    int pair_size = rank_param.pair_size();
    float hard_ratio = rank_param.hard_ratio();
    float rand_ratio = rank_param.rand_ratio();
    float margin = rank_param.margin();

    Dtype* dis_data = dis_.mutable_cpu_data();
    Dtype* mask_data = mask_.mutable_cpu_data();

    for(int i = 0; i < count; i ++ )
        bottom_diff[i] = 0;

    int cnt = neg_num * num / pair_size * 2;

    for(int i = 0; i < num; i += pair_size)
    {
        const Dtype* fori = bottom_data + i * dim;
        const Dtype* fpos = bottom_data + (i + 1) * dim;

        Dtype* fori_diff = bottom_diff + i * dim;
        Dtype* fpos_diff = bottom_diff + (i + 1) * dim;
        for(int j = 0; j < num; j ++)
        {
            if(mask_data[i * num + j] == 0) continue;
            Dtype tloss1 = max(Dtype(0), dis_data[i * num + i + 1] - dis_data[i * num + j] + Dtype(margin));
            Dtype tloss2 = max(Dtype(0), dis_data[i * num + i + 1] - dis_data[(i + 1) * num + j] + Dtype(margin));

            const Dtype* fneg = bottom_data + j * dim;
            Dtype* fneg_diff  = bottom_diff + j * dim;
            if(tloss1 > 0)
            {
                for(int k = 0; k < dim; k ++)
                {
                    fori_diff[k] += (fneg[k] - fpos[k]); // / (pairNum * 1.0 - 2.0);
                    fpos_diff[k] += -fori[k]; // / (pairNum * 1.0 - 2.0);
                    fneg_diff[k] +=  fori[k];
                }
            }
            if(tloss2 > 0)
            {
                for(int k = 0; k < dim; k ++)
                {
                    fori_diff[k] += -fpos[k]; // / (pairNum * 1.0 - 2.0);
                    fpos_diff[k] += fneg[k]-fori[k]; // / (pairNum * 1.0 - 2.0);
                    fneg_diff[k] += fpos[k];
                }
            }

        }
    }

    for (int i = 0; i < count; i ++)
    {
        bottom_diff[i] = bottom_diff[i] / cnt;
    }

}

#ifdef CPU_ONLY
STUB_GPU(RankHardLossLayer);
#endif

INSTANTIATE_CLASS(RankHardLossLayer);
REGISTER_LAYER_CLASS(RankHardLoss);

}  // namespace caffe

　　and the .cu file 

#include <vector>

#include "caffe/layer.hpp"
#include "caffe/util/io.hpp"
#include "caffe/util/math_functions.hpp"
#include "caffe/vision_layers.hpp"

namespace caffe {

template <typename Dtype>
void RankHardLossLayer<Dtype>::Forward_gpu(const vector<Blob<Dtype>*>& bottom,
    const vector<Blob<Dtype>*>& top) {
  Forward_cpu(bottom, top);
}

template <typename Dtype>
void RankHardLossLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
    const vector<bool>& propagate_down, const vector<Blob<Dtype>*>& bottom) {
   Backward_cpu(top, propagate_down, bottom);
}

INSTANTIATE_LAYER_GPU_FUNCS(RankHardLossLayer);

}  // namespace caffe

　　

　　Finally, we make the caffe file and check whether have some mistakes about it.

 

--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------

                   

　　Let's continue to talk about the triplet loss:

　　Just like the above figure showns,  the triplet loss usually have three components, i.e. the anchors, the positive, and the negative. What we are going to do is try to reduce the distance between the archor and the same, and push the different from the anchors.

　　Thus, the whole loss could be described as following:

　　Only select triplets randomly may lead to slow converage of the network, and we need to find those hard triplets, that are active and can therefore contribute to improving the model. The following section will give you an explanination about the approach.

　　Triplet Selection:

　　There are two appproaches for generate triplets, i.e.

　　1. Generate triplets offline every n steps, using the most recent newwork checkpoint and computing the argmin and argmax on a subset of the data.

　　2. Generate the triplets online. This can be done by selecting the hard positive/negative exemplars form within a mini-batch.

 

　　This paper use all anchor-positive pairs in a mini-batch while still selecting the hard negatives. the all anchor-positive method was more stable and converaged slightly faster at the begining of training.

 

　　The code could refer the github page: https://github.com/wangxiao5791509/caffe-video_triplet 

　　

layer {     
    name: "loss"    
    type: "RankHardLoss"    
    rank_param{     
        neg_num: 4  
        pair_size: 2    
        hard_ratio: 0.5     
        rand_ratio: 0.5     
        margin: 1   
    }   
    bottom: "norml2"    
    bottom: "label"     
}

 

 

---

 

Triplet Loss Implementation using Pytorch: 

　　the following document comes from: https://pytorch.org/docs/stable/nn.html#tripletmarginloss 

　　Creates a criterion that measures the triplet loss given an input tensors x1, x2, x3 and a margin with a value greater than 0. This is used for measuring a relative similarity between samples. A triplet is composed by a, p and n: anchor, positive examples and negative example respectively. The shapes of all input tensors should be (N,D)(N,D).

 

　　The distance swap is described in detail in the paper Learning shallow convolutional feature descriptors with triplet losses by V. Balntas, E. Riba et al.

 

　　The loss function for each sample in the mini-batch is:

 

　　　　L(a,p,n)=max{d(ai,pi)−d(ai,ni)+margin,0}L(a,p,n)=max{d(ai,pi)−d(ai,ni)+margin,0}

 

　　where d(xi,yi)=∥xi−yi∥pd(xi,yi)=‖xi−yi‖p.

 

Parameters:

margin (float, optional) – Default: 1. p (int, optional) – The norm degree for pairwise distance. Default: 2. swap (float, optional) – The distance swap is described in detail in the paperLearning shallow convolutional feature descriptors with triplet losses by V. Balntas, E. Riba et al. Default: False. size_average (bool, optional) – By default, the losses are averaged over observations for each minibatch. However, if the field size_average is set to False, the losses are instead summed for each minibatch. Default: True reduce (bool, optional) – By default, the losses are averaged or summed over observations for each minibatch depending on size_average. When reduce is False, returns a loss per batch element instead and ignores size_average. Default: True

 

Shape:

• Input: (N,D)(N,D) where D is the vector dimension.
• Output: scalar. If reduce is False, then (N).

>>> triplet_loss = nn.TripletMarginLoss(margin=1.0, p=2)
>>> input1 = torch.randn(100, 128, requires_grad=True)
>>> input2 = torch.randn(100, 128, requires_grad=True)
>>> input3 = torch.randn(100, 128, requires_grad=True)
>>> output = triplet_loss(input1, input2, input3)
>>> output.backward()

Example: 

import torch.nn as nn 
triplet_loss = nn.TripletMarginLoss(margin=1.2, p=2)   
# 计算特征向量
anchor = model.forward(data[0])
positive = model.forward(data[1])
negative = model.forward(data[2])
# 计算三元组loss
loss = triplet_loss.forward(anchor, positive, negative) 
loss.backward()
optimizer.step()