Data-Free Knowledge Amalgamation via Group-Stack Dual-GAN

1. Goal

The goal is to perform Data-free Knowledge distillation.

Knowledge distillation: Dealing with the problem of training a smaller model (Student) from a high capacity source model (Teacher) so as to retain most of its performance.

As the word itself, We perform knowledge distillation when there is no original dataset on which the Teacher network has been trained. It is because, in real world, most datasets are proprietary and not shared publicly due to privacy or confidentiality concerns.

To tackle this problem, it is necessary to reconstruct a dataset for training Student network. Thus, in this paper, the authors propose a data-free knowledge amalgamate strategy to craft a well-behaved multi-task student network from multiple single/multi-task teachers.

For this, the main idea is to construct the group-stack generative adversarial networks (GANs) which have two dual generators. First, one generator is trained to collect the knowledge by reconstructing the images approximating the original dataset. Then, a dual generator is trained by taking the output from the former generator as input. Finally, we treat the dual part generator as the TargetNet (Student network) and regroup it.

The architecture of Dual-GAN is shown in Fig 0.

2. Problem Definiton

In Data-Free Knowledge Amalgamation via Group-Stack Dual-GAN, The authors aim to explore a more effective approach to train the student network (TargetNet), only utilizing the knowledge amalgamated from the pre-trained teachers. The TargetNet is designed to deal with multiple tasks and learns a customized multi-branch network that can recognize all labels selected from separate teachers.

• The number of the customized categories: $C$
• Label vector: $Y_{cst}=\{y_1,\cdots ,y_C\}\subseteq \{0,1{\}}^C$
• TargetNet: $\mathcal{T}$
  • Handling multiple tasks on the $Y_{cst}$
• Pre-trained teachers: $\mathcal{A}=\{{\mathcal{A}}_1,\cdots ,{\mathcal{A}}_M\}$
• For each teacher $m$ , a $T_m$ -label classification task :$Y_m=\{y_m^1,\cdots ,y_m^{T_m}\}$
• Feature maps in the $b$-th block of the $m$-th pre-trained teacher: $F_m^b$

The teacher networks are in the constraint: $Y_{cst}\subseteq \bigcup _{m=1}^M{Y}_m$, which reveals that either the full or the subset of classification labels is an alternative for making up the customized task set.

3. Method

The process of obtaining the well-behaved TargetNet with the proposed data-free framework contains three steps.

1. The generator $G(z):z\to \mathcal{I}$ is trained with knowledge amalgamation in the adversarial way, where the images in the same distribution of the original dataset can be manufactured. Note that $z$ is the random noise and $\mathcal{I}$ denotes the image.
2. The dual generator $\mathcal{T}(\mathcal{I}):\mathcal{I}\to Y_{cst}$ is trained with generated samples from $G$ in the block-wise way to produce multiple predict labels. Note that $Y_{cst}$ is the predicted labels.
3. After training the whole dual-GAN, The dual-generator is modified as TargetNet for classifying the customized label set $Y_{cst}$

The overall training procedure of group-stack GAN is depicted in Fig 1.

3.1 Amalgamating GAN

First, introduce the arbitrary vanilla GAN. The GAN performs a minmax game between a generator $G(z):z\to \mathcal{I}$ and a discriminator $D(x):\mathcal{I}\to [0,1]$, and the objective function can be defined as:

$$L_{GAN}={\mathbb{E}}_x[log\mathcal{D}(\mathcal{I})]+\mathbb{E}[log(1-D(G(z)))].\cdots Eq.(1)$$

However, since the absence of real data, we can not perform training by Eq. (1). Then, several modification have been made as follows.

3.1.1 Group-stack GAN

The first modification is the group-stack architecture. The generator is designed to generate not only synthesized images but also the intermediated activations aligned with the teachers.

Thus, we set $B$ as the total group number of generators, which is the same as the block numbers of the teacher and the student networks. In this way, the generator can be denoted as a stack of $B$ groups $\{G^1,\cdots ,G^B\}$, from which both the image ${\mathcal{I}}_g$ and the consequent activation $F_g^j$ at group $j$ are synthesized from a random noise $z$:

$$\begin{array}{l}{F}_g^1=G^1(z)\\ F_g^j=G^j(F^{\{}j-1{\}}_g)1<j\le B,\end{array}\cdots Eq.(2)$$

when $j=B$, the output of the $B$-th group $G^B$ is $F_g^B$, which is also thought as the final generated image ${\mathcal{I}}_g$.

