Title :
Scalable learning for restricted Boltzmann machines
Author :
Barshan, Elnaz ; Fieguth, Paul
Author_Institution :
Dept. of Syst. Design Eng., Univ. of Waterloo, Waterloo, ON, Canada
Abstract :
We propose Eigen-RBM, a scalable Restricted Boltzmann Machine (RBM) for visual recognition in which the number of free parameters to learn is independent of the image size. Eigen-RBM exploits the global structure of the image and does not impose any locality or translation-invariance assumption, and regularizes the network weights to be a linear combination of a set of predefined filters. We show that, compared to basic RBM, Eigen-RBM can achieve similar or better performance in both recognition and sample generation with significantly less training time.
Keywords :
Boltzmann machines; eigenvalues and eigenfunctions; image filtering; image recognition; learning (artificial intelligence); eigen-RBM; image global structure; image size; network weights; predefined filters; scalable learning; scalable restricted Boltzmann machine; visual recognition; Computational modeling; Computer vision; Conferences; Neurons; Pattern recognition; Training; Visualization; Feature Extraction; Generative Models; Image Classification; Machine Learning; Restricted Boltzmann Machines;
Conference_Titel :
Image Processing (ICIP), 2014 IEEE International Conference on
Conference_Location :
Paris
DOI :
10.1109/ICIP.2014.7025557