论文笔记：Strong-Weak Distribution Alignment for Adaptive Object Detection

论文：Strong-Weak Distribution Alignment for Adaptive Object Detection

地址：https://arxiv.org/abs/1812.04798

一、简介

这篇文章目的在于解决无监督域适应目标检测(Unsupervised Domain Adaptation for Object Detection), 之前state of the art 是 2018年CVPR的 Domain Adaptive Faster R-CNN for Object Detection in the Wild, 对应论文笔记：DA Faster RCNN。

Motivation

在图像分类中无监督域适应方法中，大多数是尽量使得目标域数据和源域数据完全匹配。作者认为，在目标检测任务中，由于不同的域有着不同的场景布局和目标组合，完全匹配目标域和数据域的分布将降低模型的 performance。但是，由于 low-level 局部特征匹配不会改变原有的类别语义信息，所以可以将它们进行强匹配。基于上面两个原因，作者提出了：

1. 核心 contribution 弱对齐模型(weak alignment model) : 使用对齐 loss 更多的强调整体相似的样本对齐，不强调整体不相似的样本对齐。
2. 强 域对齐 ： 加强对齐feature map的局部感受野，如纹理、颜色等

具体如 Figure 1 所示，学习域不变的特征，局部特征上的强对齐，全局场景上的弱对齐：

Figure 1

二、方法

Figure 3

整体框架基于 Faster-RCNN， 在 RPN 之前抽取全局特征，局部特征来自特征抽取器中更低的层。

2.1 Weak Global Feature Alignment

Figure 2

作者使用一个域分类器去对齐目标域和源域全局特征，在特征空间中，容易区分的目标样本距离源域样本非常远，而难以区分的目标样本距离源域样本很近，如 Figure 2 左边图所示。作者提出训练域分类器去忽略容易区分的样本，而去强调难以区分的样本。
为了完成上面的这个目标，作者提出使用Focal loss，如 Figure 2 右边图所示。交叉熵的问题在于对于很容易分类的样本同样会产生不可忽视的 loss 值，而 Focal loss 通过修改交叉熵引入一个修改因子 $f(p_t)$使得对于容易区分的样本，产生一个极小的loss。 修改后的交叉熵为：

$$-f(p_t)log(p_t)$$

