CMA Light: a novel Minibatch Algorithm for large-scale non convex finite sum optimization

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Coppola, Corrado, Liuzzi, Giampaolo, Palagi, Laura
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909208386994176
author Coppola, Corrado
Liuzzi, Giampaolo
Palagi, Laura
author_facet Coppola, Corrado
Liuzzi, Giampaolo
Palagi, Laura
contents The supervised training of a deep neural network on a given dataset consists in the unconstrained minimization of the finite sum of continuously differentiable functions, commonly referred to as loss with respect to the samples. These functions depend on the network parameters and most of the times are non-convex. We develop CMA Light, a globally convergent mini-batch gradient method to tackle this problem. We consider the recently introduced Controlled Minibatch Algorithm (CMA) framework and we overcome its main bottleneck, removing the need for at least one evaluation of the whole objective function per iteration. We prove globally convergence of CMA Light under mild assumptions and we discuss extensive computational results on the same experimental test-bed used for CMA, showing that CMA Light requires less computational effort than most of the state-of-the-art optimizers. Eventually, we present early results on a large-scale Image Classification task.
format Preprint
id arxiv_https___arxiv_org_abs_2307_15775
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle CMA Light: a novel Minibatch Algorithm for large-scale non convex finite sum optimization
Coppola, Corrado
Liuzzi, Giampaolo
Palagi, Laura
Optimization and Control
The supervised training of a deep neural network on a given dataset consists in the unconstrained minimization of the finite sum of continuously differentiable functions, commonly referred to as loss with respect to the samples. These functions depend on the network parameters and most of the times are non-convex. We develop CMA Light, a globally convergent mini-batch gradient method to tackle this problem. We consider the recently introduced Controlled Minibatch Algorithm (CMA) framework and we overcome its main bottleneck, removing the need for at least one evaluation of the whole objective function per iteration. We prove globally convergence of CMA Light under mild assumptions and we discuss extensive computational results on the same experimental test-bed used for CMA, showing that CMA Light requires less computational effort than most of the state-of-the-art optimizers. Eventually, we present early results on a large-scale Image Classification task.
title CMA Light: a novel Minibatch Algorithm for large-scale non convex finite sum optimization
topic Optimization and Control
url https://arxiv.org/abs/2307.15775