Title :
Auto-encoder using the bi-firing activation function
Author :
Zihong Cao ; Guangjun Zeng ; Ng, Wing W. Y. ; Jincheng Le
Author_Institution :
Machine Learning & Cybern. Res. Center, South China Univ. of Technol., Guangzhou, China
Abstract :
Training the whole deep neural network together is restricted by the gradient diffusion problem. Greedy layer-wise training of an auto-encoder has achieved promising results in deep neural networks. However, it can not learn useful input representation from the original input directly. In this work, we propose to use the bi-firing activation function for auto-encoder with an end-to-end training scheme. It not only improves the training efficiency but also learns better features than the traditional stacked auto-encoder. Experimental results show that it extracts more representative features and also outperforms the stacked auto-encoder in supervised classification task.
Keywords :
feature extraction; gradient methods; image classification; image representation; neural nets; auto-encoder; bi-firing activation function; end-to-end training scheme; feature extraction; gradient diffusion problem; greedy layer-wise training; supervised classification task; whole deep neural network; Abstracts; Accuracy; Lead; Training; Tuning; Auto-encoder; Bi-firing function; Deep Learning; Layer-wise scheme;
Conference_Titel :
Machine Learning and Cybernetics (ICMLC), 2014 International Conference on
Conference_Location :
Lanzhou
Print_ISBN :
978-1-4799-4216-9
DOI :
10.1109/ICMLC.2014.7009128