Training Learned Optimizers with Randomly Initialized Learned Optimizers
- URL: http://arxiv.org/abs/2101.07367v1
- Date: Thu, 14 Jan 2021 19:07:17 GMT
- Title: Training Learned Optimizers with Randomly Initialized Learned Optimizers
- Authors: Luke Metz, C. Daniel Freeman, Niru Maheswaranathan, Jascha
Sohl-Dickstein
- Abstract summary: We show that a population of randomly learneds can be used to train themselves from scratch in an online fashion.
A form of population based training is used to orchestrate this self-training.
We believe feedback loops of this type will be important and powerful in the future of machine learning.
- Score: 49.67678615506608
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Learned optimizers are increasingly effective, with performance exceeding
that of hand designed optimizers such as Adam~\citep{kingma2014adam} on
specific tasks \citep{metz2019understanding}. Despite the potential gains
available, in current work the meta-training (or `outer-training') of the
learned optimizer is performed by a hand-designed optimizer, or by an optimizer
trained by a hand-designed optimizer \citep{metz2020tasks}. We show that a
population of randomly initialized learned optimizers can be used to train
themselves from scratch in an online fashion, without resorting to a hand
designed optimizer in any part of the process. A form of population based
training is used to orchestrate this self-training. Although the randomly
initialized optimizers initially make slow progress, as they improve they
experience a positive feedback loop, and become rapidly more effective at
training themselves. We believe feedback loops of this type, where an optimizer
improves itself, will be important and powerful in the future of machine
learning. These methods not only provide a path towards increased performance,
but more importantly relieve research and engineering effort.
Related papers
- Learning to Optimize for Reinforcement Learning [58.01132862590378]
Reinforcement learning (RL) is essentially different from supervised learning, and in practice, these learneds do not work well even in simple RL tasks.
Agent-gradient distribution is non-independent and identically distributed, leading to inefficient meta-training.
We show that, although only trained in toy tasks, our learned can generalize unseen complex tasks in Brax.
arXiv Detail & Related papers (2023-02-03T00:11:02Z) - VeLO: Training Versatile Learned Optimizers by Scaling Up [67.90237498659397]
We leverage the same scaling approach behind the success of deep learning to learn versatiles.
We train an ingest for deep learning which is itself a small neural network that ingests and outputs parameter updates.
We open source our learned, meta-training code, the associated train test data, and an extensive benchmark suite with baselines at velo-code.io.
arXiv Detail & Related papers (2022-11-17T18:39:07Z) - A Closer Look at Learned Optimization: Stability, Robustness, and
Inductive Biases [44.01339030872185]
Blackbox learneds often struggle with stability and generalization when applied to tasks unlike those in their meta-training set.
We investigate the inductive biases and stability properties of optimization algorithms, and apply the resulting insights to designing inductive biases for blackboxs.
We learn to a variety of neural network training tasks, where it outperforms the current state of the art learned.
arXiv Detail & Related papers (2022-09-22T17:47:21Z) - Practical tradeoffs between memory, compute, and performance in learned
optimizers [46.04132441790654]
We identify and quantify the memory, compute, and performance trade-offs for many learned and hand-designeds features.
We leverage our analysis to construct a learned is both faster and more efficient than previous work.
arXiv Detail & Related papers (2022-03-22T16:36:36Z) - Reverse engineering learned optimizers reveals known and novel
mechanisms [50.50540910474342]
Learneds are algorithms that can themselves be trained to solve optimization problems.
Our results help elucidate the previously murky understanding of how learneds work, and establish tools for interpreting future learneds.
arXiv Detail & Related papers (2020-11-04T07:12:43Z) - Tasks, stability, architecture, and compute: Training more effective
learned optimizers, and using them to train themselves [53.37905268850274]
We introduce a new, hierarchical, neural network parameterized, hierarchical with access to additional features such as validation loss to enable automatic regularization.
Most learneds have been trained on only a single task, or a small number of tasks.
We train ours on thousands of tasks, making use of orders of magnitude more compute, resulting in generalizes that perform better to unseen tasks.
arXiv Detail & Related papers (2020-09-23T16:35:09Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.