Open main menu

An autoencoder is a type of artificial neural network used to learn efficient data codings in an unsupervised manner.[1][2] The aim of an autoencoder is to learn a representation (encoding) for a set of data, typically for dimensionality reduction, by training the network to ignore signal “noise”. Along with the reduction side, a reconstructing side is learnt, where the autoencoder tries to generate from the reduced encoding a representation as close as possible to its original input, hence its name. Several variants exist to the basic model, with the aim of forcing the learned representations of the input to assume useful properties.[3] Examples are the regularized autoencoders (Sparse, Denoising and Contractive autoencoders), proven effective in learning representations for subsequent classification tasks,[4] and Variational autoencoders, with their recent applications as generative models.[5]

Contents

IntroductionEdit

An autoencoder is a neural network that learns to copy its input to its output. It has an internal (hidden) layer that describes a code used to represent the input, and it is constituted by two main parts: an encoder that maps the input into the code, and a decoder that maps the code to a reconstruction of the original input.

Performing the copying task per se would be meaningless, and this is why usually autoencoders are restricted in ways that force them to reconstruct the input only approximately, prioritizing the most relevant aspects of the data to be copied.

The idea of autoencoders has been popular in the field of neural networks for decades, and the first applications date back to the '80s.[6][3][7] Their most traditional application was dimensionality reduction or feature learning, but more recently the autoencoder concept has become more widely used for learning generative models of data.[8][9] Some of the most powerful AIs in the 2010s involved sparse autoencoders stacked inside of deep neural networks.[10]

Basic ArchitectureEdit

 
Schema of a basic Autoencoder

The simplest form of an autoencoder is a feedforward, non-recurrent neural network similar to single layer perceptrons that participate in multilayer perceptrons (MLP) – having an input layer, an output layer and one or more hidden layers connecting them – where the output layer has the same number of nodes (neurons) as the input layer, and with the purpose of reconstructing its inputs (minimizing the difference between the input and the output) instead of predicting the target value   given inputs  . Therefore, autoencoders are unsupervised learning models (do not require labeled inputs to enable learning).

An autoencoder consists of two parts, the encoder and the decoder, which can be defined as transitions   and   such that:

 
 
 

In the simplest case, given one hidden layer, the encoder stage of an autoencoder takes the input   and maps it to  :

 

This image   is usually referred to as code, latent variables, or latent representation. Here,   is an element-wise activation function such as a sigmoid function or a rectified linear unit.   is a weight matrix and   is a bias vector. Weights and biases are usually initialized randomly, and then updated iteratively during training through Backpropagation. After that, the decoder stage of the autoencoder maps   to the reconstruction   of the same shape as  :

 

where   for the decoder may be unrelated to the corresponding   for the encoder.

Autoencoders are trained to minimise reconstruction errors (such as squared errors), often referred to as the "loss":

 

where   is usually averaged over some input training set.

As mentioned before, the training of an autoencoder is performed through Backpropagation of the error, just like a regular feedforward neural network.

Should the feature space   have lower dimensionality than the input space  , the feature vector   can be regarded as a compressed representation of the input  . This is the case of undercomplete autoencoders. If the hidden layers are larger than (overcomplete autoencoders), or equal to, the input layer, or the hidden units are given enough capacity, an autoencoder can potentially learn the identity function and become useless. However, experimental results have shown that autoencoders might still learn useful features in these cases.[11] In the ideal setting, one should be able to tailor the code dimension and the model capacity on the basis of the complexity of the data distribution to be modeled. One way to do so, is to exploit the model variants known as Regularized Autoencoders.[3]

VariationsEdit

Regularized AutoencodersEdit

Various techniques exist to prevent autoencoders from learning the identity function and to improve their ability to capture important information and learn richer representations.

Sparse autoencoder (SAE)Edit

 
Simple schema of a single-layer sparse autoencoder. The hidden nodes in bright yellow are activated, while the light yellow ones are inactive. The activation depends on the input.

Recently, it has been observed that when representations are learnt in a way that encourages sparsity, improved performance is obtained on classification tasks.[12] Sparse autoencoder may include more (rather than fewer) hidden units than inputs, but only a small number of the hidden units are allowed to be active at once.[10] This sparsity constraint forces the model to respond to the unique statistical features of the input data used for training.

