Image captioning评价方法之CIDEr

文章地址:CIDEr: Consensus-based Image Description Evaluation

代码地址(非官方,且代码实现的是CIDEr-D):https://github.com/tylin/coco-caption

文章由微软和Virginia Tech联合发表在CVPR2015上。

CIDEr是专门用来评价image captioning效果的评价指标。

下面先定义一些符号。对于一张图片 I i I_i Ii,candidate sentence定义为 c i c_i ci, reference sentences定义为 S i = { s i 1 , . . . , s i m } S_i=\{s_{i1}, ..., s_{im}\} Si={ si1,...,sim}。一个n元组(n-gram)定义为 ω k \omega_k ωk,一个n元组为一个或者多个单词组成的有序序列。本文n取1,2,3,4。

一、CIDE

在进行CIDE计算之前,先将所有句子中的单词转换为原始形式,及“fishes”、“fishing”、“fished”都转换为fish。

文章认为评价指标应该考虑几点:

  1. candidate sentence中的n-grams在reference sentences中出现的频率
  2. 类似1,没在reference sentences中出现的n-grams也不应该在candidate sentence中出现
  3. 在所有图片的描述经常出现的n-grams,应该降低它们的权重(因为这种n元组没有太多针对性的信息含量,即与当前图片的信息相关性较小)

基于上述三点,文章提出一个叫TF-IDF(Term Frequency Inverse Document Frequency)的概念,TF表示在当前candidate sentence中出现的频率,IDF表示反比在所有reference sentences出现的频率。

对于一个n元组 ω k \omega_k ωk出现在某一个reference sentence s i j s_{ij} sij中的次数记为 h k ( s i j ) h_k(s_{ij}) hk(sij),同理, h k ( c i ) h_k(c_i) hk(ci)表示该n元组出现在 c i c_i ci的次数。那么对于该n元组的TF-IDF表示为

g k ( s i j ) = h k ( s i j ) ∑ ω l ∈ Ω h l ( s i j ) l o g ( ∣ I ∣ ∑ I p ∈ I m i n ( 1 , ∑ q h k ( s p q ) ) ) g_k(s_{ij})=\frac{h_k(s_{ij})}{\sum_{\omega_l\in{\Omega}} h_l(s_{ij})}log(\frac{|I|}{\sum_{I_p\in I}min(1, \sum_q h_k(s_{pq}))}) gk(sij)=ωlΩhl(sij)hk(sij)log(IpImin(1,qhk(spq))I)

式中 Ω \Omega Ω表示所有的n元组的个数, I I I表示所有的图片个数。

可以看出对于某一个n元组 ω k \omega_k ωk来说,TF-IDF就是TF*IDF, 式子的左边就是TF右边就是IDF。在 s i j s_{ij} sij中n元组出现次数越多,TF值就越大。n元组在所有的 s p q s_{pq} spq中出现的次数越多,IDF就越小。

C I D E r n CIDEr_n CIDErn则是用来计算candidate sentence和reference sentences在n长度的元组表示时的相似度,用公式表示为

C I D E r n ( c i , S i ) = 1 m ∑ j g n ( c i ) ⋅ g n ( s i j ) ∣ ∣ g n ( c i ) ∣ ∣ ⋅ ∣ ∣ g n ( s i j ) ∣ ∣ CIDEr_n(c_i, S_i)=\frac{1}{m}\sum_j\frac{\textbf{g}^n(c_i)\cdot \textbf{g}^n(s_{ij})}{||\textbf{g}^n(c_i)||\cdot ||\textbf{g}^n(s_{ij})||} CIDErn(ci,Si)=m1jgn(ci)gn(sij)gn(ci)gn(sij)

式中 g n ( c i ) \textbf{g}^n(c_i) gn(ci)是由n-gram取长度n时 g k ( c i ) g_k(c_i) gk(ci)组成的向量,对应的 ∣ ∣ g n ( c i ) ∣ ∣ ||\textbf{g}^n(c_i)|| gn(ci)表示该向量的模。 g n ( s i j ) \textbf{g}^n(s_{ij}) gn(sij)含义类似。m为对应图片 I i I_i Ii有多少个reference sentences。

从式中可以看出,就是向量的内积除以向量的模,含义就是两个向量的相似程度。

最终n变化时,所有的 C I D E r n CIDEr_n CIDErn取平均得到 C I D E r CIDEr CIDEr

C I D E r ( c i , S i ) = ∑ n = 1 N ω n C I D E r n ( c i , S i ) CIDEr(c_i, S_i)=\sum^N_{n=1}\omega_nCIDEr_n(c_i, S_i) CIDEr(ci,Si)=n=1NωnCIDErn(ci,Si)

其中 ω n = 1 / N \omega_n=1/N ωn=1/N, 文章取N=4。

二、CIDEr-D

文章提出了CIDEr的修改版本CIDEr-D,目的是为了防止评价指标的gaming问题。gaming问题就是防止针对某一种评价指标来优化算法使得当人类给出的分很低,但是评价指标给出的分很高。

首先,去除了将句子中的单词转换为原始形式这一步,为了就是让句子生成时对应的状态是对的。

第二,在CIDEr评价指标中,对一些重复单词得到的长句子,CIDEr的分数越高,为了减少这种影响,文章加入了高斯因子,用于惩罚candidate和reference sentence长度相差较大的情况。

最后,给一些重复高置信度(即信息含量较多的句子,比如图片有一条鱼,那句子重复说fish的情况)的单词的句子加惩罚。例如对于某个指定的n元组来说,取该n元组在candidate和reference sentence出现次数最小的数。(类似于Bleu的操作)

那么 C I D E r − D n ( c i , S i ) CIDEr-D_n(c_i, S_i) CIDErDn(ci,Si)

C I D E r − D n ( c i , S i ) = 10 m ∑ j e − ( l ( c i ) − l ( s i j ) 2 2 σ 2 ∑ j m i n ( g n ( c i ) , g n ( s i j ) ) ⋅ g n ( s i j ) ∣ ∣ g n ( c i ) ∣ ∣ ⋅ ∣ ∣ g n ( s i j ) ∣ ∣ CIDEr-D_n(c_i, S_i)=\frac{10}{m}\sum_j e^{\frac{-(l(c_i)-l(s_{ij})^2}{2\sigma^2}}\sum_j\frac{min(\textbf{g}^n(c_i),\textbf{g}^n(s_{ij}))\cdot \textbf{g}^n(s_{ij})}{||\textbf{g}^n(c_i)||\cdot ||\textbf{g}^n(s_{ij})||} CIDErDn(ci,Si)=m10je2σ2(l(ci)l(sij)2jgn(ci)gn(sij)min(gn(ci),gn(sij))gn(sij)

式中, l ( c i ) l(c_i) l(ci) l ( s i j ) l_(s_{ij}) l(sij)表示candidate和reference sentences的长度,10是为了将CIDEr-D的值和其它评价指标的值大小相似, σ = 6 \sigma=6 σ=6

那么CIDEr-D的计算方式为

C I D E r − D ( c i , S i ) = ∑ n = 1 N ω n C I D E r − D n ( c i , S i ) CIDEr-D(c_i, S_i)=\sum^N_{n=1}\omega_nCIDEr-D_n(c_i, S_i) CIDErD(ci,Si)=n=1NωnCIDErDn(ci,Si)

其中 ω n = 1 / N \omega_n=1/N ωn=1/N

猜你喜欢

转载自blog.csdn.net/liuxiaoheng1992/article/details/120939385