Random-reshuffled SARAH does not need a full gradient computations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Beznosikov, Aleksandr, Takáč, Martin
Format: Preprint
Published: 2021
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914640484630528
author Beznosikov, Aleksandr
Takáč, Martin
author_facet Beznosikov, Aleksandr
Takáč, Martin
contents The StochAstic Recursive grAdient algoritHm (SARAH) algorithm is a variance reduced variant of the Stochastic Gradient Descent (SGD) algorithm that needs a gradient of the objective function from time to time. In this paper, we remove the necessity of a full gradient computation. This is achieved by using a randomized reshuffling strategy and aggregating stochastic gradients obtained in each epoch. The aggregated stochastic gradients serve as an estimate of a full gradient in the SARAH algorithm. We provide a theoretical analysis of the proposed approach and conclude the paper with numerical experiments that demonstrate the efficiency of this approach.
format Preprint
id arxiv_https___arxiv_org_abs_2111_13322
institution arXiv
publishDate 2021
record_format arxiv
spellingShingle Random-reshuffled SARAH does not need a full gradient computations
Beznosikov, Aleksandr
Takáč, Martin
Machine Learning
Optimization and Control
The StochAstic Recursive grAdient algoritHm (SARAH) algorithm is a variance reduced variant of the Stochastic Gradient Descent (SGD) algorithm that needs a gradient of the objective function from time to time. In this paper, we remove the necessity of a full gradient computation. This is achieved by using a randomized reshuffling strategy and aggregating stochastic gradients obtained in each epoch. The aggregated stochastic gradients serve as an estimate of a full gradient in the SARAH algorithm. We provide a theoretical analysis of the proposed approach and conclude the paper with numerical experiments that demonstrate the efficiency of this approach.
title Random-reshuffled SARAH does not need a full gradient computations
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2111.13322