Specifically, a sparse autoencoder is an autoencoder whose training criterion involves a sparsity penalty  on the code layer  .

 

Recalling that  , the penalty encourages the model to activate (i.e. output value close to 1) some specific areas of the network on the basis of the input data, while forcing all other neurons to be inactive (i.e. to have an output value close to 0).[13]

This sparsity of activation can be achieved by formulating the penalty terms in different ways.

 

be the average activation of the hidden unit   (averaged over the   training examples). Note that the notation  makes explicit what was the input affecting the activation, i.e. it identifies which input value the activation is function of. To encourage most of the neurons to be inactive, we would like  to be as close to 0 as possible. Therefore, this method enforces the constraint   where  is the sparsity parameter, a value close to zero, leading the activation of the hidden units to be mostly zero as well. The penalty term  will then take a form that penalizes  for deviating significantly from  , exploiting the KL divergence:

  where   is summing over the   hidden nodes in the hidden layer, and  is the KL-divergence between a Bernoulli random variable with mean  and a Bernoulli random variable with mean  .[13]

  • Another way to achieve sparsity in the activation of the hidden unit, is by applying L1 or L2 regularization terms on the activation, scaled by a certain parameter  .[16] For instance, in the case of L1 the loss function would become

 

  • A further proposed strategy to force sparsity in the model is that of manually zeroing all but the strongest hidden unit activations (k-sparse autoencoder).[17] The k-sparse autoencoder is based on a linear autoencoder (i.e. with linear activation function) and tied weights. The identification of the strongest activations can be achieved by sorting the activities and keeping only the first k values, or by using ReLU hidden units with thresholds that are adaptively adjusted until the k larges activities are identified. This selection acts like the previously mentioned regularization terms in that it prevents the model from reconstructing the input using too many neurons.[17]

Denoising autoencoder (DAE)Edit

Differently from sparse autoencoders or undercomplete autoencoders that constrain representation, Denoising autoencoders (DAE) try to achieve a good representation by changing the reconstruction criterion.[3]

Indeed, DAEs take a partially corrupted input and are trained to recover the original undistorted input. In practice, the objective of denoising autoencoders is that of cleaning the corrupted input, or denoising. Two underlying assumptions are inherent to this approach:

  • Higher level representations are relatively stable and robust to the corruption of the input;
  • To perform denoising well, the model needs to extract features that capture useful structure in the distribution of the input.[4]

In other words, denoising is advocated as a training criterion for learning to extract useful features that will constitute better higher level representations of the input.[4]

The training process of a DAE works as follows:

  • The initial input  is corrupted into   through stochastic mapping  .
  • The corrupted input   is then mapped to a hidden representation with the same process of the standard autoencoder,  .
  • From the hidden representation the model reconstructs  .[4]

The model's parameters  and  are trained to minimize the average reconstruction error over the training data, specifically, minimizing the difference between  and the original uncorrupted input  .[4] Note that each time a random example  is presented to the model, a new corrupted version is generated stochastically on the basis of  .

The above mentioned training process could be developed with any kind of corruption process. Some examples might be additive isotropic Gaussian noise, Masking noise (a fraction of the input chosen at random for each example is forced to 0) or Salt-and-pepper noise (a fraction of the input chosen at random for each example is set to its minimum or maximum value with uniform probability).[4]

Finally, notice that the corruption of the input is performed only during the training phase of the DAE. Once the model has learnt the optimal parameters, in order to extract the representations from the original data no corruption is added.

Contractive autoencoder (CAE)Edit

Contractive autoencoder adds an explicit regularizer in their objective function that forces the model to learn a function that is robust to slight variations of input values. This regularizer corresponds to the Frobenius norm of the Jacobian matrix of the encoder activations with respect to the input. Since the penalty is applied to training examples only, this term forces the model to learn useful information about the training distribution. The final objective function has the following form:

 

The name contractive comes from the fact that the CAE is encouraged to map a neighborhood of input points to a smaller neighborhood of output points.

There is a connection between the denoising autoencoder (DAE) and the contractive autoencoder (CAE): in the limit of small Gaussian input noise, DAE make the reconstruction function resist small but finite-sized perturbations of the input, while CAE make the extracted features resist infinitesimal perturbations of the input.

