Semi-supervised Disentanglement with Independent Vector Variational Autoencoders

2020·Arxiv

Abstract

Abstract

We aim to separate the generative factors of data into two latent vectors in a variational autoencoder. One vector captures class factors relevant to target classification tasks, while the other vector captures style factors relevant to the remaining information. To learn the discrete class features, we introduce supervision using a small amount of labeled data, which can simply yet effectively reduce the effort required for hyperparameter tuning performed in existing unsupervised methods. Furthermore, we introduce a learning objective to encourage statistical independence between the vectors. We show that (i) this vector independence term exists within the result obtained on decomposing the evidence lower bound with multiple latent vectors, and (ii) encouraging such independence along with reducing the total correlation within the vectors enhances disentanglement performance. Experiments conducted on several image datasets demonstrate that the disentanglement achieved via our method can improve classifica-tion performance and generation controllability.

1. Introduction

A desirably disentangled representation contains individual units, each corresponding to a single generative factor of data while being invariant to changes in other units (Bengio et al., 2013). Such interpretable and invariant properties lead to benefits in downstream tasks including image classifica-tion and generation (Kingma et al., 2014; Makhzani et al., 2016; Narayanaswamy et al., 2017; Zheng & Sun, 2019).

Variational autoencoders (VAEs) (Kingma & Welling, 2014) have been actively utilized for unsupervised disentanglement learning (Kim & Mnih, 2018; Chen et al., 2018; Es- maeili et al., 2019; Kumar et al., 2018; Gao et al., 2019;

Figure 1. Proposed method with discrete and continuous random vectors, y and z. Weak classification supervision helps y capture discrete class factors. Our vector independence objective forces y and z to capture different information. Reducing this objective along with the TC improves disentanglement.

Burgess et al., 2017). To capture the generative factors that are assumed to be statistically independent, many studies have encouraged the independence of latent variables within a representation (Higgins et al., 2017; Kim & Mnih, 2018; Chen et al., 2018). Despite their promising results, the usage of only continuous variables frequently causes difficulty in the discovery of discrete factors (e.g., object categories).

To address this issue, researchers have utilized discrete variables together with continuous variables to separately capture discrete and continuous factors (Dupont, 2018; Kingma et al., 2014) and trained their models by maximizing the evidence lower bound (ELBO). In this paper, we show that the ability of disentanglement in their models is derived from not only using the two types of variables but also encouraging several sources of disentanglement. We expose these sources by decomposing the ELBO.

Unsupervised learning with discrete and continuous variables is difficult, because continuous units with a large informational capacity often store all information, causing discrete units to store nothing and be neglected by models (Dupont, 2018). Previously, this issue was mostly solved using hyperparameter-sensitive capacity controls (Dupont, 2018) or additional steps for inferring discrete features during training (Jeong & Song, 2019). In contrast, we simply inject weak supervision with a few class labels and effectively resolve the difficulty in learning of discrete variables. Locatello et al. (2019) also suggested the exploitation of available supervision for improving disentanglement.

We introduce a semi-supervised disentanglement learning method, which is shown in Figure 1. A VAE was used to extract two feature vectors: one containing discrete variables to capture class factors and the other vector containing continuous variables to capture style factors. The contributions of this work to the relevant field of study are as follows:

• We introduce the vector independence objective that measures the statistical independence between latent vectors in VAEs and enables different vectors to store different information. We name our model an Independent Vector VAE (IV-VAE).

• We decompose the ELBO containing multiple latent vectors to reveal the vector independence term along with well-known total correlation (TC) terms that measure independence between the variables within the vectors. We show that these terms are the sources of disentanglement in jointly learning discrete and continuous units.

• We introduce supervision with a small number of class labels, significantly reducing the difficulty in learning discrete class features.

• We empirically show that our method enhances disentanglement learning on several image datasets, positively affecting the classification and controllable generation.

The supplementary material includes derivation details and additional results, and the sections are indicated with S (e.g., Section S1).

2. Related Work: Disentanglement Learning with VAEs

A J-dimensional random vector, z of scalar-valued random variables, . Previous methods have encouraged independence between variables, whereas our method encourages independence between vectors (by introducing another vector, y) along with independence between the variables within the vectors.

2.1. Promoting Independence of Continuous Variables

Given dataset containing N i.i.d. samples of random vector x, a VAE learns latent vector z involved in the data generation process. The VAE consists of encoder and decoder trained by maximizing the ELBO on , where the em- pirical distribution is represented as x. The training objective consists of the reconstruction term and the KL divergence from the prior to individual posteriors:

where ) represent the vanilla VAE and Higgins et al., 2017) objectives, respectively. Under factorized prior enhances the independence of latent variables, leading to disentanglement.

To understand the disentangling mechanism in VAEs, the KL term (2) was decomposed into (3) in (Hoffman & Johnson, 2016) and further into (4) under the factorized prior in (Chen et al., 2018):

where is the aggregate posterior (Makhzani et al., 2016) that describes the latent structure for all data samples, DKLis the mutual information (MI) between the data and latent vectors under empirical distribution , and is the regularization term for penalizing individual latent dimensions that excessively differ from their priors.

The second term in (4) represents the TC (Watanabe, 1960) that measures the statistical dependency of more than two random variables. Kim & Mnih (2018) and Chen et al. (2018) argued that (i) -VAEs heavily penalize the MI term, , along with the other terms, causing z to be less informative about x, and (ii) the TC term is the source of disentanglement and thus should be strongly penalized. To address these issues, Kim & Mnih (2018) proposed the FactorVAE by adding the TC term to the vanilla VAE objective, and Chen et al. (2018) proposed the -TCVAE by separately controlling the three terms in (4) using individual weights.

The previous methods (Higgins et al., 2017; Kim & Mnih, 2018; Chen et al., 2018) learned disentangled representations by encouraging the independence of continuous variables. They primarily employed Gaussian distributions for variational posteriors and consequently focused on modeling continuous generative factors. However, most datasets naturally contain discrete factors of variation (e.g., object classes), which these methods frequently fail to capture. We address this issue by incorporating discrete random variables and further enhance the disentanglement by promoting the independence between one set of discrete variables and another set of continuous variables.

