An Ensemble of Epoch-Wise Empirical Bayes for Few-Shot Learning

By Yaoyao Liu, Bernt Schiele, Qianru Sun. 2019

136 citations recorded by the source corpus.

Few-shot learning aims to train efficient predictive models with a few examples. The lack of training data leads to poor models that perform high-variance or low-confidence predictions. In this paper, we propose to meta-learn the ensemble of epoch-wise empirical Bayes models (E3BM) to achieve robust predictions. "Epoch-wise" means that each training epoch has a Bayes model whose parameters are specifically learned and deployed. "Empirical" means that the hyperparameters, e.g., used for learning and ensembling the epoch-wise models, are generated by hyperprior learners conditional on task-specific data. We introduce four kinds of hyperprior learners by considering inductive vs. transductive, and epoch-dependent vs. epoch-independent, in the paradigm of meta-learning. We conduct extensive experiments for five-class few-shot tasks on three challenging benchmarks: miniImageNet, tieredImageNet, and FC100, and achieve top performance using the epoch-dependent transductive hyperprior learner, which captures the richest information. Our ablation study shows that both "epoch-wise ensemble" and "empirical" encourage high efficiency and robustness in the model performance.

Subfields: Probabilistic & Bayesian Methods, Meta & Continual Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Probabilistic & Bayesian Methods

  1. Xiansheng Hua
  2. Chi Zhang
  3. Guosheng Lin
  4. Yaoyao Liu
  5. Qianru Sun
  6. Bernt Schiele
  7. Hanwang Zhang

In Meta & Continual Learning

  1. Bernt Schiele
  2. Yaoyao Liu
  3. Qianru Sun
  4. Chi Zhang
  5. Guosheng Lin
  6. Zhenyu Weng
  7. Zhiping Lin
  8. Huiping Zhuang

Citations in the corpus