AdaNorm: Adaptive Gradient Norm Correction based Optimizer for CNNs

Document Type

Conference Article

Publication Title

Proceedings - 2023 IEEE Winter Conference on Applications of Computer Vision, WACV 2023

Abstract

The stochastic gradient descent (SGD) optimizers are generally used to train the convolutional neural networks (CNNs). In recent years, several adaptive momentum based SGD optimizers have been introduced, such as Adam, diffGrad, Radam and AdaBelief. However, the existing SGD optimizers do not exploit the gradient norm of past iterations and lead to poor convergence and performance. In this paper, we propose a novel AdaNorm based SGD optimizers by correcting the norm of gradient in each iteration based on the adaptive training history of gradient norm. By doing so, the proposed optimizers are able to maintain high and representive gradient throughout the training and solves the low and atypical gradient problems. The proposed concept is generic and can be used with any existing SGD optimizer. We show the efficacy of the proposed AdaNorm with four state-of-the-art optimizers, including Adam, diffGrad, Radam and AdaBelief. We depict the performance improvement due to the proposed optimizers using three CNN models, including VGG16, ResNet18 and ResNet50, on three benchmark object recognition datasets, including CIFAR10, CIFAR100 and TinyImageNet.

First Page

5273

Last Page

5282

DOI

10.1109/WACV56688.2023.00525

Publication Date

1-1-2023

Comments

Open Access, Green

This document is currently not available here.

Share

COinS