arXiv · 2006.11764
Gradient-EM Bayesian Meta-learning
Abstract
Bayesian meta-learning enables robust and fast adaptation to new tasks with uncertainty assessment. The key idea behind Bayesian meta-learning is empirical Bayes inference of hierarchical model. In this work, we extend this framework to include a variety of existing methods, before proposing our variant based on gradient-EM algorithm. Our method improves computational efficiency by avoiding back-propagation computation in the meta-update step, which is exhausting for deep neural networks. Furthermore, it provides flexibility to the inner-update optimization procedure by decoupling it from meta-update. Experiments on sinusoidal regression, few-shot image classification, and policy-based reinforcement learning show that our method not only achieves better accuracy with less computation cost, but is also more robust to uncertainty.
Explore related subjects
Keep this discovery
Yayi Zou, Xiaoqi Lu. 2020-06-21. Gradient-EM Bayesian Meta-learning. https://arxiv.org/abs/2006.11764
Cite the original work for its findings. Save a collection to share your selection of sources.