2.2. Utilizing Discrete and Continuous Variables

To separately capture discrete and continuous factors, re- searchers have proposed the simultaneous utilization of discrete and continuous variables, which are stored in two latent vectors, y and z, respectively. By introducing joint prior p(y, z), approximate posterior , and likelihood , the ELBO containing the reconstruction and KL terms becomes (see Section S1 for the derivation)

By assuming factorized prior p(y, z) = p(y)p(z), the KL can be decomposed depending on the factorized form of ; Kingma et al. (2014) assumed , and Dupont (2018) assumed

In contrast, we decompose the KL term (6) to reveal the existence of the vector independence term between y and z as well as the TC terms within the vectors (see our decomposition in (11)). Moreover, our method explicitly encourages the vector independence term while not penalizing the latent-data MI terms, and , to obtain disentangled informative features. We show that our method outperforms the Higgins et al., 2017) and its variant (Dupont, 2018), which strongly penalize the KL term (7) and consequently minimize the latent-data MI terms.

In addition, unsupervised learning with discrete and continuous variables often causes the discrete units to capture less information and be discarded by the model (Dupont, 2018) because of a larger informational capacity of continuous units than that of discrete units. To address this issue, existing unsupervised methods involve sophisticated hyper-parameter tunings or additional computations. For example, Dupont (2018) modified (7) using capacity control terms separately for discrete and continuous units, and Jeong & Song (2019) proposed an alternating optimization between inferring probable discrete features and updating the encoder. In contrast, we introduce weak classification supervision to guide the encoder to store class factors in discrete units; this method simply but significantly reduces the effort needed for designing inference steps.

3. Proposed Approach

The problem scenario is identical to that described in Section 2.2, with discrete random vector y for capturing class factors and continuous random vector z for capturing style factors. First, we show that the ELBO can be decomposed into the proposed vector independence term and the others. Then, we present our semi-supervised learning (SSL) strategy.

3.1. Learning of Independent Latent Vectors

3.1.1. VECTOR INDEPENDENCE OBJECTIVE

Assuming the conditional independence of , an encoder produces the parameters for variational posteriors and , for the n-th sample, x. The aggregate posteriors that capture the entire latent space under data distribution q(x) are defined as

where is computed using its decomposition form.2 Then, we define our vector independence objective as the MI between the two vectors that measures their statistical dependency:

Thr reduction of this term can enforce y and z to capture different semantics. Here, we emphasize that our method is applicable to cases with multiple L latent vectors by extending (9) to DKL, which has a form similar to the TC computed over the variables (i.e., DKL) but is computed over the vectors. The relationship between the TC and the vector independence term is similar to that between the objectives of independent component analysis (ICA; Jutten & Herault (1991); Amari et al. (1996)) and independent vector analysis (IVA; Kim et al. (2006b;a)).

3.1.2. OUR ELBO DECOMPOSITION

Chen et al. (2018) showed that the TC term measuring the dependency between latent variables (i) exists in the decomposion of the ELBO containing a single latent vector, z, and (ii) is a source of disentanglement in VAEs. Similarly, we reveal that the vector independence term (i) exists in the decomposition of the ELBO containing two latent vectors, y and z, and (ii) is another source of disentanglement.

Concretely, we decompose the KL term (6) of the ELBO into (10) under p(y, z) = p(y)p(z) and further into (11) under

where is the MI between x and latent vectors y and z, and and are the dimension-wise regularization terms. The derivation from (10) to (11) is motivated by that from (3) to (4). See Section S2 for the derivation details.

As suggested by Kim & Mnih (2018) and Chen et al. (2018), data-latent MI is not penalized during training so as to allow the latent vectors to capture data information. The second term in the RHS is our vector independence objective, and the third and fifth terms are the TC terms for the variables in y and z, respectively. We empirically show that simultaneously reducing these three terms provides better disentanglement compared to penalizing only the TC terms without considering the vector independence. The regularization terms, , forbid individual latent variables from deviating largely from the priors.

A concurrent work (Esmaeili et al., 2019) introduced a decomposition similar to our result in (11). Their derivation was initiated by augmenting the ELBO with a data entropy term (i.e., ). In contrast to our method, their method with discrete and continuous units uses purely unsupervised learning, which often causes the discrete units to be ignored by the model.

3.1.3. RELATIONSHIP BETWEEN THE VECTOR INDEPENDENCE OBJECTIVE AND TC

Here, we investigate the relationship between vector inde- pendence objective ) and the following TC terms:

