【论文笔记】Unsupervised Generative Adversarial Cross-Modal Hashing

Pipeline

文章基于GAN网络,提出一种无监督跨模态检索算法,算法框架图如下

网络接收原始文本和图片作为输入,采用神经网络分别提取特征文本和图片特征,

生成器

生成器采用两路网络,分别接收上面提取到的文本和图片特征向量,后面接两个全连接层,一层用于将特征向量映射到公共空间,一层用于生成哈希码。

给定某个模态中的查询样例,生成器目的在于拟合Manifold Pairs的分布,从另一个模态的样例中找到和查询样例相关的样例。相关程度采用下面的概率判断
p θ ( x U ∣ q ) = exp ⁡ ( − ∥ h ( q ) − h ( x U ) ∥ 2 ) ∑ x U exp ⁡ ( − ∥ h ( q ) − h ( x U ) ∥ 2 ) p_{\theta}\left(x^{U} | q\right)=\frac{\exp \left(-\left\|h(q)-h\left(x^{U}\right)\right\|^{2}\right)}{\sum_{x^{U}} \exp \left(-\left\|h(q)-h\left(x^{U}\right)\right\|^{2}\right)} pθ(xUq)=xUexp(h(q)h(xU)2)exp(h(q)h(xU)2)
q q q为查询样例, x U x^{U} xU为另一模态的样例, h ( ∗ ) h(*) h()为哈希码

鉴别器

鉴别器接收生成器生成的’假’样本和关联图采样得到的Manifold Pairs(’真‘样本)作为输入,采用triplet ranking loss作为损失函数。

构建关联图
首先分别为文本和图片模态构建两个无向图, G r a p h i = ( V , W i ) Graph_{i}=(V,W_{i}) Graphi=(V,Wi) G r a p h i = ( V , W i ) Graph_{i}=(V,W_{i}) Graphi=(V,Wi) V V V为顶点, W i W_{i} Wi W j W_{j} Wj为相似度矩阵,定义如下
w ( p , q ) = { 1 : x p ∈ N N k ( x q ) 0 :  otherwise  w(p, q)=\left\{\begin{array}{cc} 1: & x_{p} \in N N_{k}\left(x_{q}\right) \\ 0: & \text { otherwise } \end{array}\right. w(p,q)={1:0:xpNNk(xq) otherwise 
N N k ( x q ) N N_{k}\left(x_{q}\right) NNk(xq) x q x_{q} xq k k k邻近节点,需要注意的是,跨模态数据集中成对信息自然存在,若文本查询样例 q j q_{j} qj和文本 t k t_{k} tk有相同的潜在结构,即语义相似,则与 t k t_{k} tk成对的图像 i k i_{k} ik q j q_{j} qj也有相同的潜在结构,反之亦然。

判别概率
鉴别器接收生成器和关联图产生的样本作为输入,输出每个对的得分做判别结果。给定查询样例 q q q和实例 x U x^{U} xU,三元组损失定义如下
f ϕ ( x G , q ) = max ⁡ ( 0 , m + ∥ h ( q ) − h ( x M ) ∥ 2 − ∥ h ( q ) − h ( x G ) ∥ 2 ) f_\phi(x^G,q) = \max(0, m+\|h(q) - h(x^M)\|^2 - \|h(q) - h(x^G)\|^2) fϕ(xG,q)=max(0,m+h(q)h(xM)2h(q)h(xG)2)

  • x U x^{U} xU为相关图生成的真实样本
  • x G x^{G} xG为生成器生成的样本
  • m m m为margin值,文章中设为1
  • 三元组损失使得查询样例与真实样本间的距离比生成器生成的样本之间的距离要近

给定 q q q,样本 x x x的预测概率定义如下:
D ( x ∣ q ) = s i g m o i d ( f ϕ ( x , q ) ) = exp ⁡ ( f ϕ ( x , q ) ) 1 + exp ⁡ ( f ϕ ( x , q ) ) D(x|q) = sigmoid(f_\phi(x,q)) = \frac{\exp(f_\phi(x,q))}{1+\exp(f_\phi(x,q))} D(xq)=sigmoid(fϕ(x,q))=1+exp(fϕ(x,q))exp(fϕ(x,q))

鉴别器就是使得 D ( x M ∣ q ) D(x^M|q) D(xMq)更大 , D ( x G ∣ q ) ,D(x^G|q) D(xGq)更小,使用鉴别器模型进行跨模态检索。

生成对抗学习

V ( G , D ) = min ⁡ θ max ⁡ ϕ ∑ j = 1 n ( E x ∼ p t r u e ( x M ∣ q j ) [ log ⁡ ( D ( x M ∣ q j ) ) ] + E x ∼ p θ ( x G ∣ q j ) [ log ⁡ ( 1 − D ( x G ∣ q j ) ) ] ) \begin{aligned} \mathcal{V}(G, D)=& \min _{\theta} \max _{\phi} \sum_{j=1}^{n}\left(E_{x \sim p_{t r u e}\left(x^{M} | q^{j}\right)}\left[\log \left(D\left(x^{M} | q^{j}\right)\right)\right]\right.\\ &\left.+E_{x \sim p_{\theta}\left(x^{G} | q^{j}\right)}\left[\log \left(1-D\left(x^{G} | q^{j}\right)\right)\right]\right) \end{aligned} V(G,D)=θminϕmaxj=1n(Exptrue(xMqj)[log(D(xMqj))]+Expθ(xGqj)[log(1D(xGqj))])

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值