Stacked convolutional auto-encoders for hierarchical feature extraction

  • Authors:
  • Jonathan Masci;Ueli Meier;Dan Cireşan;Jürgen Schmidhuber

  • Affiliations:
  • Istituto Dalle Molle di Studi sull'Intelligenza Artificiale, Lugano, Switzerland;Istituto Dalle Molle di Studi sull'Intelligenza Artificiale, Lugano, Switzerland;Istituto Dalle Molle di Studi sull'Intelligenza Artificiale, Lugano, Switzerland;Istituto Dalle Molle di Studi sull'Intelligenza Artificiale, Lugano, Switzerland

  • Venue:
  • ICANN'11 Proceedings of the 21th international conference on Artificial neural networks - Volume Part I
  • Year:
  • 2011

Quantified Score

Hi-index 0.00

Visualization

Abstract

We present a novel convolutional auto-encoder (CAE) for unsupervised feature learning. A stack of CAEs forms a convolutional neural network (CNN). Each CAE is trained using conventional on-line gradient descent without additional regularization terms. A max-pooling layer is essential to learn biologically plausible features consistent with those found by previous approaches. Initializing a CNN with filters of a trained CAE stack yields superior performance on a digit (MNIST) and an object recognition (CIFAR10) benchmark.