Self-supervised Representation Learning from Videos for Facial Action Unit Detection(CVPR 2019)

Self-supervised Representation Learning from Videos for Facial Action Unit Detection(CVPR 2019)
Paper PDF

Introduction

Facial Action -> Functional -> usual methods -> problem -> proposed method

人脸作为人类之间交流所面对的人体器官，可以显示出无数种信息。为了全面的研究面部的运动模式，Ekman和Friesen开发了面部动作编码系统(FACS)，它定义了一套独特的约40个原子级别的面部肌肉动作，称为动作单元（Action Units, AUs）。通过对人脸面部运动单元的检测，我们可以从中获取大量与人内心活动有关的信息，并具有广泛的应用场景。比如，表情识别、健康评估、人机交互等等。通常对AUs的检测算法按照类型可以分为以下三种：

1. Descriptor：
2. Supervised：
3. Self-supervised：

由于给视频中人脸打面部运动单元标签的任务困难，导致在AUs检测方面的数据集极为缺乏，小样本的监督学习也容易造成过拟合。对此，本篇论文提出基于检测运动信号的自监督学习方式。考虑到不同帧中人物头部的差异由面部的AUs变化和头部姿势的改变两者共同造成，本文提出一种Twin-Cycle Autoencoder（ACAE）来同时学习头部姿势相关和面部运动单元相关的运动信息，并以此来解耦两者，达到提取discriminative的AUs表征。具体来讲，TCAE通过预测AUs相关和头部姿势相关的运动，将源帧的AUs和头姿态转换为目标帧的AUs和头部姿态。

---

Innovation

1. 用像素移动来表示AU以及头部姿态的运动，并以这种运用信号，来进行自监督的学习。具体见##Cycle with AU/pose changed
2. 对AU和头部的运动进行解耦，获得更加‘纯粹’的AU表征。

---

Method

整个网络架构如上图所示，可以大致分为以下三个模块：

Feature disentangling

TCAE首先对源人脸图像和目标人脸图像使用编码器分别获得各自的AU特征（${\varphi }_s,{\varphi }_t$）和头部姿势特征（${\psi }_s,{\psi }_t$）。TCAE将${\varphi }_s,{\varphi }_t$级联在一起输入AU-related解码器$D_A$，$D_A$用来解码AUs是如何从源脸变换为目标脸。同理，将${\psi }_s,{\psi }_t$级联在一起输入pose-related解码器$D_P$，$D_P$用来解码头部姿势是如何从源脸变换为目标脸。

论文中将这种变换定义为像素的移动。比如AUs造成的变换可以定义为一个矩阵${\tau }^A\in {\mathbb{R}}^{W\times H\times 2}$，其中$W,H$分别代表图像的宽和高。${\tau }_{xy}^A=({\delta }_x,{\delta }_y)$，其中${\delta }_x,{\delta }_y$分别表示在位置$(x,y)$上的像素在x方向和y方向的位移量。因此，${\tau }^A$作为一种变换，将原始人脸改变为AU-changed的人脸，即$I_s\mapsto I_A$。

与之相似，我们从pose-related解码器$D_P$从获得与头部姿态相关的像素的移动${\tau }^P$。${\tau }^P$作为一种变换，将原始人脸改变为pose-changed的人脸，即$I_s\mapsto I_P$。

为了区分AU和位姿变化引起的位移，我们在${\tau }^A$上加入L₁正则化，以保持AU-related运动的稀疏性和精微性，如式所示：
$${\mathfrak{L}}_1^A=\sum _{x,y}\Vert {\tau }_{xy}^A{\Vert }_1\text{\hspace{1em}(1)}$$

Target reconstruction

TCAE通过线性组合的方式来整合由${\tau }^A$和${\tau }^P$造成的帧间差异。
$$\begin{array}{rl}& {\tau }_{xy}={\alpha }_{xy}^A{\tau }_{xy}^A+{\alpha }_{xy}^P{\tau }_{xy}^A\\ s.t.& \\ & {\alpha }_{xy}^A+{\alpha }_{xy}^P=1\end{array}$$

$\tau$作为变换，将源人脸映射为预测的目标人脸。因此，我们需要限制变换后的人脸与目标人脸的相似度，如式2所示：
$${\mathfrak{L}}_{rec}=\Vert \tau (I_s)-I_t{\Vert }_1\text{\hspace{1em}(2)}$$
其中，$I_s,I_t$分别为源人脸图片和目标人脸图片。

