4.7 Article

Block-cyclic stochastic coordinate descent for deep neural networks

Journal

NEURAL NETWORKS
Volume 139, Issue -, Pages 348-357

Publisher

PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.neunet.2021.04.001

Keywords

Coordinate descent; Deep neural network; Energy optimization; Stochastic gradient descent

Funding

  1. National Research Foundation of Korea (NRF), Republic of Korea [2017R1A2B4006023, 2019K1A3A1A77074958]
  2. Office of Naval Research (ONR), USA [N00014-19-1-2229]
  3. National Research Foundation of Korea [2019K1A3A1A77074958, 2017R1A2B4006023] Funding Source: Korea Institute of Science & Technology Information (KISTI), National Science & Technology Information Service (NTIS)

Ask authors/readers for more resources

BCSC is a stochastic first-order optimization algorithm that adds a cyclic constraint to the selection of data and parameters, resulting in higher accuracy in image classification. It effectively limits the impact of outliers in the training set and provides better generalization performance within the same number of update iterations.
We present a stochastic first-order optimization algorithm, named block-cyclic stochastic coordinate descent (BCSC), that adds a cyclic constraint to stochastic block-coordinate descent in the selection of both data and parameters. It uses different subsets of the data to update different subsets of the parameters, thus limiting the detrimental effect of outliers in the training set. Empirical tests in image classification benchmark datasets show that BCSC outperforms state-of-the-art optimization methods in generalization leading to higher accuracy within the same number of update iterations. The improvements are consistent across different architectures and datasets, and can be combined with other training techniques and regularizations. (C) 2021 Elsevier Ltd. All rights reserved.

Authors

I am an author on this paper
Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.7
Not enough ratings

Secondary Ratings

Novelty
-
Significance
-
Scientific rigor
-
Rate this paper

Recommended

No Data Available
No Data Available