A novel pruning algorithm for smoothing feedforward neural networks based on group lasso method

J Wang, C Xu, X Yang… - IEEE transactions on …, 2017 - ieeexplore.ieee.org
J Wang, C Xu, X Yang, JM Zurada
IEEE transactions on neural networks and learning systems, 2017ieeexplore.ieee.org
In this paper, we propose four new variants of the backpropagation algorithm to improve the
generalization ability for feedforward neural networks. The basic idea of these methods
stems from the Group Lasso concept which deals with the variable selection problem at the
group level. There are two main drawbacks when the Group Lasso penalty has been directly
employed during network training. They are numerical oscillations and theoretical
challenges in computing the gradients at the origin. To overcome these obstacles …
In this paper, we propose four new variants of the backpropagation algorithm to improve the generalization ability for feedforward neural networks. The basic idea of these methods stems from the Group Lasso concept which deals with the variable selection problem at the group level. There are two main drawbacks when the Group Lasso penalty has been directly employed during network training. They are numerical oscillations and theoretical challenges in computing the gradients at the origin. To overcome these obstacles, smoothing functions have then been introduced by approximating the Group Lasso penalty. Numerical experiments for classification and regression problems demonstrate that the proposed algorithms perform better than the other three classical penalization methods, Weight Decay, Weight Elimination, and Approximate Smoother, on both generalization and pruning efficiency. In addition, detailed simulations based on a specific data set have been performed to compare with some other common pruning strategies, which verify the advantages of the proposed algorithm. The pruning abilities of the proposed strategy have been investigated in detail for a relatively large data set, MNIST, in terms of various smoothing approximation cases.
ieeexplore.ieee.org
以上显示的是最相近的搜索结果。 查看全部搜索结果