Dynamic Estimation of Learning Rates Using a Non-Linear Autoregressive Model

Fuente: arXiv
Saved in:
Bibliographic Details
Main Author: Okhrati, Ramin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912140292521984
author Okhrati, Ramin
author_facet Okhrati, Ramin
contents We introduce a new class of adaptive non-linear autoregressive (Nlar) models incorporating the concept of momentum, which dynamically estimate both the learning rates and momentum as the number of iterations increases. In our method, the growth of the gradients is controlled using a scaling (clipping) function, leading to stable convergence. Within this framework, we propose three distinct estimators for learning rates and provide theoretical proof of their convergence. We further demonstrate how these estimators underpin the development of effective Nlar optimizers. The performance of the proposed estimators and optimizers is rigorously evaluated through extensive experiments across several datasets and a reinforcement learning environment. The results highlight two key features of the Nlar optimizers: robust convergence despite variations in underlying parameters, including large initial learning rates, and strong adaptability with rapid convergence during the initial epochs.
format Preprint
id arxiv_https___arxiv_org_abs_2410_09943
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dynamic Estimation of Learning Rates Using a Non-Linear Autoregressive Model
Okhrati, Ramin
Machine Learning
Optimization and Control
Applications
We introduce a new class of adaptive non-linear autoregressive (Nlar) models incorporating the concept of momentum, which dynamically estimate both the learning rates and momentum as the number of iterations increases. In our method, the growth of the gradients is controlled using a scaling (clipping) function, leading to stable convergence. Within this framework, we propose three distinct estimators for learning rates and provide theoretical proof of their convergence. We further demonstrate how these estimators underpin the development of effective Nlar optimizers. The performance of the proposed estimators and optimizers is rigorously evaluated through extensive experiments across several datasets and a reinforcement learning environment. The results highlight two key features of the Nlar optimizers: robust convergence despite variations in underlying parameters, including large initial learning rates, and strong adaptability with rapid convergence during the initial epochs.
title Dynamic Estimation of Learning Rates Using a Non-Linear Autoregressive Model
topic Machine Learning
Optimization and Control
Applications
url https://arxiv.org/abs/2410.09943