A sparsity-based stochastic pooling mechanism for deep convolutional neural networks.

Journal: Neural networks : the official journal of the International Neural Network Society
Published Date:

Abstract

A novel sparsity-based stochastic pooling which integrates the advantages of max-pooling, average-pooling and stochastic pooling is introduced. The proposed pooling is designed to balance the advantages and disadvantages of max-pooling and average-pooling by using the degree of sparsity of activations and a control function to obtain an optimized representative feature value ranging from average value to maximum value of a pooling region. The optimized representative feature value is employed for probability weights assignment of activations in normal distribution. The proposed pooling also adopts weighted random sampling with a reservoir for the sampling process to preserve the advantages of stochastic pooling. This proposed pooling is evaluated on several standard datasets in deep learning framework to compare with various classic pooling methods. Experimental results show that it has good performance on improving recognition accuracy. The influence of changes to the feature parameter on recognition accuracy is also investigated.

Authors

  • Zhenhua Song
    The Third Affiliated Hospital of Southern Medical University, Guangzhou, 510630, China.
  • Yan Liu
    Department of Clinical Microbiology, Shanghai Tenth People's Hospital, School of Medicine, Tongji University, Shanghai, 200072, People's Republic of China.
  • Rong Song
  • Zhenguang Chen
    The First Affiliated Hospital of Sun Yat-sen University, Sun Yat-sen University, Guangzhou 510080, PR China.
  • Jianyong Yang
    The First Affiliated Hospital of Sun Yat-sen University, Sun Yat-sen University, Guangzhou 510080, PR China.
  • Chao Zhang
    School of Information Engineering, Suqian University, Suqian, Jiangsu, China.
  • Qing Jiang
    Department of Urology, The Second Affiliated Hospital of Chongqing Medical University, Chongqing, China.