IDEAS home Printed from https://ideas.repec.org/a/spr/coopap/v77y2020i1d10.1007_s10589-020-00202-1.html
   My bibliography  Save this article

Make $$\ell _1$$ ℓ 1 regularization effective in training sparse CNN

Author

Listed:
  • Juncai He

    (Pennsylvania State University)

  • Xiaodong Jia

    (Pennsylvania State University)

  • Jinchao Xu

    (Pennsylvania State University)

  • Lian Zhang

    (Pennsylvania State University)

  • Liang Zhao

    (Chinese Academy of Sciences, and University of Chinese Academy of Sciences)

Abstract

Compressed Sensing using $$\ell _1$$ ℓ 1 regularization is among the most powerful and popular sparsification technique in many applications, but why has it not been used to obtain sparse deep learning model such as convolutional neural network (CNN)? This paper is aimed to provide an answer to this question and to show how to make it work. Following Xiao (J Mach Learn Res 11(Oct):2543–2596, 2010), We first demonstrate that the commonly used stochastic gradient decent and variants training algorithm is not an appropriate match with $$\ell _1$$ ℓ 1 regularization and then replace it with a different training algorithm based on a regularized dual averaging (RDA) method. The RDA method of Xiao (J Mach Learn Res 11(Oct):2543–2596, 2010) was originally designed specifically for convex problem, but with new theoretical insight and algorithmic modifications (using proper initialization and adaptivity), we have made it an effective match with $$\ell _1$$ ℓ 1 regularization to achieve a state-of-the-art sparsity for the highly non-convex CNN compared to other weight pruning methods without compromising accuracy (achieving 95% sparsity for ResNet-18 on CIFAR-10, for example).

Suggested Citation

  • Juncai He & Xiaodong Jia & Jinchao Xu & Lian Zhang & Liang Zhao, 2020. "Make $$\ell _1$$ ℓ 1 regularization effective in training sparse CNN," Computational Optimization and Applications, Springer, vol. 77(1), pages 163-182, September.
  • Handle: RePEc:spr:coopap:v:77:y:2020:i:1:d:10.1007_s10589-020-00202-1
    DOI: 10.1007/s10589-020-00202-1
    as

    Download full text from publisher

    File URL: http://link.springer.com/10.1007/s10589-020-00202-1
    File Function: Abstract
    Download Restriction: Access to the full text of the articles in this series is restricted.

    File URL: https://libkey.io/10.1007/s10589-020-00202-1?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:spr:coopap:v:77:y:2020:i:1:d:10.1007_s10589-020-00202-1. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Sonal Shukla or Springer Nature Abstracting and Indexing (email available below). General contact details of provider: http://www.springer.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.