(12) measures the independence of the variables within each vector (hereafter called the “separate” TC). In addition, (13) simultaneously considers the variables in y and z (hereafter called the “collective” TC), and it can be viewed as the TC on concatenated vector h = [y; z] (i.e., DKL

We introduce two relationships. First, perfectly penalized collective TC indicates perfect vector independence:

by letting and . In this case, the vector independence objective would be naturally satisfied, resulting in unnecessary optimization. However, this case is rare because of the existence of other loss terms (e.g., a reconstruction term) that often prevent the collective TC from being zero. Furthermore, under the factorized prior, the perfectly penalized TC may be undesirable because it could imply the occurrence of posterior collapse (i.e., learning a trivial posterior that collapses to the prior and fails to capture data features). We present the experimental setup and results regarding this relationship in Sections S5 and S6.

Second, perfect vector independence does not ensure that all variables within and between the vectors are perfectly independent, i.e., . However, perfect vector independence ensures that the collective TC is the sum of the two separate TCs, i.e., (see Section S3 for the derivation).

3.2. Semi-supervised Learning (SSL)

A weak classification supervision guides y to suitably represent discrete class factors. In addition, our vector independence objective further enforces y and z to capture different types of information. In our experiments, we simplify the problem setup by assuming that a given dataset involves a single classification task with C classes. This enabled us to design y as a single categorical variable, , where K = 1.3 We represent y as a C-dimensional one-hot vector. For z , we assume the existence of multiple style factors and expect each factor to be captured by each variable,

The training image dataset consists of labeled set L = , where the n-th image, x, is paired with the corresponding class label, {1, ..., C}, and unlabeled set . Here, and are the numbers of samples in datasets L and U, respectively, and . The empirical data distributions over L and U are denoted by and , respectively.

3.2.1. SEMI-SUPERVISED LEARNING OBJECTIVE

To update encoder parameter and decoder parameter , the objectives for the labeled and unlabeled sets are given as

For the following reconstruction terms, we use true label t for L and inferred feature y for U. This strategy helps the decoder accurately recognize one-hot class vectors.

We compute the classification term for L as

where hyper-parameter controls the effect of discrimina-

tive learning. With scaling constant (Kingma et al., 2014; Maaløe et al., 2016).

Next, we introduce the following commonly used objective for L and U to learn disentangled features and regularize the encoder:

This is identical to applying individual loss weights to the vector independence, TC, and dimension-wise regularization terms in our KL decomposition (11). Here, the expectation over the empirical distribution, i.e., or , is included in computing the aggregated posteriors, . Note that disappears for a single class variable, y (i.e., K = 1 in (12)), and data-latent MI is removed so as to allow y and z to properly store the information about x.

The final optimization function is given as

4. Data and Experimental Settings

4.1. Data and Experimental Settings

We used the dSprites (Matthey et al., 2017), Fashion-MNIST (Xiao et al., 2017), and MNIST (LeCun et al., 2010) datasets. For SSL, the labeled data were selected to be distributed evenly across classes, and the entire training set was used for the unlabeled set. We prevented overfitting to training data in classification tasks by introducing validation data. For the dSprites dataset, we divided the images in a ratio of 10:1:1 for training, validation, and testing and tested two SSL setups with 2% and 0.25% labeled training data. For the Fashion-MNIST and MNIST datasets, we divided the training set in a ratio of 5:1 for training and validation while maintaining the original test set and tested the SSL setup with 2% labeled training data.

The architectures of the encoder and decoder were the same as the convolutional networks used in (Dupont, 2018; Jeong & Song, 2019). The priors were set as p(z) = N(0, I) and , where denotes evenly distributed class probabilities. We employed the Gumbel-Softmax distribution (Jang et al., 2017; Maddison et al., 2017) for reparametrizing categorical y. We trained networks with minibatch weighted sampling (Chen et al., 2018). Further details of experimental settings are described in Section S4.

We considered the vanilla VAE (Kingma & Welling, 2014), Higgins et al., 2017), Chen et al., 2018), and jointVAE (Dupont, 2018) as the baselines. For fair comparison, we augmented their original unsupervised objectives with the classification term in (19) and the same loss weight, . For the VAE, -VAE, and -TCVAE, we augmented continuous vector z in their original objectives with discrete variable y. Note that the main difference between the -TCVAE and our IV-VAE is the existence of the vector independence term in (20). We also removed the data-latent MI term from the -TCVAE objective, as applied in our objective. See Section S5 for the baseline details.

4.2. Performance Metrics

We measured the classification error with y to assess the amount of class information in y and the ELBO to examine the generative modeling ability. As the disentanglement score, we computed the MI gap (Chen et al., 2018) based on the empirical MI between latent variables and known generative factors: MIG , where M is the number of known factors and MIG

for each factor, , for quantifying the gap between the top two variables with the highest MI. Here, is the normalized MI between latent variable and factor , and it is theoretically bounded between 0 and 1. Additionally, . This vanilla MIG is denoted by MIGall. Figure 2 shows the example of the normalized MI computed on dSprites with different models, where the top two MI values for each factor are indicated.

The entangled results for the case where one variable captures both style and class factors (e.g., obtain fairly good MIGall scores. To alleviate this issue, we separately computed the MIG for the class and style factors as MIGclass (i.e., the MIG of the first row in Figure 2) and MIGstyle (i.e., the mean MIG averaged from the second to last rows), where t denotes ground truth class labels and S is the set of style factors. For the cases without known generative factors but with available labels, t, we also computed the MI that assesses

Figure 2. Examples of disentanglement scores (top right) and empirical mutual information (bottom) between the ground truth generative factors of dSprites (y-axis) and inferred features (x-axis). (a) Unsupervised ). (b) Semi-supervised (c) Semi-supervised IV-VAE (ours; ). The three networks were initiated from the same random seed. The dotted-line boxes indicate undesirable results, while the solid-line boxes indicate improved results. The SSL with weak class supervision helps y better capture discrete class factors (pink boxes). The vector independence objective helps y and z store different information, leading to better disentanglement (red boxes).