Variational autoencoder (VAE)Edit

Unlike classical (sparse, denoising, etc.) autoencoders, Variational autoencoders (VAEs) are generative models, like Generative Adversarial Networks.[18] Their association with this group of models derives mainly from the architectural affinity with the basic autoencoder (the final training objective has an encoder and a decoder), but their mathematical formulation differs significantly.[19] VAEs are directed probabilistic graphical models (DPGM) whose posterior is approximated by a neural network, forming an autoencoder-like architecture.[18] Differently from discriminative modeling that aims to learn a predictor given the observation, generative modeling tries to simulate how the data is generated, in order to understand the underlying causal relations. Causal relations have indeed the great potential of being generalizable.[5]

Variational autoencoder models make strong assumptions concerning the distribution of latent variables. They use a variational approach for latent representation learning, which results in an additional loss component and a specific estimator for the training algorithm called the Stochastic Gradient Variational Bayes (SGVB) estimator.[8] It assumes that the data is generated by a directed graphical model   and that the encoder is learning an approximation   to the posterior distribution   where   and   denote the parameters of the encoder (recognition model) and decoder (generative model) respectively. The probability distribution of the latent vector of a VAE typically matches that of the training data much closer than a standard autoencoder. The objective of VAE has the following form:

 

Here,   stands for the Kullback–Leibler divergence. The prior over the latent variables is usually set to be the centred isotropic multivariate Gaussian  ; however, alternative configurations have been considered.[20]

Commonly, the shape of the variational and the likelihood distributions are chosen such that they are factorized Gaussians:

 

where   and   are the encoder outputs, while   and   are the decoder outputs. This choice is justified by the simplifications[8] that it produces when evaluating both the KL divergence and the likelihood term in variational objective defined above.

VAE have been criticized because they generate blurry images.[21] However, researchers employing this model were showing only the mean of the distributions,  , rather than a sample of the learned Gaussian distribution

 .

These samples were shown to be overly noisy due to the choice of a factorized Gaussian distribution.[22][21] Employing a Gaussian distribution with a full covariance matrix,

 

could solve this issue, but is computationally intractable and numerically unstable, as it requires estimating a covariance matrix from a single data sample. However, later research[22][21] showed that a restricted approach where the inverse matrix   is sparse, could be tractably employed to generate images with high-frequency details.

Advantages of DepthEdit

 
Schematic structure of an autoencoder with 3 fully connected hidden layers. The code (z, or h for reference in the text) is the most internal layer.

Autoencoders are often trained with only a single layer encoder and a single layer decoder, but using deep encoders and decoders offers many advantages.[3]

  • Depth can exponentially reduce the computational cost of representing some functions.[3]
  • Depth can exponentially decrease the amount of training data needed to learn some functions.[3]
  • Experimentally, deep autoencoders yield better compression compared to shallow or linear autoencoders.[23]

Training Deep ArchitecturesEdit

Geoffrey Hinton developed a pretraining technique for training many-layered deep autoencoders. This method involves treating each neighbouring set of two layers as a restricted Boltzmann machine so that the pretraining approximates a good solution, then using a backpropagation technique to fine-tune the results.[23] This model takes the name of deep belief network.

Recently, researchers have debated whether joint training (i.e. training the whole architecture together with a single global reconstruction objective to optimize) would be better for deep auto-encoders.[24] A study published in 2015 empirically showed that the joint training method not only learns better data models, but also learned more representative features for classification as compared to the layerwise method.[24] However, their experiments highlighted how the success of joint training for deep autoencoder architectures depends heavily on the regularization strategies adopted in the modern variants of the model.[24][25]

ApplicationsEdit

The two main applications of autoencoders since the 80s have been dimensionality reduction and information retrieval,[3] but modern variations of the basic model were proven successful when applied to different domains and tasks.

Dimensionality ReductionEdit

 
Plot of the first two Principal Components (left) and a two-dimension hidden layer of a Linear Autoencoder (Right) applied to the Fashion MNIST dataset.[26] The two models being both linear learn to span the same subspace. The projection of the data points is indeed identical, a part from rotation of the subspace - to which PCA is invariant.

Dimensionality Reduction was one of the first applications of deep learning, and one of the early motivations to study autoencoders.[3] In a nutshell, the objective is to find a proper projection method, that maps data from high feature space to low feature space.[3]

