Notes

← Back to home

A collection of fragments of understanding in the pursuit of deeper questions.

Meta-Learning with ANNs - Optimization-Based (Model-Agnostic Meta-Learning)

Optimization-Based ML Deep Learning models learn through backpropagation of gradients. However, the gradient-based optimization is neither designed to cope with a small number of training sample, nor to converge within a small number of optimization steps. Is there a way to adjust the optimization algorithm so that the model can be good at learning with a few examples? This is what optimization-based approach meta-learning algorithms intend for. Look at Model-Agnostic Meta-Learning (MAML) and LSTM Meta-Learner, Reptile for further information (Not covered in this class).

Model-Agnostic Meta-Learning (MAML) This is a fairly general optimization algorithm, compatible with any model that learns through gradient descent. Let's say our model is fθf_{\theta} with parameters θ\theta. Given a task τi\tau_{i} and its associated dataset (D(i)\mathcal{D}^{(i)} train, D(i)\mathcal{D}^{(i)} test), we can update the model parameters by one or more gradient descent steps (the following example only contains one step):

image181

where L(0)\mathcal{L}^{(0)} is the loss computed using the mini data batch with id (0). The above formula only optimizes for one task. To achieve a good generalization across a variety of tasks, we would like to find the optimal θ\theta^{*} so that the task-specific fine-tuning is more efficient. Now, we sample a new data batch with id (1) for updating the meta-objective. The loss, denoted as L(1)\mathcal{L}^{(1)}, depends on the mini batch (1). The superscripts in L(0)\mathcal{L}^{(0)} and L(1)\mathcal{L}^{(1)} only indicate different data batches, and they refer to the same loss objective for the same task.

image182
image184 image185
image183