A NEW TRAINING ALGORITHM OF AUTOENCODER COMBINING LATERAL INHIBITION MECHANISM

Jiacheng Ni,Yuelei Xu,Shiping Ma,Shuai Li
DOI: https://doi.org/10.3969/j.issn.1000-386x.2015.09.038
2015-01-01
Abstract:Deep learning has emerged as one of the most popular machine learning means.When in training,the autoencoder in deep learning is easy to produce the problems of excess network model complexity,insufficient output matrix sparsity and over-fitting in small sample training,etc.Aiming at such issues,we present a new training algorithm for autoencoder which combines the lateral inhibition mechanism.The algorithm builds the lateral inhibition neuron screening model used in hidden layer,it first sets up the inhibition threshold for seeking the neurons satisfying inhibition condition,then inhibits the fast output of neurons meeting the condition with lateral inhibition function,and employs back propagation algorithm to optimise the model and finally outputs the weight characteristics.Experimental result proves that the algorithm can make the hidden layer output approximately meet the sparse condition and get more robust characteristics through learning,while improving correctness rate of classification,it is able to restrain over-fitting phenomenon to certain extent as well.
What problem does this paper attempt to address?