Faster Stochastic Optimization with Arbitrary Delays via Asynchronous Mini-Batching
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866913900442681344 |
|---|---|
| author | Attia, Amit Gaash, Ofir Koren, Tomer |
| author_facet | Attia, Amit Gaash, Ofir Koren, Tomer |
| contents | We consider the problem of asynchronous stochastic optimization, where an optimization algorithm makes updates based on stale stochastic gradients of the objective that are subject to an arbitrary (possibly adversarial) sequence of delays. We present a procedure which, for any given $q \in (0,1]$, transforms any standard stochastic first-order method to an asynchronous method with convergence guarantee depending on the $q$-quantile delay of the sequence. This approach leads to convergence rates of the form $O(τ_q/qT+σ/\sqrt{qT})$ for non-convex and $O(τ_q^2/(q T)^2+σ/\sqrt{qT})$ for convex smooth problems, where $τ_q$ is the $q$-quantile delay, generalizing and improving on existing results that depend on the average delay. We further show a method that automatically adapts to all quantiles simultaneously, without any prior knowledge of the delays, achieving convergence rates of the form $O(\inf_{q} τ_q/qT+σ/\sqrt{qT})$ for non-convex and $O(\inf_{q} τ_q^2/(q T)^2+σ/\sqrt{qT})$ for convex smooth problems. Our technique is based on asynchronous mini-batching with a careful batch-size selection and filtering of stale gradients. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2408_07503 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Faster Stochastic Optimization with Arbitrary Delays via Asynchronous Mini-Batching Attia, Amit Gaash, Ofir Koren, Tomer Optimization and Control Machine Learning We consider the problem of asynchronous stochastic optimization, where an optimization algorithm makes updates based on stale stochastic gradients of the objective that are subject to an arbitrary (possibly adversarial) sequence of delays. We present a procedure which, for any given $q \in (0,1]$, transforms any standard stochastic first-order method to an asynchronous method with convergence guarantee depending on the $q$-quantile delay of the sequence. This approach leads to convergence rates of the form $O(τ_q/qT+σ/\sqrt{qT})$ for non-convex and $O(τ_q^2/(q T)^2+σ/\sqrt{qT})$ for convex smooth problems, where $τ_q$ is the $q$-quantile delay, generalizing and improving on existing results that depend on the average delay. We further show a method that automatically adapts to all quantiles simultaneously, without any prior knowledge of the delays, achieving convergence rates of the form $O(\inf_{q} τ_q/qT+σ/\sqrt{qT})$ for non-convex and $O(\inf_{q} τ_q^2/(q T)^2+σ/\sqrt{qT})$ for convex smooth problems. Our technique is based on asynchronous mini-batching with a careful batch-size selection and filtering of stale gradients. |
| title | Faster Stochastic Optimization with Arbitrary Delays via Asynchronous Mini-Batching |
| topic | Optimization and Control Machine Learning |
| url | https://arxiv.org/abs/2408.07503 |