DocumentCode
1797407
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
Volume
1
fYear
2014
fDate
13-16 July 2014
Firstpage
271
Lastpage
277
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;
fLanguage
English
Publisher
ieee
Conference_Titel
Machine Learning and Cybernetics (ICMLC), 2014 International Conference on
Conference_Location
Lanzhou
ISSN
2160-133X
Print_ISBN
978-1-4799-4216-9
Type
conf
DOI
10.1109/ICMLC.2014.7009128
Filename
7009128
Link To Document