一、改进预测函数,使其具有对称性
原来的预测函数是:
其不满足对称性,下面将其改进为具有对称性的预测函数
二元组和三元组模型的损失函数与原来类似
方案一:
实际修改,train阶段修改SCIR_pairwise_net.prototxt
增加与concat22层功能相同层,输入将f(xi),f(xj)反过来,即将pool2-2,pool2-1作为concat22b(新增加的层)的输入,相应增加conv31b的输入,relu31bb,pool3bb,dropbb,fc1_bb,relu3_b,fc3bb,并将fc3bb(wTg(xj,xi))添加到doublet loss的输入中。更改doublet_loss_layer.cu
test阶段同样对应修改SCIR_pairwise_test_net.prototxt
二、将CIR的预测值从一次项扩展为二次项(下一步工作)
由于CIR的预测值为一次项,为了增强模型对CIR特征的拟合能力,我们引入CIR特征的二次项,将预测函数变为:
通过添加全连接层,获得M