【论文】Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representations

Oct 20, 2020· · 1 min read

论文题目:Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representations

作者:Francesco Locatello, Stefan Bauer, Mario Lucic, Gunnar Rätsch, Sylvain Gelly, Bernhard Schölkopf, Olivier Bachem

会议/时间:ICML2019

链接:https://arxiv.org/pdf/1811.12359.pdf

论文目标

近年来关于disentangled representation 的unsupervised learning相关的研究非常多,其核心观点为现实世界中的数据是通过一些单独的可解释的元素生成的,可以通过无监督学习的方式学习到数据的一种表示。本文通过理论分析,证明在模型和数据不添加归纳偏置的情况下这种无监督学习是不可能实现的,同时文中在多个数据集上训练了近年来的相关模型。最终提出了进一步研究的方向。

相关工作

目前效果比较好的模型大多是基于变分自编码器(Variational Autoencoder,VAE),即认为现实世界中的数据来源于一个高维隐变量zP(z)z \sim P(z),实际观察到的数据为xx,通过建立一个自编码器,包括编码器和解码器学习得到P(xz)P(x|z)Q(zx)Q(z|x),其中使用Q(zx)Q(z|x)近似实际的P(zx)P(z|x),从而从数据xx中学到一个表示r(x)r(x)去寻找隐变量zz

这一领域之前的研究包括独立成分分析(Independent Component Analysis,ICA)等。

本文思路/解决方案

给定如下的理论

For d>1d > 1 , let zPz \sim P denote any distribution which admits a density p(z)=i=1dp(zi)p(z) = \prod_{i=1}^{d}p(z_i). Then, there exists an infinite family of bijective functions f : supp(z) → supp(z) such that fi(u)uj\frac {\partial f_i(u)}{\partial u_j} almost everywhere for all ii and jj (i.e., zz and f(z)f(z) are completely entangled) and P(zu)=P(f(z)u)P(z \leq u) = P(f(z) \leq u) for all usupp(z)u \in supp(z) (i.e., they have the same marginal distribution).

可以得到,对于给定的数据xx,可以找到无限多的p(z)p(z)满足要求,且这些p(z)p(z)相互等价并满足disentangled的要求,一个无监督的模型难以区分这些p(z)p(z)。因此在实际的模型中,需要为模型结构和数据集引入合理的归纳偏置(inductive bias)。

在实验设计中,使用到的模型为添加了正则项的VAE,包括betaVAE、AnnealedVAE、FactorVAE、beta-TCVAE、DIP-VAE等。在模型的度量标准中,使用了包括BetaVAE Metric、FactorVAE Metric、MIG(Mutual Information Gap)、Modularity、SAP Score等方法。使用到的数据集包括四个3D空间变化的数据集和三个随机的噪声数据集。

在归纳偏置中,为了控制变量,对于所有的模型使用相同的卷积结构、优化算法、超参数,使用相同的Gaussian Encoder和Bernoulli Decoder,使用相同的隐变量维度(=10),构建模型进行测试。仅使用不同的正则项和不同的正则项权重。

结果

  1. 当正则化系数增大时,使用拟合高斯的采样得到的相关性减小,使用均值表示的相关性增大。可以证明上述模型可以得到不同维度相关性比较弱的聚合先验,但是并不能表示均值表示也是不相关的。
  2. 在不同的性能度量中,除了Modularity以外的几种都是相关的,只是在不同的数据集上的相关性有差异。
  3. 模型的性能受到随机初始化和超参数的影响比较大,目标函数在其中影响比较小。
  4. 如何选择无监督模型仍然等待得到解决。在不同数据集和度量指标之间的超参数迁移用处不大。
  5. 实验并不能证明这些disentangled表示对于下游的任务有所帮助。

总结

在将来的研究中可以将重点更多放在如下三个方面:1)归纳偏置和显/隐性监督,2)disentangled representation的实际效益,3)实验设置和数据集的多样性。