KOALA: A Kalman Optimization Algorithm with Loss Adaptivity
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2021
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866913751766138880 |
|---|---|
| author | Davtyan, Aram Sameni, Sepehr Cerkezi, Llukman Meishvilli, Givi Bielski, Adam Favaro, Paolo |
| author_facet | Davtyan, Aram Sameni, Sepehr Cerkezi, Llukman Meishvilli, Givi Bielski, Adam Favaro, Paolo |
| contents | Optimization is often cast as a deterministic problem, where the solution is found through some iterative procedure such as gradient descent. However, when training neural networks the loss function changes over (iteration) time due to the randomized selection of a subset of the samples. This randomization turns the optimization problem into a stochastic one. We propose to consider the loss as a noisy observation with respect to some reference optimum. This interpretation of the loss allows us to adopt Kalman filtering as an optimizer, as its recursive formulation is designed to estimate unknown parameters from noisy measurements. Moreover, we show that the Kalman Filter dynamical model for the evolution of the unknown parameters can be used to capture the gradient dynamics of advanced methods such as Momentum and Adam. We call this stochastic optimization method KOALA, which is short for Kalman Optimization Algorithm with Loss Adaptivity. KOALA is an easy to implement, scalable, and efficient method to train neural networks. We provide convergence analysis and show experimentally that it yields parameter estimates that are on par with or better than existing state of the art optimization algorithms across several neural network architectures and machine learning tasks, such as computer vision and language modeling. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2107_03331 |
| institution | arXiv |
| publishDate | 2021 |
| record_format | arxiv |
| spellingShingle | KOALA: A Kalman Optimization Algorithm with Loss Adaptivity Davtyan, Aram Sameni, Sepehr Cerkezi, Llukman Meishvilli, Givi Bielski, Adam Favaro, Paolo Machine Learning Computer Vision and Pattern Recognition Optimization and Control Optimization is often cast as a deterministic problem, where the solution is found through some iterative procedure such as gradient descent. However, when training neural networks the loss function changes over (iteration) time due to the randomized selection of a subset of the samples. This randomization turns the optimization problem into a stochastic one. We propose to consider the loss as a noisy observation with respect to some reference optimum. This interpretation of the loss allows us to adopt Kalman filtering as an optimizer, as its recursive formulation is designed to estimate unknown parameters from noisy measurements. Moreover, we show that the Kalman Filter dynamical model for the evolution of the unknown parameters can be used to capture the gradient dynamics of advanced methods such as Momentum and Adam. We call this stochastic optimization method KOALA, which is short for Kalman Optimization Algorithm with Loss Adaptivity. KOALA is an easy to implement, scalable, and efficient method to train neural networks. We provide convergence analysis and show experimentally that it yields parameter estimates that are on par with or better than existing state of the art optimization algorithms across several neural network architectures and machine learning tasks, such as computer vision and language modeling. |
| title | KOALA: A Kalman Optimization Algorithm with Loss Adaptivity |
| topic | Machine Learning Computer Vision and Pattern Recognition Optimization and Control |
| url | https://arxiv.org/abs/2107.03331 |