Since both the architecture of generator $G^j$ and discriminator $D^j$ are symmetrical, the group adversarial pair is presented as $[\{G^1,D^1\},\cdots ,\{G^B,D^B\}]$. In this way, the satisfying $G^{j\ast }$ can be acquired by:

$$G^{j\ast }=argmi{n}_{\{}{G}^j\}{\mathbb{E}}_z[log(1-D^{j\ast }(G^j(F_g^{j-1})))]\cdots Eq.(3)$$

where $1\le j\le B$ and $D^{j\ast }$ is the optimal $j$-group discriminator. We transfer the discriminator to designing a plausible loss function to calculate the difference between the generated samples and the real ones. Thus, we take the teacher network $\mathcal{A}$ to constitute each $D^j$:

$$D^j\leftarrow \bigcup _{i=1}^j\{{\mathcal{A}}^{B-j+i}\}\cdots Eq.(4)$$

During the training for the group pair $\{G^j,D^j\}$, only $G^j$ is optimized with discriminator $D^j$ is fixed, whose output is for classifying multiple labels. We make use of several losses to constraint the output of $D^j$ to motivate the real data's response.

The output for $D$ is $\mathcal{O}(F_g)=\{y^1,\cdots ,y^C\}$, with the predict label as $t^i$:

$$t^i=\{\begin{array}{ll}1& y^i\ge ϵ,\\ 0& y^i<ϵ,\end{array}\cdots Eq.(5)$$

where $1\le i\le C$ and $ϵ$ is set to be 0.5 in the experiment. Then the one-hot loss function can be defined as:

$$L_{oh}=\frac{1}{C}\sum _{i}l(y^i,t^i),\cdots Eq.(6)$$

where $l$ is the cross-entropy and $L_{oh}$ enforces the outputs of the generated samples to be close to one-hot vectors.

In addition, the outputs need to be sparse, since an image in the real world can't be tagged with dense labels which are the descriptions for different situations. So, an extra discrete loss function is proposed:

$$L_{dis}=\frac{1}{C}\sum _i|y^i|,\cdots Eq.(7)$$

which is known as L1-norm loss function. Finally, combining all the losses, the final objective function can be obtained:

$$L_{gan}=L_{oh}+\alpha L_a+\beta L_{ie}+\gamma L_{dis},\cdots Eq.(8)$$

where $\alpha ,\beta$ and $\gamma$ are the hyperparameters for balancing different loss items. $L_a$ and $L_{ie}$ are the activation loss and information entropy loss functions and described in the paper "Data-Free Learning of Student Networks"[H. Chen et al., 2019].

3.1.2 Multiple Targets

Since the TargetNet is customized to perform multi-label classifications learning from multiple teachers, the generator should generate samples containing multiple targets. As a result, for the $j$-th group generator $G^j$, authors construct multiple group-stack discriminators $\{D_1^j,\cdots ,D_M^j\}$ in concert with teachers specializing in different task sets by Eq. (4).

 

In order to amalgamate multi-knowledge into the generator by $\{D_m^j{\}}_{m=1}^M$, the teacher-level filtering is applied to $F_g^j$. And this filtering is conducted as:

$$F_g^{f,m}=f_m^j(F_g^j),\cdots Eq.(9)$$

where the filtering function $f_m^j$ is realized by a light learnable module consisting of a global pooling layer and two fully connected layers. $F_g^{j,m}$ is the filtered generated features that approach the output feature distribution of ${\mathcal{A}}_m^{B-j}$. The output of discriminator is donated as ${\mathcal{O}}_m(F_g^j)=D_m^j(F_g^{j,m}$.

The for the generated features $F_g^j$ from $G^j$, we collect from the multi-discriminator the $M$ prediction sets $\{{\mathcal{O}}_1(F_g^j),\cdots ,{\mathcal{O}}_M(F_g^j)\}$, which are:

$${\mathcal{O}}_g(F_g^j)=\bigcup _m^M{\mathcal{O}}_m(F_g^j),\cdots Eq.(10)$$

which is treated as new input to the loss Eq. 8, then $L_{gan}^i$ is the adversarial loss for each $G^j$. Despite the fact that the generated features should appear like the ones extracted
from the real data, they should also lead to the same predictions from the same input span style="color:DodgerBlue">$z$. Thus, the stack-generator $\{G^1,\cdots ,G^B\}$ can be jointly optimized by the final loss:

$$L_{joint}=L_{gan}^B+\frac{1}{B-1}\sum _{j=1}^{B-1}l({\mathcal{O}}_g(F_g^j),{\mathcal{O}}_g({\mathcal{I}}_g)),\cdots Eq.(11)$$

where the adversarial loss $L_{gan}$ only calculates from the last group $\{G^B,D^B\}$. The rest part of the loss is the cross-entropy loss that restrains the intermediate features generated from $G^1$ to $G^{B-1}$ to make the same prediction as ${\mathcal{I}}_g$, which offsets the adversarial loss $\{L_{gan}^1,\cdots ,L_{gan}^{B-1}\}$.

By minimizing $L_{joint}$, the optimal generator $G$ can synthesis the images that have similar activations as the real data fed to the teacher.

3.2 Dual-generator Training

After training the group-stack GAN, a set of generated samples are obtained, which are in the form of $\mathcal{R}=\{F_g^1,\cdots ,F_g^{B-1}\}\cup \{{\mathcal{I}}_g\}$ including both the generated intermediate features and the RGB images. Then, the next step is to train the dual-generator $\mathcal{T}$. Also, the group-discriminator for each ${\mathcal{T}}^b$ is constructed as:

$$D_d^{b,m}\to \bigcup _i^{B-b}\{{\mathcal{A}}_m^{b+i}\},\cdots Eq.(12)$$

where the dual generator $\mathcal{T}$ is trained to synthesis the samples that $D_d$ can't distinguished from ${\mathcal{I}}_g$ produced by the trained $G$.

We divide dual-generator into $B$ blocks as $\{{\mathcal{T}}^1,\cdots ,{\mathcal{T}}^B\}$, and during the training process of ${\mathcal{T}}^b(1<b\le B)$, the whole generator $G$ and the blocks of the dual-generator $\mathcal{T}$ from $1$ to $b-1$ keep fixed.

Taking training ${\mathcal{T}}^b$ for example, the problem is to learn features $F_u^b$ generated by the $b$-th block of the dual-generator. As shown in Fig 5, the generated $F_u^b$ is treated as 'fake' to the discriminator with ${\mathcal{I}}_g$ as the 'real' data in this dual part GAN. To do this, a two-level filtering method is applied to $F_u^b$.

1. Teacher-level filtering for multi-target demand $f_m$
   • Transforming $F_u^b$ to $m$ teacher streams $F_u^{b,m}=f_m^b(F_u^b)$, as defined in Eq. (9).
2. Task-level filtering $g_m$
   • Conducting after the last few fully connected layers of the corresponding discriminator, which is established for the constraint of $Y_{cst}\subseteq \bigcup _m^M{Y}_m$.

And, the authors feed the generated features $F_u^b$ to the corresponding discriminator $D_d^{b,m}$, and derive the predictions ${\mathcal{O}}_m(F_u^b)$. Then, the task-level filtering $g_m$ is applied to meet the task customizable demand ( $Y_{cst}\subseteq \bigcup _m^M{Y}_m$ )and embedded into the $m$-th branch block-wise adversarial loss:

$$L_d^{b,m}=l(g_m({\mathcal{O}}_m(F_u^b),g_m({\mathcal{O}}_m({\mathcal{I}}_g)),\cdots Eq.(13)$$

So, the block-wise loss for updating dual-generator ${\mathcal{T}}^b$ from multiple branches can be defined as:

$$L_d^b=\sum _m{\lambda }_m\cdot L_{\{}^{b,m}d\},\cdots Eq.(14)$$

where $\lambda =1$ for $m\in \{1,\cdots ,M\}$. Since the existence of the links between the generators $G$ and $\mathcal{T}$, the input has two streams:

$$\begin{array}{l}{F}_i^1={\mathcal{T}}^{b-1}{\mathcal{T}}^{b-2}\cdots {\mathcal{T}}^1({\mathcal{I}}_g),\\ F_i^2=G^{B+1-b}{G}^{B-b}\cdots G^1(z).\end{array}\cdots Eq.(15)$$

Then according to the different inputs to ${\mathcal{T}}^b$, the final loss can be rewritten from Eq. (14) as:

$$L_u^b={\lambda }_i^1{L}_d^{b,m}(F_i^1)+{\lambda }_i^2{L}_d^{b,m}(F_i^2),\cdots Eq.(16)$$

Reference

Ye, Jingwen, et al. "Data-free knowledge amalgamation via group-stack dual-gan." Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2020.