One milestone paper on the subject was that of Geoffrey Hinton with his publication in Science Magazine in 2006:[23] in that study, he pretrained a multi-layer autoencoder with a stack of RBMs and then used their weights to initialize a deep autoencoder with gradually smaller hidden layers until a bottleneck of 30 neurons. The resulting 30 dimensions of the code yielded a smaller reconstruction error compared to the first 30 principal components of a PCA, and learned a representation that was qualitatively easier to interpret, clearly separating clusters in the original data.[23][3]

Representing data in a lower-dimensional space can improve performance on different tasks, such as classification.[3] Indeed, many forms of dimensionality reduction place semantically related examples near each other,[27] aiding generalization.

Relationship with principal component analysis (PCA)Edit

 
Reconstruction of 28x28pixel images by an Autoencoder with a code size of two (two-units hidden layer) and the reconstruction from the first two Principal Components of PCA. Images come from the Fashion MNIST dataset.[26]

If linear activations are used, or only a single sigmoid hidden layer, then the optimal solution to an autoencoder is strongly related to principal component analysis (PCA).[28][29] The weights of an autoencoder with a single hidden layer of size   (where   is less than the size of the input) span the same vector subspace as the one spanned by the first   principal components, and the output of the autoencoder is an orthogonal projection onto this subspace. The autoencoder weights are not equal to the principal components, and are generally not orthogonal, yet the principal components may be recovered from them using the singular value decomposition.[30]

However, the potential of Autoencoders resides in their non-linearity, allowing the model to learn more powerful generalizations compared to PCA, and to reconstruct back the input with a significantly lower loss of information.[23]

Information RetrievalEdit

Information Retrieval benefits particularly from dimensionality reduction in that search can become extremely efficient in certain kinds of low dimensional spaces. Autoencoders were indeed applied to semantic hashing, proposed by Salakhutdinov and Hinton in 2007.[27] In a nutshell, training the algorithm to produce a low-dimensional binary code, then all database entries could be stored in a hash table mapping binary code vectors to entries. This table would then allow to perform information retrieval by returning all entries with the same binary code as the query, or slightly less similar entries by flipping some bits from the encoding of the query.

Anomaly DetectionEdit

Another field of application for autoencoders is anomaly detection.[31][32][33][34] By learning to replicate the most salient features in the training data under some of the constraints described previously, the model is encourage to learn how to precisely reproduce the most frequent characteristics of the observations. When facing anomalies, the model should worsen its reconstruction performance. In most cases, only data with normal instances are used to train the autoencoder; in others, the frequency of anomalies is so small compared to the whole population of observations, that its contribution to the representation learnt by the model could be ignored. After training, the autoencoder will reconstruct normal data very well, while failing to do so with anomaly data which the autoencoder has not encountered.[32] Reconstruction error of a data point, which is the error between the original data point and its low dimensional reconstruction, is used as an anomaly score to detect anomalies.[32]

Image ProcessingEdit

The peculiar characteristics of autoencoders have rendered these model extremely useful in the processing of images for various tasks.

One example can be found in lossy image compression task, where autoencoders demonstrated their potential by outperforming other approaches and being proven competitive against JPEG 2000.[35]

Another useful application of autoencoders in the field of image preprocessing is image denoising.[36][37] The need for efficient image restoration methods has grown with the massive production of digital images and movies of all kinds, often taken in poor conditions,[38] but autoencoders proved their reliability even in more delicate contexts such as medical images denoising.[39]

Lastly, other successful experiments have been carried out exploiting variations of the basic autoencoder for image super-resolution tasks.[40]

See alsoEdit

