《PROTOTYPICAL CONTRASTIVE LEARNING OF UNSUPERVISED REPRESENTATIONS》学习笔记

引言

本文提出了一种无监督表征学习方法，主要针对对比损失(contrastive loss) 和聚类方法(deep cluster)方法改进，对比损失主要基于噪声对比估计(noise contrastive estimator)，将同分布的embedding 拉近，不同分布的embedding距离推远。这些方法的主要问题是对比损失针对的实例判别任务只需浅层特征就可以处理，因为实验表明在10epochs内就可以达到很高精度，而继续训练只提供少量信息。而且提取的特征在下游任务中表现不好。所以实例对比学习到的特征缺少高层语义信息。另外对比损失需要构造大量负样本，而与正样本相似的负样本会损害表征学习。对于基于聚类的无监督学习方法，其结构如图1，重复 聚类-分类训练 的步骤，但cnn提取的特征还需要先pca降维，而且最后的分类器需要频繁初始化。

本文针对这些问题，结合InfoNCE 和EM（最大期望）算法改进deepcluster方法。聚类时，直接使用backbone的特征，将聚类中心作为一组相似样本的prototype（原型）特征，图2中蓝绿实心小圆，希望提取到的原型特征能包含一类样本共有的语义特征。如图2，实例对比损失的方法会将同实例的训练样本靠近，而本文的方法是将样本往聚类中心靠近。

方法

目标函数使用似然函数，使用$\theta$参数描述$p(x_i;\theta )$，选择最优参数使得似然函数最大（已有样本发生可能性最大）。即公式$1$：
$${\theta }^{\ast }=\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\log p\left(x_i;\theta \right)\text{\hspace{1em}(1)}$$
引入隐变量$c_i$，表示样本属于哪个原型特征对应的聚类中。公式1变为下式：
$${\theta }^{\ast }=\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\log p\left(x_i;\theta \right)=\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\log \sum _{c_i\in C}p\left(x_i,c_i;\theta \right)\text{\hspace{1em}(2)}$$
对于等式右边的项利用凹函数log的jensen不等式可以确定其下界
$$\sum _{i=1}^n\log \sum _{c_i\in C}p\left(x_i,c_i;\theta \right)=\sum _{i=1}^n\log \sum _{c_i\in C}Q\left(c_i\right)\frac{p\left(x_i,c_i;\theta \right)}{Q\left(c_i\right)}\ge \sum _{i=1}^n\sum _{c_i\in C}Q\left(c_i\right)\log \frac{p\left(x_i,c_i;\theta \right)}{Q\left(c_i\right)}\text{\hspace{1em}(3)}$$
通过最大化下界来求解参数，求解过程即EM算法过程，当$Q(c_i)$满足下式时下界最大，参考PCL。
$$Q\left(c_i\right)=p\left(c_i;x_i,\theta \right)\text{\hspace{1em}(4)}$$
因为$Q(c_i)$与$\theta$无关，所以${\sum }_{i=1}^n{\sum }_{c_i\in C}Q\left(c_i\right)\log Q\left(c_i\right)$为常数
$$\begin{gathered} {\theta }^{\ast }=\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\log p\left(x_i;\theta \right)=\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\sum _{c_i\in C}Q\left(c_i\right)\log p\left(x_i,c_i;\theta \right)-const \\ =\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\sum _{c_i\in C}Q\left(c_i\right)\log p\left(x_i,c_i;\theta \right) \\ =\underset{\theta }{\mathrm{argmax}}\sum _{i=1}^n\sum _{c_i\in C}p\left(c_i;x_i,\theta \right)\log p\left(x_i,c_i;\theta \right)\text{\hspace{1em}(5)} \end{gathered}$$
现在问题是如何知道$Q(c_i)=p\left(c_i;x_i,\theta \right)$与$p\left(x_i,c_i;\theta \right)$，在聚类完成后，我们就可以知道每个样本与哪个原型相关，属于哪个聚类中。用指示函数给出每个样本与原型的关系。
$$p\left(c_i;x_i,\theta \right)=1\left(x_i\in c_i\right)\text{\hspace{1em}(6)}$$
对于$p\left(x_i,c_i;\theta \right)$，使用概率公式转换，$p\left(c_i;\theta \right)$使用均匀分布。
$$p\left(x_i,c_i;\theta \right)=p\left(x_i;c_i,\theta \right)p\left(c_i;\theta \right)=\frac{1}{k}\cdot p\left(x_i;c_i,\theta \right)\text{\hspace{1em}(7)}$$
$p\left(x_i;c_i,\theta \right)$采用高斯分布
$$p\left(x_i;c_i,\theta \right)=\exp \left(\frac{-{\left(v_i-c_s\right)}^2}{2{\sigma }_s^2}\right)/\sum _{j=1}^k\exp \left(\frac{-{\left(v_i-c_j\right)}^2}{2{\sigma }_j^2}\right)\text{\hspace{1em}(8)}$$
综合上面公式，最后目标函数为：
$${\theta }^{\ast }=\underset{\theta }{\mathrm{argmin}}\sum _{i=1}^n-\log \frac{\exp \left(v_i\cdot c_s/{\varphi }_s\right)}{{\sum }_{j=1}^k\exp \left(v_i\cdot c_j/{\varphi }_j\right)}\text{\hspace{1em}(9)}$$
因为要聚类，不知道需要多少个聚类数，所以聚类M次，每个聚类数为$k_m$，然后取平均，同时加上InfoNCE的Loss，最后总损失函数如下：
$${\mathcal{L}}_{\text{ProtoNCE }}=\sum _{i=1}^n-\left(\log \frac{\exp \left(v_i\cdot v_i^{\prime }/\tau \right)}{{\sum }_{j=0}^r\exp \left(v_i\cdot v_j^{\prime }/\tau \right)}+\frac{1}{M}\sum _{m=1}^M\log \frac{\exp \left(v_i\cdot c_s^m/{\varphi }_s^m\right)}{{\sum }_{j=0}^r\exp \left(v_i\cdot c_j^m/{\varphi }_j^m\right)}\right)\text{\hspace{1em}(10)}$$
其中InfoNCE loss为
$${\mathcal{L}}_{\text{InfoNCE }}=\sum _{i=1}^n-\log \frac{\exp \left(v_i\cdot v_i^{\prime }/\tau \right)}{{\sum }_{j=0}^r\exp \left(v_i\cdot v_j^{\prime }/\tau \right)}\text{\hspace{1em}(11)}$$ 参考Noise Contrastive Estimation和Contrastive Predictive Coding，注意里面构造的正负样本关系。
损失函数中有个$\varphi$参数，类似于温度系数，当聚类样本数越多越集中，$\varphi$越小，这样样本间的相似度会放大，因此，紧密的簇（$\varphi$ 值小）相似度增大，松散的簇（$\varphi$值大）相似度降低。同温度系数改变Loss，产生更平衡的cluster且其集中程度（concentration）相似。

实验

基于学习的特征在有限的微调下做后续任务

Low-shot classification

每个类别只有k个样本时的分类任务。ImageNet上无监督表征学习，在少样本中训练svm分类器。指标为mAP。参数设置同MoCo方法。

特征可视化

在ImageNet无监督训练完后，使用t-SNE工具可视化特征分布。对比MoCo方法

消融实验

分析infoNCE，warm-up策略，ProtoNCE的效果