Харитонов Валерий Дмитриевич
Bayesian Pruning of Neural Networks
Науки о данных
We study the empirical Bayes approach to hyperparameter optimization and model selection in Bayesian neural networks. We explore its automatic relevance determination and pruning effect when Gaussian prior and approximate posterior for neural network weights are used. The connection is drawn between Sparse Variational Dropout and Gaussian Automatic Relevance Determination objectives. Based on the empirical Bayes procedure, we propose a novel type of prior that encourages structured sparsity and allows for compression and speeding up of modern convolutional deep neural network architectures. We provide experimental results showing that the proposed structured pruning procedure removes most of the filters or neurons in convolutional and dense layers in LeNet5 and VGG-like convolutional neural networks with almost no drop in accuracy of classification on MNIST and CIFAR-10 datasets. The new approach outperforms previously reported results in terms of sparsity levels in convolutional layers.