ReferencesEdit

  1. ^ Liou, Cheng-Yuan; Huang, Jau-Chi; Yang, Wen-Chie (2008). "Modeling word perception using the Elman network". Neurocomputing. 71 (16–18): 3150. doi:10.1016/j.neucom.2008.04.030.
  2. ^ Liou, Cheng-Yuan; Cheng, Wei-Chen; Liou, Jiun-Wei; Liou, Daw-Ran (2014). "Autoencoder for words". Neurocomputing. 139: 84–96. doi:10.1016/j.neucom.2013.09.055.
  3. ^ a b c d e f g h i j k l Goodfellow, Ian; Bengio, Yoshua; Courville, Aaron (2016). Deep Learning. MIT Press. ISBN 978-0262035613.
  4. ^ a b c d e f Vincent, Pascal; Larochelle, Hugo (2010). "Stacked Denoising Autoencoders: Learning Useful Representations in a Deep Network with a Local Denoising Criterion". Journal of Machine Learning Research. 11: 3371–3408.
  5. ^ a b Diederik P. Kingma, Max Welling, An Introduction to Variational Autoencoders, arXiv:1906.02691
  6. ^ Schmidhuber, Jürgen (January 2015). "Deep learning in neural networks: An overview". Neural Networks. 61: 85–117. arXiv:1404.7828. doi:10.1016/j.neunet.2014.09.003. PMID 25462637.
  7. ^ Hinton, G. E., & Zemel, R. S. (1994). Autoencoders, minimum description length and Helmholtz free energy. In Advances in neural information processing systems (pp. 3-10).
  8. ^ a b c Diederik P Kingma; Welling, Max (2013). "Auto-Encoding Variational Bayes". arXiv:1312.6114 [stat.ML].
  9. ^ Generating Faces with Torch, Boesen A., Larsen L. and Sonderby S.K., 2015 torch.ch/blog/2015/11/13/gan.html
  10. ^ a b Domingos, Pedro (2015). "4". The Master Algorithm: How the Quest for the Ultimate Learning Machine Will Remake Our World. Basic Books. "Deeper into the Brain" subsection. ISBN 978-046506192-1.
  11. ^ Bengio, Y. (2009). "Learning Deep Architectures for AI" (PDF). Foundations and Trends in Machine Learning. 2: 1–127. CiteSeerX 10.1.1.701.9550. doi:10.1561/2200000006.
  12. ^ a b Frey, Brendan; Makhzani, Alireza (2013-12-19). "k-Sparse Autoencoders". arXiv:1312.5663v2. Bibcode:2013arXiv1312.5663M.
  13. ^ a b c Ng, A. (2011). Sparse autoencoder. CS294A Lecture notes, 72(2011), 1-19.
  14. ^ Nair, Vinod; Hinton, Geoffrey E. (2009). "3D Object Recognition with Deep Belief Nets". Proceedings of the 22Nd International Conference on Neural Information Processing Systems. NIPS'09. USA: Curran Associates Inc.: 1339–1347. ISBN 9781615679119.
  15. ^ Zeng, Nianyin; Zhang, Hong; Song, Baoye; Liu, Weibo; Li, Yurong; Dobaie, Abdullah M. (2018-01-17). "Facial expression recognition via learning deep sparse autoencoders". Neurocomputing. 273: 643–649. doi:10.1016/j.neucom.2017.08.043. ISSN 0925-2312.
  16. ^ Arpit, D., Zhou, Y., Ngo, H., & Govindaraju, V. (2015). Why regularized auto-encoders learn sparse representation?. arXiv preprint arXiv:1505.05561.
  17. ^ a b Makhzani, A., & Frey, B. (2013). K-sparse autoencoders. arXiv preprint arXiv:1312.5663.
  18. ^ a b An, J., & Cho, S. (2015). Variational autoencoder based anomaly detection using reconstruction probability. Special Lecture on IE, 2(1).
  19. ^ Doersch, C. (2016). Tutorial on variational autoencoders. arXiv preprint arXiv:1606.05908.
  20. ^ Partaourides, Harris; Chatzis, Sotirios P. (June 2017). "Asymmetric deep generative models". Neurocomputing. 241: 90–96. doi:10.1016/j.neucom.2017.02.028.
  21. ^ a b c Dorta, G.; Vicente, S.; Agapito, L.; Campbell, N. D. F.; Simpson, I. (2018). "Training VAEs Under Structured Residuals". arXiv:1804.01050 [stat.ML].
  22. ^ a b Dorta, G.; Vicente, S.; Agapito, L.; Campbell, N. D. F.; Simpson, I. (2018). "Structured Uncertainty Prediction Networks". The IEEE Conference on Computer Vision and Pattern Recognition (CVPR): 5477–5485. arXiv:1802.07079. Bibcode:2018arXiv180207079D.
  23. ^ a b c d e Hinton, G. E.; Salakhutdinov, R.R. (28 July 2006). "Reducing the Dimensionality of Data with Neural Networks". Science. 313 (5786): 504–507. Bibcode:2006Sci...313..504H. doi:10.1126/science.1127647. PMID 16873662.
  24. ^ a b c Zhou, Y., Arpit, D., Nwogu, I., & Govindaraju, V. (2014). Is joint training better for deep auto-encoders?. arXiv preprint arXiv:1405.1380.
  25. ^ R. Salakhutdinov and G. E. Hinton, “Deep boltzmann machines,” in AISTATS, 2009, pp. 448–455.
  26. ^ a b "Fashion MNIST". 2019-07-12.
  27. ^ a b Salakhutdinov, Ruslan; Hinton, Geoffrey (2009-07-01). "Semantic hashing". International Journal of Approximate Reasoning. Special Section on Graphical Models and Information Retrieval. 50 (7): 969–978. doi:10.1016/j.ijar.2008.11.006. ISSN 0888-613X.
  28. ^ Bourlard, H.; Kamp, Y. (1988). "Auto-association by multilayer perceptrons and singular value decomposition". Biological Cybernetics. 59 (4–5): 291–294. doi:10.1007/BF00332918. PMID 3196773.
  29. ^ Chicco, Davide; Sadowski, Peter; Baldi, Pierre (2014). "Deep autoencoder neural networks for gene ontology annotation predictions". Proceedings of the 5th ACM Conference on Bioinformatics, Computational Biology, and Health Informatics - BCB '14. p. 533. doi:10.1145/2649387.2649442. hdl:11311/964622. ISBN 9781450328944.
  30. ^ Plaut, E (2018). "From Principal Subspaces to Principal Components with Linear Autoencoders". arXiv:1804.10253 [stat.ML].
  31. ^ Sakurada, M., & Yairi, T. (2014, December). Anomaly detection using autoencoders with nonlinear dimensionality reduction. In Proceedings of the MLSDA 2014 2nd Workshop on Machine Learning for Sensory Data Analysis (p. 4). ACM.
  32. ^ a b c An, J., & Cho, S. (2015). Variational autoencoder based anomaly detection using reconstruction probability. Special Lecture on IE, 2, 1-18.
  33. ^ Zhou, C., & Paffenroth, R. C. (2017, August). Anomaly detection with robust deep autoencoders. In Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (pp. 665-674). ACM.
  34. ^ Ribeiro, M., Lazzaretti, A. E., & Lopes, H. S. (2018). A study of deep convolutional auto-encoders for anomaly detection in videos. Pattern Recognition Letters, 105, 13-22.
  35. ^ Theis, L., Shi, W., Cunningham, A., & Huszár, F. (2017). Lossy image compression with compressive autoencoders. arXiv preprint arXiv:1703.00395.
  36. ^ Cho, K. (2013, February). Simple sparsification improves sparse denoising autoencoders in denoising highly corrupted images. In International Conference on Machine Learning (pp. 432-440).
  37. ^ Cho, K. (2013). Boltzmann machines and denoising autoencoders for image denoising. arXiv preprint arXiv:1301.3468.
  38. ^ Antoni Buades, Bartomeu Coll, Jean-Michel Morel. A review of image denoising algorithms, with a new one. Multiscale Modeling and Simulation: A SIAM Interdisciplinary Journal, Society for Industrial and Applied Mathematics, 2005, 4 (2), pp.490-530. hal-00271141
  39. ^ Gondara, Lovedeep (December 2016). "Medical Image Denoising Using Convolutional Denoising Autoencoders". 2016 IEEE 16th International Conference on Data Mining Workshops (ICDMW). Barcelona, Spain: IEEE: 241–246. arXiv:1608.04667. Bibcode:2016arXiv160804667G. doi:10.1109/ICDMW.2016.0041. ISBN 9781509059102.
  40. ^ Zeng, Kun; Yu, Jun; Wang, Ruxin; Li, Cuihua; Tao, Dacheng (January 2017). "Coupled Deep Autoencoder for Single Image Super-Resolution". IEEE Transactions on Cybernetics. 47 (1): 27–37. doi:10.1109/TCYB.2015.2501373. ISSN 2168-2267. PMID 26625442.