其中 $p_t$ 为：
$$p_{\mathrm{t}}=\{\begin{array}{ll}p& \text{ if }d=1\\ 1-p& \text{ otherwise }\end{array}$$
$d=1$ 代表该样本属于目标域，$p$ 为对于的概率。作者选择一个函数 $f$ 使得随着 $p_t$ 增加，$f$值减小。

$$\mathrm{F}\mathrm{L}\left(p_{\mathrm{t}}\right)=-f\left(p_{\mathrm{t}}\right)\log \left(p_{\mathrm{t}}\right),f\left(p_{\mathrm{t}}\right)={\left(1-p_{\mathrm{t}}\right)}^{\gamma }$$
其中 $\gamma$ 是一个难区分类的权重值。

整个弱对齐域分类器 loss 定义如下：
$${\mathcal{L}}_{{\text{global}}_s}=-\frac{1}{n_s}\sum _{i=1}^{n_s}(1-D_g{\left(F\left(x_i^s\right)\right)}^{\gamma }\log \left(D_g\left(F\left(x_i^s\right)\right)\right)$$
$${\mathcal{L}}_{globa{l}_t}=-\frac{1}{n_t}\sum _{i=1}^{n_t}{D}_g{\left(F\left(x_i^t\right)\right)}^{\gamma }\log \left(1-D_g\left(F\left(x_i^t\right)\right)\right)$$
$${\mathcal{L}}_{global}\left(F,D_g\right)=\frac{1}{2}\left({\mathcal{L}}_{globa{l}_s}+{\mathcal{L}}_{globa{l}_t}\right)$$
$n_s$ 为 source 样本数量，$n_t$ 为 target 样本数量。

除此之外，整个 Faster RCNN 部分的 loss 可以总结为：
$${\mathcal{L}}_{cls}(F,R)=-\frac{1}{n_s}\sum _{i=1}^{n_s}{\mathcal{L}}_{det}\left(R\left(F\left(x_i^s\right)\right),y_i^s\right)$$

2.2 Strong Local Feature Alignment

作者将整个特征提取模块 $F$ 一分为 2 ，分别为 $F_1$ 和 $F_2$, 使用低层的 $F_1$ 提取的特征作为 local feature 域分类器的输入，输出一个和输入同样宽高的 feature map ，使用最小平方 loss 来训练域分类器，定义如下：

$${\mathcal{L}}_{lo{c}_s}=\frac{1}{n_{s}HW}\sum _{i=1}^{n_s}\sum _{w=1}^W\sum _{h=1}^H{D}_l{\left(F_1\left(x_i^s\right)\right)}_{wh}^2$$

$${\mathcal{L}}_{lo{c}_t}=\frac{1}{n_{t}HW}\sum _{i=1}^{n_t}\sum _{w=1}^W\sum _{h=1}^H{\left(1-D_l{\left(F_1\left(x_i^t\right)\right)}_{wh}\right)}^2$$

$${\mathcal{L}}_{loc}\left(F,D_l\right)=\frac{1}{2}\left({\mathcal{L}}_{lo{c}_s}+{\mathcal{L}}_{lo{c}_t}\right)$$

3.3 Context Vector based Regularization

主要是从两个域分类器的中间层抽取特征，然后根据不同的区域去跟RPN的输出对应(region-wise)进行拼接，作者称之为 “context” ，用来预测类别和bounding box。作者认为这是一个正则项，有助于稳定对抗训练。但我现在还不知道为什么：）

3.4 Overall Objective

所有对抗 loss 定义如下：
$${\mathcal{L}}_{adv}(F,D)={\mathcal{L}}_{loc}\left(F_1,D_l\right)+{\mathcal{L}}_{global}\left(F,D_g\right)$$

整个目标函数定义为最大最小化loss：

$$\underset{D}{\max }\underset{F,R}{\min }{\mathcal{L}}_{cls}(F,R)-\lambda {\mathcal{L}}_{adv}(F,D)$$

三、实验

3.1 不相似的域之间的适应

作者选择的源域数据集为 VOC， 目标域数据集为一些艺术图片，Watercolor、Clipart，结果如 Table 1 所示。

Table 1

其中 Source Only 指的是仅仅使用训练集图片训练，然后直接在目标域进行测试。BDC-Faster 是使用了域分类器的baseline，DA-Faster 是前面提到的2018年CVPR的Domain Adaptive Faster RCNN。

主要证明了如下几点：

1. 仅仅使用focal loss 就使得 map 上升了 10.8%(25.6 to 36.4)
2. context 向量(CTX)和局部特征对齐(L) 都是有效果的，进一步提高了MAP
3. 与 Source Only比较，BDC-Faster 和 DA-Faster 使得 performance 巨降，证明了在两个不同的域之间进行严格的特征对齐会扰乱目标检测任务的训练，而作者提出的弱对齐不会降低性能。

3.2 相似的域之间的适应

作者用了 Cityscrapes 到 Foggy Cityscrapes，效果如 Table 4 所示：

Table 4

这两个数据集的差别很小，而且目标域数据集Foggy Cityscrapes 来自源域数据集 Cityscrapes，只是在Cityscrapes 数据集上加入了一些 foggy 噪声，所以仅仅一个强的局部对齐使得MAP有极大的提高。

3.3 Adaptation from synthetic to real images

选择合成的数据SIM10K为源数据域，Cityscrapes为目标域，结果如 Table 5所示：

Table 4

主要证明一下几点：

1. 弱对齐的效果不只是有focal loss可以达成，其他和focal loss一样的如指数focal loss(EFL $f(p_t)=e^{\eta p_t}$) 也有效果
2. context vector 是有正则的效果，将其全部置为 0，基本不会影响结果。
3. DA-Faster 的 Instance-level 对齐没有什么作用，反而会降低模型的performance。

其中 P 指的是使用 CycleGAN 将源数据转化为目标数据风格后拿去训练。

发现一个问题：

• 根据实验结果，似乎强局部对齐分类器(L)的效果没有使用 CycleGAN 生成的数据来训练效果好。还不如直接拿 CycleGAN 生成的图片训练，而不使用强局部对齐分类器(L)。特别是在 $FL(\gamma =3)$ 的时候，有 P 和 没有 P的实验效果相差太大。