arXiv · 1910.07368
Model-Agnostic Meta-Learning using Runge-Kutta Methods
Abstract
Meta-learning has emerged as an important framework for learning new tasks from just a few examples. The success of any meta-learning model depends on (i) its fast adaptation to new tasks, as well as (ii) having a shared representation across similar tasks. Here we extend the model-agnostic meta-learning (MAML) framework introduced by Finn et al. (2017) to achieve improved performance by analyzing the temporal dynamics of the optimization procedure via the Runge-Kutta method. This method enables us to gain fine-grained control over the optimization and helps us achieve both the adaptation and representation goals across tasks. By leveraging this refined control, we demonstrate that there are multiple principled ways to update MAML and show that the classic MAML optimization is simply a special case of second-order Runge-Kutta method that mainly focuses on fast-adaptation. Experiments on benchmark classification, regression and reinforcement learning tasks show that this refined control helps attain improved results.
Explore related subjects
Keep this discovery
Daniel Jiwoong Im, Yibo Jiang, Nakul Verma. 2019-10-16. Model-Agnostic Meta-Learning using Runge-Kutta Methods. https://arxiv.org/abs/1910.07368
Cite the original work for its findings. Save a collection to share your selection of sources.