English

A Variant of Gradient Descent Algorithm Based on Gradient Averaging

Machine Learning 2020-12-11 v2 Optimization and Control Machine Learning

Abstract

In this work, we study an optimizer, Grad-Avg to optimize error functions. We establish the convergence of the sequence of iterates of Grad-Avg mathematically to a minimizer (under boundedness assumption). We apply Grad-Avg along with some of the popular optimizers on regression as well as classification tasks. In regression tasks, it is observed that the behaviour of Grad-Avg is almost identical with Stochastic Gradient Descent (SGD). We present a mathematical justification of this fact. In case of classification tasks, it is observed that the performance of Grad-Avg can be enhanced by suitably scaling the parameters. Experimental results demonstrate that Grad-Avg converges faster than the other state-of-the-art optimizers for the classification task on two benchmark datasets.

Keywords

Cite

@article{arxiv.2012.02387,
  title  = {A Variant of Gradient Descent Algorithm Based on Gradient Averaging},
  author = {Saugata Purkayastha and Sukannya Purkayastha},
  journal= {arXiv preprint arXiv:2012.02387},
  year   = {2020}
}

Comments

9 pages, 4 figures. Accepted at OPT2020: 12th Annual Workshop on Optimization for Machine Learning @ NeurIPS, 2020

R2 v1 2026-06-23T20:43:29.450Z