Adaptive Weight Decay for Deep Neural Networksopen access
- Authors
- Nakamura, Kensuke; Hong, Byung-Woo
- Issue Date
- Aug-2019
- Publisher
- IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
- Keywords
- Adaptive regularization; deep learning; neural networks; stochastic gradient descent; weight-decay
- Citation
- IEEE ACCESS, v.7, pp 118857 - 118865
- Pages
- 9
- Journal Title
- IEEE ACCESS
- Volume
- 7
- Start Page
- 118857
- End Page
- 118865
- URI
- https://scholarworks.bwise.kr/cau/handle/2019.sw.cau/39048
- DOI
- 10.1109/ACCESS.2019.2937139
- ISSN
- 2169-3536
- Abstract
- Regularization in the optimization of deep neural networks is often critical to avoid undesirable over-fitting leading to better generalization of model. One of the most popular regularization algorithms is to impose L-2 penalty on the model parameters resulting in the decay of parameters, called weight-decay, and the decay rate is generally constant to all the model parameters in the course of optimization. In contrast to the previous approach based on the constant rate of weight-decay, we propose to consider the residual that measures dissimilarity between the current state of model and observations in the determination of the weight-decay for each parameter in an adaptive way, called adaptive weight-decay (AdaDecay) where the gradient norms are normalized within each layer and the degree of regularization for each parameter is determined in proportional to the magnitude of its gradient using the sigmoid function. We empirically demonstrate the effectiveness of AdaDecay in comparison to the state-of-the-art optimization algorithms using popular benchmark datasets: MNIST, Fashion-MNIST, and CIFAR-10 with conventional neural network models ranging from shallow to deep. The quantitative evaluation of our proposed algorithm indicates that AdaDecay improves generalization leading to better accuracy across all the datasets and models.
- Files in This Item
- There are no files associated with this item.
- Appears in
Collections - College of Software > Department of Artificial Intelligence > 1. Journal Articles
![qrcode](https://api.qrserver.com/v1/create-qr-code/?size=55x55&data=https://scholarworks.bwise.kr/cau/handle/2019.sw.cau/39048)
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.