Is All Learning (Natural) Gradient Descent?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Shoji, Lucas, Suzuki, Kenta, Kozachkov, Leo
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910619140096000
author Shoji, Lucas
Suzuki, Kenta
Kozachkov, Leo
author_facet Shoji, Lucas
Suzuki, Kenta
Kozachkov, Leo
contents This paper shows that a wide class of effective learning rules -- those that improve a scalar performance measure over a given time window -- can be rewritten as natural gradient descent with respect to a suitably defined loss function and metric. Specifically, we show that parameter updates within this class of learning rules can be expressed as the product of a symmetric positive definite matrix (i.e., a metric) and the negative gradient of a loss function. We also demonstrate that these metrics have a canonical form and identify several optimal ones, including the metric that achieves the minimum possible condition number. The proofs of the main results are straightforward, relying only on elementary linear algebra and calculus, and are applicable to continuous-time, discrete-time, stochastic, and higher-order learning rules, as well as loss functions that explicitly depend on time.
format Preprint
id arxiv_https___arxiv_org_abs_2409_16422
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Is All Learning (Natural) Gradient Descent?
Shoji, Lucas
Suzuki, Kenta
Kozachkov, Leo
Machine Learning
Dynamical Systems
Neurons and Cognition
This paper shows that a wide class of effective learning rules -- those that improve a scalar performance measure over a given time window -- can be rewritten as natural gradient descent with respect to a suitably defined loss function and metric. Specifically, we show that parameter updates within this class of learning rules can be expressed as the product of a symmetric positive definite matrix (i.e., a metric) and the negative gradient of a loss function. We also demonstrate that these metrics have a canonical form and identify several optimal ones, including the metric that achieves the minimum possible condition number. The proofs of the main results are straightforward, relying only on elementary linear algebra and calculus, and are applicable to continuous-time, discrete-time, stochastic, and higher-order learning rules, as well as loss functions that explicitly depend on time.
title Is All Learning (Natural) Gradient Descent?
topic Machine Learning
Dynamical Systems
Neurons and Cognition
url https://arxiv.org/abs/2409.16422