Cycle with AU/pose changed

在 AU-changed 的循环中:
给定输入$({\varphi }_s{\varphi }_t)$，$D_A$解码到由$I_s$到$I_A$的变换${\tau }^A$。这个变换具体为一个($W\times H\times 3$)矩阵，包含每个像素点的mask和偏移量。基于变换${\tau }^A$，我们可以得到AU-changed人脸$I_A={\tau }^A(I_s)$。并定义$I_A$经过编码器得到AU相关特征${\varphi }_s^A$，头部姿态相关特征${\psi }_s^A$。

那么，给定输入$({\varphi }_s{\varphi }_s^A)$，$D_A$可以解码为到由$I_A$到$I_s$的变换${\tau }^{-A}$。如果我们在AU-changed人脸$I_A$上使用变换${\tau }^{-A}$，得到的人脸图像${\tau }^{-A}(I_A)={\tau }^{-A}({\tau }^A(I_s))$一定和源人脸$I_s$相似。因此，得到另一约束条件，如公式3所示：
$${\mathfrak{L}}_{cyc}^A=\Vert {\tau }^{-A}({\tau }^A(I_s))-I_s{\Vert }_1\text{\hspace{1em}(3)}$$
除此之外，我们还要考虑特征级别的一致性。考虑到图像$I_s$经历AU相关的变换得到图像$I_A$，因此$I_s$和$I_A$在头部姿态特征上应该具有相似性，$I_t$和$I_A$在AU特征上应该具有相似性，因此我们得到公式5：
$${\mathfrak{L}}_{emb}^A=\Vert {\psi }_s^A-{\psi }_s{\Vert }^2+\Vert {\varphi }_s^A-{\varphi }_t{\Vert }^2\text{\hspace{1em}(5)}$$

pose-changed的循环中:
给定输入$({\psi }_s{\psi }_t)$，$D_P$解码到由$I_s$到$I_P$的变换${\tau }^P$。基于变换${\tau }^P$，我们可以得到pose-changed人脸$I_P={\tau }^P(I_s)$。并定义$I_P$经过编码器得到AU相关特征${\varphi }_s^P$，头部姿态相关特征${\psi }_s^P$。

那么，给定输入$({\psi }_s{\psi }_s^A)$，$D_P$可以解码为到由$I_P$到$I_s$的变换${\tau }^{-P}$。如果我们在pose-changed人脸$I_P$上使用变换${\tau }^{-P}$，得到的人脸图像${\tau }^{-P}(I_P)={\tau }^{-P}({\tau }^P(I_s))$一定和源人脸$I_s$相似。因此，得到另一约束条件，如公式4所示：
$${\mathfrak{L}}_{cyc}^P=\Vert {\tau }^{-P}({\tau }^P(I_s))-I_s{\Vert }_1\text{\hspace{1em}(4)}$$
除此之外，我们还要考虑特征级别的一致性。考虑到图像$I_s$经历头部姿态相关的变换得到图像$I_P$，因此$I_s$和$I_P$在AU特征上应该具有相似性，$I_t$和$I_P$在头部姿态特征上应该具有相似性，因此我们得到公式6：
$${\mathfrak{L}}_{emb}^P=\Vert {\varphi }_s^P-{\varphi }_s{\Vert }^2+\Vert {\psi }_s^P-{\psi }_t{\Vert }^2\text{\hspace{1em}(6)}$$

---

Experiment

Implementation details Detailed

编解码器的设计如下图所示：

Loss 函数如公式7所示
$$\mathfrak{L}=\frac{1}{W\times H\times 3}{\mathfrak{L}}_{rec}+\frac{{\lambda }_1}{W\times H\times 2}{\mathfrak{L}}_1^A+\frac{{\lambda }_2}{W\times H\times 3}({\mathfrak{L}}_{cyc}^A+{\mathfrak{L}}_{cyc}^P)+\frac{{\lambda }_3}{256}({\mathfrak{L}}_{emb}^A+{\mathfrak{L}}_{emb}^P)\text{\hspace{1em}(7)}$$

训练的过程中采用了Curriculum Learning的方式，即难度递增的策略进行训练。

Comparisons with other methods

表中数值均为F1-score：

Analysis