Transformed _1 regularization for learning sparse deep neural networks

From MaRDI portal
Publication:2185659

DOI10.1016/J.NEUNET.2019.08.015zbMATH Open1434.68512DBLPjournals/nn/MaMNZ19arXiv1901.01021OpenAlexW2970738028WikidataQ93198988 ScholiaQ93198988MaRDI QIDQ2185659FDOQ2185659


Authors: Yanyan Li Edit this on Wikidata


Publication date: 5 June 2020

Published in: Neural Networks (Search for Journal in Brave)

Abstract: Deep neural networks (DNNs) have achieved extraordinary success in numerous areas. However, to attain this success, DNNs often carry a large number of weight parameters, leading to heavy costs of memory and computation resources. Overfitting is also likely to happen in such network when the training data are insufficient. These shortcomings severely hinder the application of DNNs in resource-constrained platforms. In fact, many network weights are known to be redundant and can be removed from the network without much loss of performance. To this end, we introduce a new non-convex integrated transformed ell1 regularizer to promote sparsity for DNNs, which removes both redundant connections and unnecessary neurons simultaneously. To be specific, we apply the transformed ell1 to the matrix space of network weights and utilize it to remove redundant connections. Besides, group sparsity is also employed as an auxiliary to remove unnecessary neurons. An efficient stochastic proximal gradient algorithm is presented to solve the new model at the same time. To the best of our knowledge, this is the first work to utilize a non-convex regularizer in sparse optimization based method to promote sparsity for DNNs. Experiments on several public datasets demonstrate the effectiveness of the proposed method.


Full work available at URL: https://arxiv.org/abs/1901.01021




Recommendations




Cites Work


Cited In (18)

Uses Software





This page was built for publication: Transformed \(\ell_1\) regularization for learning sparse deep neural networks

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q2185659)