Figure 3. Comparison on dSprites under the unsupervised (black) and semi-supervised setups with 0.25% (red) and 2% (blue) labeled data. For each network setting, the mean and std scores from 7 random seeds are shown. The x-axis shows -TCVAE, and our IV-VAE trained with the same weight of ELBO KL or TC term (weight for vector independence varies as 1, 4, and 8. Unsupervised learning causes the class factor to be stored in the continuous z instead of the discrete y (i.e., high I(z; t) and low I(y; t)). Furthermore, the class information is captured in multiple variables, causing lower MIGclass than MIGstyle. Introducing weak supervision with a few labeled data effectively relieves this issue (i.e., low I(z; t) and high I(y; t)). Our vector independence objective allows y and z to capture different information, improving most of the scores. Keep in mind that the baselines were also trained under the same SSL setup using z and the discrete y.

how much y represents the class information and z does not as , respectively.

5. Experimental Results

All results presented in this section were obtained using test data. See our supplementary material for additional results.

5.1. Results on dSprites

5.1.1. EFFECT OF SSL AND VECTOR INDEPENDENCE

Figure 2 depicts the benefit of SSL and vector independence. Using purely unsupervised learning failed to capture the class factor with discrete variable y, but employing SSL with 2% class labels easily resolved this issue. Encouraging vector independence helped z better capture the style factors by forcing y and z to store different information.

Figure 3 shows the results on dSprites obtained under various SSL setups. The unsupervised setup yielded good vanilla MIGall scores, which were similar to those of the 2%-labeled setup. However, the unsupervised setup caused the continuous vector z to mostly captured the class infor- mation and the discrete y to fail to store it. This is evidenced by higher I(z; t) and lower I(y; t) than those of the SSL setups. Furthermore, as indicated by lower MIGclass scores, more than two variables undesirably captured the class factor under the unsupervised setup.

Injecting weak class supervision to the training process simply yet effectively alleviated this difficulty, as shown in the decreased I(z; t) and increased I(y; t) in Figure 3. In addition, our IV-VAEs with proper weights outperformed -VAEs and -TCVAEs for most scores. In particular, the score improvements with the 0.25% labels were larger than those with the 2% labels, indicating the benefit of vector independence under a few class labels. In terms of the ELBO, our IV-VAEs did not outperform -TCVAEs but showed less trade-off between density modeling and disentanglement than -VAE (i.e., higher ELBO and MIG scores).

5.1.2. RELATIONSHIPS BETWEEN VECTOR INDEPENDENCE AND EVALUATION METRICS

Figure 4 shows the relationships between the vector independence objective and evaluation metrics. In Figure 4-1 (left), we depicted scatter plots, where each circle represents the

Figure 4. Relationship between the vector independence objective and the evaluation metrics. The x-axis shows vector independence KL , where lower KL values indicate stronger independence. The y-axis shows (a) ELBO, (b) MIGall, (c) MIGclass, (d) MIGstyle, (e) classification error, (f) I(y; t), and (g)I(z; t) obtained under the SSL setups with 0.25% and 2% labeled data (denoted by SSL L0.25% and SSL L2%, respectively). Lower is better for (e) and (g), and higher is better for the other metrics. In Figure 4-1 (left), each circle shows the median score of seven networks initiated from different random seeds. As shown in the legend, different colors indicate different values for controlling the effect of TC in training, and bigger circles indicate bigger values for causing stronger vector independence. In Figure 4-2 (right), the different settings are further merged to analyze the overall tendency, and each circle shows the median score of 21 networks (i.e., 7 random seeds settings). Given each setting, the best performing network is indicated with text. As shown in Figure 4-1, given the value of , IV-VAEs often outperform -TCVAEs. As shown in Figure 4-2, IV-VAEs with the of 1 for the 0.25% SSL setup and those with the of 4 for the 2% SSL setup generally work well.

median score of seven networks trained with the same loss weights but initiated from different random seeds. In Figure 4-2 (right), the different values were further integrated to analyze the general result trends. Notice that in the fig-ures, a larger led to a smaller KL of vector independence, i.e., a stronger independence between y and z. The analyses are summarized below. See Section S6 for the additional analyses with extended baselines.

• ELBO in Figure 4-1(a). The ELBO values of IV-VAEs were similar to those of -TCVAEs for lower 0.5 and 1. Increasing caused slightly decreased ELBO values, because the models focused more on learning disentangled features than maximizing the reconstruction term. Nevertheless, under higher values of 4 and 8, IV-VAEs yielded better ELBO than did not penalize the latent-data MI term, which led to an eased trade-off between reconstruction and disentanglement.

• MIGall, MIGclass, and MIGstyle in Figure 4-1(b), (c), and (d). For most of the setups, IV-VAEs with the values of 0.5, 1, or 4 achieved better MIG scores than -TCVAEs, showing that reducing the vector independence objective along with the TC helps disentanglement. Higher that heavily penalized the TC often led to higher MIG scores (i.e., the blue and green lines showed better scores than the red lines). The optimal yielding the highest MIG differed depending on the value of , but the of 0.5, 1, and 4 generally worked well.

• Classification error and I(y; t) in Figure 4-1(e) and (f). Given the value of , the lowest classification error and highest I(y; t) score were often obtained with IVVAE, in comparison to -TCVAE. This result implies that vector independence encourages y to better capture the class factor. In the SSL setup with 0.25% labels, excessively increasing values harmed the classification performance.

• I(z; t) in Figure 4-1(g). IV-VAEs often achieved lower I(z; t) than that of -TCVAEs. This result implies that vector independence prevents z from storing the class factor by enforcing z and y to represent different information, leading to better disentanglement.

• Overall result trends in Figure 4-2. Our IV-VAEs outperformed -TCVAEs for most scores, demonstrating the benefit of vector independence. In general, the of 1 for the 0.25% setup and the of 4 for the 2% setup worked well.

Figure 5. Comparison on Fashion-MNIST with 2% labeled data. For each network setting, the mean and std scores from 10 random seeds are shown. The horizontal lines indicate the mean performance of vanilla VAE and jointVAE. The x-axis shows TCVAE, and our IV-VAEs. Our IV-VAEs outperform the baselines for most scores, demonstrating the benefit of vector independence. Keep in mind that the baselines were also trained under the same SSL setup using z and the discrete y.

Figure 6. Comparison on MNIST with 2% labeled data. The settings and results are similar to those of Figure 5.

5.2. Results on Fashion-MNIST and MNIST

Figure 5 shows the results on Fashion-MNIST with 2% labeled data. Because of the absence of ground truth style factors, we only measured the MI between the latent variables and class labels. For most values of , promoting vector independence with greater ture different factors, causing more class information to be stored in y. Our IV-VAEs achieved better classification errors and I(y; t) as well as lower I(z; t) than all the baselines, indicating improved disentanglement. Figure 6 shows the results on MNIST with 2% labeled data. We also observed the benefit of our vector independence under most of the weight settings.

Figures 7 and 8 depict the qualitative results of the two

Figure 7. Qualitative results on Fashion-MNIST. (a) Corrected clas-sification. Given an input (left), inferred class probabilities (middle) or one-hot labels (right) from the class encoding y are used for reconstruction. Below each example, inferred labels without and with the vector independence loss are indicated. (b) Style-conditional generation. The style vector z is extracted from an input (leftmost), and the y is set as desired item labels (right). Without promoting vector independence, the network often fails to generate Trouser (the 2nd column in the generation results), Sneaker (the 8th), and Bag (the 9th) classes. Encouraging vector independence forces y to better capture class information in (a) and suitably separates styles from clothing classes in (b).

Figure 8. Qualitative results on MNIST. The analysis is similar to that of Figure 7. (a) Corrected classification. (b) Style-conditional generation. In (b), the absolute differences of pixel values are also shown. Encouraging vector independence better exhibits the digit styles (e.g., writing angle, line width, and shape width) by helping z effectively capture style information.

networks trained without and with the vector independence objective. The networks were obtained from the same random seed and loss weights except the See Section S7 and S8 for additional examples.

To visually show the corrected class labels by encouraging vector independence, Figures 7(a) and 8(a) depict reconstruction examples with estimated class information, i.e., inferred class probabilities or one-hot labels via the argmax operation. The results obtained from the class probabilities were often blurry, indicating that the inputs were confusing to be classified. Employing vector independence often corrects the classification results by enforcing y to better store the class factor.

Figures 7(b) and 8(b) depict the generation examples given input styles. We extracted z from the input and set y as each of the one-hot class labels. Enhancing vector independence better displays the fashion and digit styles by helping z effectively capture style information.

6. Conclusion

We have proposed an approach for semi-supervised disentanglement learning with a variational autoencoder. In our method, two latent vectors separately capture class and style factors. To boost disentanglement, we have proposed the vector independence objective that enforces the vectors to be statistically independent. We have revealed that, along with the total correlation term, our vector independence term is another source of disentanglement in the evidence lower bound. Furthermore, the difficulty in the learning of discrete factors can be reduced by exploiting a small number of class labels. The experiments on the dSprites, Fashion-MNIST, and MNIST datasets have confirmed the effectiveness of our method for image classification and generation.

Acknowledgement

We thank Prof. Sung Ju Hwang (KAIST) for providing his valuable comments. This work was supported by an Institute for Information and Communications Technology Promotion (IITP) grant funded by the Korean Government (MSIT) under grant no. 2016-0-00562 (R0124-16-0002).

References

Amari, S.-i., Cichocki, A., and Yang, H. H. A new learning algorithm for blind signal separation. In Proc. NIPS, pp. 757–763, 1996.

5In Figure 7, the of 4 was used for training the IV-VAE, while

the of 2 were commonly used for both

networks. In Figure 8, the of 8 was used, while the

of 1, and of 2 were the same for both networks.

Bengio, Y., Courville, A., and Vincent, P. Representation learning: A review and new perspectives. IEEE Trans. Pattern Anal. Mach. Intell., 35(8):1798–1828, 2013.

Burgess, C. P., Higgins, I., Pal, A., Matthey, L., Watters, N., Desjardins, G., and Lerchner, A. Understanding disentangling in beta-vae. In Proc. NIPS Workshop, 2017.

Chen, T. Q., Li, X., Grosse, R. B., and Duvenaud, D. K. Isolating sources of disentanglement in variational autoencoders. In Proc. NIPS, pp. 2610–2620, 2018.

Dupont, E. Learning disentangled joint continuous and discrete representations. In Proc. NIPS, pp. 710–720, 2018.

Esmaeili, B., Wu, H., Jain, S., Bozkurt, A., Siddharth, N., Paige, B., Brooks, D. H., Dy, J., and van de Meent, J.-W. Structured disentangled representations. In Proc. AISTATS, pp. 2525–2534, 2019.

Gao, S., Brekelmans, R., Steeg, G. V., and Galstyan, A. Auto-encoding total correlation explanation. In Proc. AISTATS, pp. 1157–1166, 2019.

Higgins, I., Matthey, L., Pal, A., Burgess, C., Glorot, X., Botvinick, M., Mohamed, S., and Lerchner, A. beta-vae: Learning basic visual concepts with a constrained variational framework. In Proc. ICLR, 2017.

Hoffman, M. D. and Johnson, M. J. Elbo surgery: yet another way to carve up the variational evidence lower bound. In Proc. NIPS Workshop, 2016.

Jang, E., Gu, S., and Poole, B. Categorical reparameteriza- tion with gumbel-softmax. In Proc. ICLR, 2017.

Jeong, Y. and Song, H. O. Learning discrete and continuous factors of data via alternating disentanglement. In Proc. ICML, pp. 3091–3099, 2019.

Jutten, C. and Herault, J. Blind separation of sources, part i: An adaptive algorithm based on neuromimetic architecture. Signal Processing, 24(1):1–10, 1991.

Kim, H. and Mnih, A. Disentangling by factorising. In Proc. ICML, pp. 2649–2658, 2018.

Kim, T., Attias, H. T., Lee, S.-Y., and Lee, T.-W. Blind source separation exploiting higher-order frequency dependencies. IEEE Trans. Audio, Speech, Language Process., 15(1):70–79, 2006a.

Kim, T., Eltoft, T., and Lee, T.-W. Independent vector analysis: An extension of ica to multivariate components. In Proc. Int. Conf. on ICA, pp. 165–172, 2006b.

Kingma, D. P. and Welling, M. Auto-encoding variational bayes. In Proc. ICLR, 2014.

Kingma, D. P., Mohamed, S., Rezende, D. J., and Welling, M. Semi-supervised learning with deep generative models. In Proc. NIPS, pp. 3581–3589, 2014.

Kumar, A., Sattigeri, P., and Balakrishnan, A. Variational inference of disentangled latent concepts from unlabeled observations. In Proc. ICLR, 2018.

LeCun, Y., Cortes, C., and Burges, C. Mnist handwritten digit database. http://yann.lecun.com/exdb/mnist/, 2010.

Locatello, F., Bauer, S., Lucic, M., Raetsch, G., Gelly, S., Sch¨olkopf, B., and Bachem, O. Challenging common assumptions in the unsupervised learning of disentangled representations. In Proc. ICML, pp. 4114–4124, 2019.

Maaløe, L., Sønderby, C. K., Sønderby, S. K., and Winther, O. Auxiliary deep generative models. In Proc. ICML, pp. 1445–1453, 2016.

Maddison, C. J., Mnih, A., and Teh, Y. W. The concrete distribution: A continuous relaxation of discrete random variables. In Proc. ICLR, 2017.

Makhzani, A., Shlens, J., Jaitly, N., Goodfellow, I., and Frey, B. Adversarial autoencoders. In Proc. ICLR Workshop, 2016.

Matthey, L., Higgins, I., Hassabis, D., and Lerchner, A. dsprites: Disentanglement testing sprites dataset. https://github.com/deepmind/dsprites-dataset/, 2017.

Narayanaswamy, S., Paige, T. B., Van de Meent, J.-W., Desmaison, A., Goodman, N., Kohli, P., Wood, F., and Torr, P. Learning disentangled representations with semi-supervised deep generative models. In Proc. NIPS, pp. 5925–5935, 2017.

Watanabe, S. Information theoretical analysis of multivari- ate correlation. IBM Journal of research and development, 4(1):66–82, 1960.

Xiao, H., Rasul, K., and Vollgraf, R. Fashion-mnist: a novel image dataset for benchmarking machine learning algorithms. In arXiv preprint arXiv:1708.07747, 2017.

Zheng, Z. and Sun, L. Disentangling latent space for vae by label relevant/irrelevant dimensions. In Proc. CVPR, pp. 12192–12201, 2019.

List of Figures

S1. (Derivation) Our decomposition of the KL term in the variational bound with two latent vectors S2. (Derivation) Relationship between the vector independence objective and the separate TC S3. Network architecture S4. Comparison of the collective TC and the separate TC S5. Corrected classification examples on Fashion-MNIST S6. Style-controlled generation on Fashion-MNIST S7. Style-controlled generation on Fashion-MNIST: continued from Figure S6 S8. Latent traversal examples on Fashion-MNIST S9. Latent traversal examples on Fashion-MNIST: continued from Figure S8

S10. Corrected classification examples on MNIST S11. Style-controlled generation on MNIST S12. Latent traversal examples on MNIST

S1. Variational Bound with Two Latent Vectors

Suppose that the generation process of the data involves two latent vectors, y and z, with a joint prior, p(y, z). By introducing an approximate posterior, , and a likelihood, , the ELBO for a single data sample becomes

S2. Our ELBO Decomposition

Figure S1 verifies our decomposition of the KL term, , in the ELBO on

S3. Relationships between Vector Independence and TC

Figure S2 verifies the following relationship: perfect vector independence ensures that the collective TC becomes the sum of the two separate TCs (i.e.,

Moreover, in this document, we provide the experimental results regarding the relationship described in (14) of the main paper: . We show that (i) a perfectly penalized collective TC (i.e., ) rarely occurs because of the existence of other loss terms (e.g., a reconstruction term) and (ii) encouraging vector independence under penalizing either the separate TC or the collective TC can improve disentanglement performance. See Section S5 for the experimental settings and Section S6 for the results.

S4. Data and Experimental Settings

We used the dSprites (Matthey et al., 2017), MNIST (LeCun et al., 2010), and Fashion-MNIST (Xiao et al., 2017) datasets. For semi-supervised learning, the labeled data were selected to be distributed evenly across classes. The size of labeled set L was either 2% or 0.25% of the entire training data, which we used for unlabeled set U. The classification loss weight, , was set as for the 2% and 0.25%-labeled dSprites setups. Here, for the 0.25% setup, we failed to apply scaling constant (which caused an extremely large ), and we instead used as applied in the 2% setup. We used (i.e., ) for the 2%-labeled MNIST. We used (i.e., ) for the 2%-labeled Fashion-MNIST.

Figure S3 shows the network architectures used in our experiments, which are identical to the convolutional architectures used in (Dupont, 2018; Jeong & Song, 2019). The priors were set as evenly distributed class probabilities. We employed the Gumbel-Softmax distribution (Jang et al., 2017; Maddison et al., 2017) for reparametrizing categorical y. We will release our experimental codes on GitHub.

S4.1. dSprites: 3-class Shape Classification

The dSprites dataset contains 2D-shape binary images with a size of 6464, which were synthetically generated with five independent factors: shape (3 classes; heart, oval, and square), position X (32 values), position Y (32), scale (6), and rotation (40). We divided the 737,280 images into training, validation, and test sets in a ratio of 10:1:1. We tested two cases with the 2% and 0.25% labels available in the training data.

One 6-dimensional isotropic Gaussian vector was used for , and one categorical variable representing 3 classes was used for . The Gumbel-Softmax temperature parameter was set as 0.75. The Adam optimizer was used with an initial learning rate of 0.001 and a minibatch size of 2048. Every batch contained 1024 labeled samples. We trained networks for 100 epochs and reported the results measured at the epoch of the best validation loss. We tested 7 different random weeds for network weight initialization.

Figure S1. Our decomposition of the KL term in the variational bound with two latent vectors. The terms for vector independence (highlighted in red) and variable independence (in green) are depicted.

Figure S2. Relationship between the vector independence objective and the separate TC: perfect vector independence ensures that the collective TC becomes the sum of the two separate TCs.

Figure S3. Network architecture for 6464 (left) and 3232 (right) images. The yellow and blue boxes indicate the layers of encoder and decoder , respectively. Each convolutional (C) or up-convolutional (uC) layer is identified by the size and number of filters. The stride size is 2 for the C and uC layers. Each fully-connected (Fc) layer is identified by the number of output neurons. The concatenation is denoted by “concat.” Each layer is followed by ReLU nonlinearity except that the last encoding layer and output layer are followed by linear and sigmoid activations, respectively. The dashed lines indicate the sampling operation for distributions

S4.2. MNIST: 10-class Digit Classification

The MNIST dataset contains 0–9 handwritten digit images with a size of 2828. We divided the original training set into 50,000 training and 10,000 validation images while maintaining the test set of 10,000 images. The images were normalized to have [0, 1] continuous values and resized to 3232 by following (Dupont, 2018). We tested the case with the 2% labels available in the training data.

One 10-dimensional isotropic Gaussian vector was used for , and one categorical variable representing 10 classes was used for . The Gumbel-Softmax temperature parameter was set as 0.67. The Adam optimizer was used with an initial learning rate of 0.001 and a minibatch size of 512. Every batch contained 256 labeled samples. We trained networks for 200 epochs and reported the results measured at the epoch of the best validation loss. We tested 10 different random seeds for network weight initialization.

S4.3. Fashion-MNIST: 10-class Fashion Item Classification

The Fashion-MNIST dataset (Xiao et al., 2017) contains grayscale images with a size of 2828 and 10 fashion categories (e.g., t-shirt, trouser, sandal, and bag). The experimental settings were the same as those used in the MNIST dataset, except that the Gumbel-Softmax temperature parameter was set as 0.75.

S5. Baseline Methods

We assumed that discrete random variable and continuous random vector z were involved in the data generation process. Under this scenario, we compared our IV-VAE with the vanilla VAE (Kingma & Welling, 2014), (Higgins et al., 2017), -TCVAE (Chen et al., 2018), and JointVAE (Dupont, 2018). For fair comparison, we applied the following settings that were used in our method to the baselines.

• We utilized the same dimensional latent units and network architectures. For the VAE, -VAE, and -TCVAE, we augmented the original continuous vector, z, in their objectives with discrete variable y. The jointVAE was originally designed to incorporate both of the continuous z and discrete y.

• We assumed the conditional independence of by following (Dupont, 2018). • We modified their original unsupervised setups to use weak classification supervision for semi-supervised learning (SSL). Concretely, as applied in our method, their objectives for labeled set L and unlabeled set U became

The reconstruction and classification terms were also identical to those of our method, as shown below. For the reconstruction terms, true label t for L and inferred feature y for U were used to enable the decoder to better recognize one-hot class vectors.

The methods differ in the definition of to learn disentangled features and regularize the encoder, as shown below. For notational brevity, we omit the dependence of objectives J and L on their parameters . The terms related to independence of latent units are indicated by colors (i.e., , and ). Note that we did not penalize the latent-data MI terms by setting in order to help y and z capture the information about x.

• VAE: the case with in the below term of •

• • • JointVAE: • IV-VAE-1 (ours): • IV-VAE-2 (ours):

where is the separate TC on z, is the collective TC that simultaneously considers y and the variables in z, is the vector independence term between y and z, and and are the dimension-wise regularization terms.

The term of is obtained by applying the KL decomposition of (4) in the main paper separately to and in the -VAE objective and further assigning individual loss weights to the decomposed terms. Notice that does not exist for the single-dimensional variable y (i.e., DKLwith K = 1 is zero). The term of is obtained by applying the KL decomposition of (4) in the main paper to -VAE objective, where the concatenation [y; z] is considered as a single latent vector, and further assigning individual loss weights to the decomposed terms. For fair comparison, we assigned and separately to and instead of assigning a single weight to . The term of JointVAE is identical to the KL term in the original JointVAE objective, where the channel capacity weights control the amount of information that y and z can capture. The is identical to (20) in the main paper. The is obtained by changing the TC term in the IV-VAE-1 objective from

The -TCVAE and IV-VAE in the main paper correspond to the -TCVAE-1 and IV-VAE-1. In this document, we will present the additional results of -TCVAE-2 and IV-VAE-2 along with -TCVAE-1 and IV-VAE-1 in Section S6. We will analyze (i) the relationships between the vector independence and evaluation metrics and (2) the effect of penalizing collective TC or separate TC under encouraging vector independence term

S6. In-depth Analysis of Relationships between Vector Independence and Evaluation Metrics

In the main paper, Figure 4 shows the results of -TCVAE-1 and IV-VAE-1 on the dSprites dataset. Here, Figure S4 shows the results of -TCVAE-2 and IV-VAE-2 (trained by penalizing the collective TC, ) that are merged to those of -TCVAE-1 and IV-VAE-1 (trained by penalizing the separate TC, ). The x-axis shows vector independence KL , where lower KL values indicate stronger independence. The y-axis shows (a) ELBO, (b) MIGall, (c) MIGclass, (d) MIGstyle, (e) classification error, (f) I(y; t), and (g)I(z; t) obtained under the SSL setups with 0.25% and 2% labeled data (denoted by SSL L0.25% and SSL L2%, respectively). Lower is better for (e) and (g), and higher is better for the other metrics. In Figure S4-1 (left), each circle shows the median score of seven networks initiated from different random seeds. As shown in the legend, different colors indicate different values for controlling the effect of TC in training, and bigger circles indicate bigger values for causing stronger vector independence. In Figure S4-2 (right), the different settings are further merged to analyze the overall tendency, and each circle shows the median score of 21 networks (i.e., 7 random seeds settings).

We compare the effect of the collective TC on disentanglement learning with that of the separate TC. We also show the benefit of encouraging vector independence under penalizing not only the separate TC but also the collective TC. The observations are summarized below.

• In Figure S4-1, given a larger value of , penalizing the collective TC for -TCVAE-2 and IV-VAE-2 yielded stronger vector independence than penalizing the separate TC for -TCVAE-1 and IV-VAE-1, e.g., the results for (colored by cyan in Figure S4-1) were placed to the left of those for (by blue), whereas the results for (by pink) and those for (by red) were similarly placed on the x-axis,. The cause of this phenomenon is that promoting independence between y and all the variables in z by decreasing the collective TC naturally encourages vector independence between y and z (e.g., independence between ensures independence between y and z ). Nevertheless, increasing for IV-VAE-2 under each setting often led to better vector independence than -TCVAE-2 (i.e., ). This phenomenon is caused by the imperfect minimization of the collective TC because of the existence of the other loss terms (e.g., the reconstruction term).

• In terms of the disentanglement scores, penalizing the separate TC on z was competitive with or sometimes outperformed penalizing the collective TC . For example, under the 0.25% SSL setup in Figure S4-2, the best scores of MIGall, MIGclass, MIGstyle, classification error, I(y; t), and I(z; t) obtained using the separate TC (text in orange) were better than those using the collective TC (text in purple). Rather than promoting multiple independences via the collective TC (e.g., , and ), promoting vector independence between y and the entire z (i.e., ) may guide the encoder better to separate discrete and continuous information. Under this guidance, penalizing the separate TC (e.g., ) may help each variable capture each continuous factor.

• In Figure S4-2, IV-VAE-2 networks outperformed or were competitive with -TCVAE-2 networks for most scores, demonstrating the benefit of vector independence even under penalizing the collective TC. In particular, the of 0.5 for the 0.25% SSL setup worked well, as shown in the purple text in Figure S4-2. Under the 2% SSL setup, the best performing IV-VAE-2 networks yielded similar results with those of the best -TCVAE-2 networks.

S7. Additional Results on Fashion-MNIST

Figures S5, S6, S7, S8, and S9 depict the qualitative results of the two networks trained on Fashion-MNIST without and with the vector independence objective (i.e., -TCVAE-1 and IV-VAE-1). The networks were obtained from the same random seed and loss weights except the weight: the of 4 was used for training the IV-VAE-1, while the of 2 were commonly used for both networks.

To visually show the corrected class labels by encouraging vector independence, Figure S5 depicts reconstruction examples with estimated class information, i.e., inferred class probabilities or one-hot labels via the argmax operation. The results obtained from the class probabilities were often blurry, indicating that the inputs were confusing to be classified. Employing vector independence often corrects the classification results by enforcing y to better store the class factor. Figures S6 and S7 depict generated fashion images given input clothing styles (e.g., brightness and width). We extracted z from the input and set y as each of the one-hot item labels. Without promoting vector independence, the generation of non-clothing items (e.g., sneaker and bag) frequently failed. Encouraging vector independence better disentangled styles from class information, improving the synthesis controllability. The latent traversals in Figures S8 and S9 show that the two dimensions of z captured the continuous factors corresponding to shapes and brightness.

S8. Additional Results on MNIST

Figures S10, S11, and S12 depict the qualitative results of the two networks trained on MNIST without and with the vector independence objective (i.e., -TCVAE-1 and IV-VAE-1). The networks were obtained from the same random seed and loss weights except the weight: the of 8 was used for training the IV-VAE-1, while the of 32, of 1, and of 2 were commonly used for both networks.

Figure S10 shows corrected classification examples, which are visualized using the reconstruction outputs with estimated class information. Figure S11 depicts generation examples of 0–9 digit images given input styles. Encouraging vector independence better displays the digit styles (e.g., writing angle and line width) by helping z effectively capture style information. In Figure S12, the latent traversals on z show the discovered continuous factors of variation.

Figure S4. Comparison of the collective TC and the separate TC. The -TCVAE-2 and IV-VAE-2 were trained by penalizing the collective TC and the variables in z, whereas the -TCVAE-1 and IV-VAE-1 were trained by penalizing the separate TC Given each setting, the best performing network is indicated with text. As shown in Figure S4-1, given a larger value of , penalizing the collective TC yields stronger vector independence than penalizing the separate TC. Nevertheless, because of the imperfect optimization of the collective TC, increasing for IV-VAE-2 under each setting often yields better vector independence than -TCVAE-2 (i.e., ). As shown in Figure S4-2, IV-VAE-2 outperforms or is competitive with -TCVAE-2 for most scores, demonstrating the benefit of vector independence even under penalizing the collective TC.

Figure S5. Corrected classification examples on Fashion-MNIST. Given an input (left), inferred class probabilities (middle; denoted by “soft”) or one-hot labels (right; denoted by “hard”) from the class encoding y are used for reconstruction. Below each example, inferred labels without and with the vector independence loss are indicated. Encouraging vector independence forces y to better capture class information, improving the classification performance.

Figure S6. Style-controlled generation on Fashion-MNIST. The style feature z is extracted from an input (leftmost), and the class feature y is set as desired item labels (right). Here, the style inputs correspond to top- and whole-body clothing items, and their class number and name are indicated (e.g., L6. Shirt). The order of generation follows the order of original label numbers. Without promoting vector independence, the network often fails to generate Trouser (the 2nd column in the generation results), Sneaker (the 8th), and Bag (the 9th) classes. Encouraging vector independence forces z to better capture style information by suitably separating styles from clothing classes.

Figure S7. Style-controlled generation on Fashion-MNIST: continued from Figure S6. The style feature z is extracted from an input (leftmost), and the class feature y is set as desired item labels (right). Here, the style inputs correspond to bottom clothing and non-clothing fashion items, and their class number and name are indicated (e.g., L1. Trouser). The order of generation follows the order of original label numbers. Without promoting vector independence, the generation with shoe styles frequently fails, as shown in the results of L5, L7, and L9. Encouraging vector independence forces z to better capture style information by suitably separating styles from clothing classes.

Figure S8. Latent traversal examples on Fashion-MNIST. The class feature y and style feature z are extracted from the given inputs (top-most), and the latent traversal results on a single dimension are depicted (bottom; traversal range is from -1 to 1). The models capture the continuous factor related to clothing shapes (e.g., width, sleeve length, heel height, and bag size).

Figure S9. Latent traversal examples on Fashion-MNIST: continued from Figure S8 . The class feature y and style feature z are extracted from the given inputs (top-most), and the latent traversal results on a single dimension are depicted (bottom; traversal range is from -1 to 1). The models capture the continuous factor related to brightness.

Figure S10. Corrected classification examples on MNIST. Given an input (left), inferred class probabilities (middle; denoted by “soft”) or one-hot labels (right; denoted by “hard”) from the class encoding y are used for reconstruction. Encouraging vector independence forces y to better capture class information, improving the classification performance.

Figure S11. Style-controlled generation on MNIST. The style feature z is extracted from an input (leftmost), and the class feature y is set as desired digit labels (right). The order of generation follows the order of original label numbers. Encouraging vector independence forces z to better capture digit styles such as (a) writing angle, (b) line width, and (c) shape width.

Figure S12. Latent traversal examples on MNIST. The class feature y and style feature z are extracted from the given inputs (top-most), and the latent traversal results on the three dimensions of z are depicted (bottom; traversal range is from -3 to 3). The continuous factors related to digit styles are stored in z.