李宏毅学习笔记44.Meta Learning

介绍

元学习Meta learning = 学习如何去学习Learn to learn
和Life-long方法有所不一样:

方法 区别
Life-long one model for all the tasks
Meta How to learn a new model

李宏毅学习笔记44.Meta Learning
从过去的任务中学习到一些经验,在新的任务上学得更快(不是更好)
李宏毅学习笔记44.Meta Learning

公式输入请参考:在线Latex公式

Meta Learning概念

机器学习:用Training Data训练由我们设计的Learning Algorithm,得到一个最优算法ff^*,可以用来完成相应的任务(猫狗识别)

李宏毅学习笔记44.Meta Learning
也就是:Machine Learning ≈ 根据数据找一个函数 f 的能力
李宏毅学习笔记44.Meta Learning

原学习:用DtrainD_{train}训练由我们设计的F,可以得到一个完成相应任务的ff^*,怎么感觉和上面没什么区别?
李宏毅学习笔记44.Meta Learning
其实不一样,Meta Learning≈ 根据数据找一个找一个函数 f 的函数 F 的能力。
李宏毅学习笔记44.Meta Learning
机器学习中是知道函数f,而是训练函数f的参数;
绪论里面的图:
李宏毅学习笔记44.Meta Learning
元学习是不知道函数f,而是训练函数F找到f(含参数)。
把上图中的function 代换为 F就变成了元学习
李宏毅学习笔记44.Meta Learning

Meta Learning的三板斧

第一板

• Define a set of learning algorithm
以基于GD优化的算法为例,下图中的每一步的gradient gg其实不一样
李宏毅学习笔记44.Meta Learning
对于上图中有很多东西都是我们人来定义的,具体看下图中红色框框,网络构建是我们定义的,初始值是我们定义的,定义的lr不一样,更新的结果也不一样。
红框中如果我们定义不同的东西,实际上就是不同的算法。我们的想法是让机器来帮助我们设计这些东西。
李宏毅学习笔记44.Meta Learning

第二板

Defining the goodness of a function F
在机器学习中衡量一个函数f的好坏是用一组testing data来进行测试,那么要知道一个生成函数的函数F的好坏当然是要准备一把函数来进行测试咯。
李宏毅学习笔记44.Meta Learning
李宏毅学习笔记44.Meta Learning
从这里可以看到机器学习和元学习在数据上有所不一样:
机器学习的数据:
李宏毅学习笔记44.Meta Learning
元学习的数据:
李宏毅学习笔记44.Meta Learning
这里要说明:

  1. 由于元学习有多个任务,每个任务如果有很多数据,那么训练时间会很长很长,因此,元学习中每个任务的数据不会很多,所以元学习也叫few-shot learning,为了和机器学习区分开,训练和测试数据分别叫Support set和Query set。
  2. 和机器学习一样,当我们的元学习中的训练任务很多的时候,我们可以将其中一部分切出来作为验证任务:validation tasks。
  3. 元学习中的testing task可以和training task一样,也可以不一样。

第三板

• Defining the goodness of a function FF
完成了N个任务后,可以计算F的loss
L(F)=n=1NlnL(F)=\sum_{n=1}^Nl^n
我们就是要找到一个FF^*,使得L最小。
F=argminFL(F)F^*=arg\underset{F}{\text{min}}L(F)
找到之后:
李宏毅学习笔记44.Meta Learning

Meta Learning实例:Omniglot

https://github.com/brendenlake/omniglot
• 1623 characters,部分字符:
李宏毅学习笔记44.Meta Learning
• Each has 20 examples
李宏毅学习笔记44.Meta Learning
把这个学习过程看做是:N-ways K-shot classification,意思是 In each training and test tasks, there are N classes, each has K examples.
例如:20 ways 1 shot就是有20个类别,每个类别只有一个example。
李宏毅学习笔记44.Meta Learning
具体做法:
• Split your characters into training and testing characters
• Sample N training characters, sample K examples from each sampled characters → one training task
• Sample N testing characters, sample K examples from each sampled characters → one testing task

Techniques Today

• MAML
• Chelsea Finn, Pieter Abbeel, and Sergey Levine, “Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks”, ICML, 2017
李宏毅学习笔记44.Meta Learning

• Reptile
• Alex Nichol, Joshua Achiam, John Schulman, On FirstOrder Meta-Learning Algorithms, arXiv, 2018
李宏毅学习笔记44.Meta Learning

MAML

MAML主要是关注初始化参数ϕ\phi的选择(所有task的Network Structure都是一样的)。其损失函数为:
L(ϕ)=n=1Nln(θ^n)L(\phi)=\sum_{n=1}^Nl^n(\hat\theta^n)
其中:
θ^n\hat\theta^n: model learned from task n, θ^n\hat\theta^n depends on ϕ\phi
ln(θ^n)l^n(\hat\theta^n):loss of task n on the testing set of task n
李宏毅学习笔记44.Meta Learning
使用Gradient Descent来最小化L(ϕ)L(\phi)
ϕϕηϕL(ϕ)\phi\leftarrow\phi-\eta\triangledown_{\phi}L(\phi)
这里要和transfer learning中的pre-train model的损失函数进行区分:
L(ϕ)=n=1Nln(ϕ)L(\phi)=\sum_{n=1}^Nl^n(\phi)
可以看到transfer learning是用现有的模型去计算Loss(看模型的当前表现)
而MAML是用ϕ\phi训练之后的模型来计算Loss(看模型潜力)
用图形来表示二者的区别吧

MAML vs transfer learning

对于transfer learning,我们寻找在所有task都最好的ϕ\phi,但并不能保证把ϕ\phi拿去训练以后会得到最好的θn\theta^n,例如下图中ϕ\phi在task 2上得到最好的结果,但是拿到task 1上却只能得到一个局部最小值。
李宏毅学习笔记44.Meta Learning
对于MAML,我们不在意 ϕ\phi 在 training task 上表现如何,我们在意用 ϕ\phi 训练出来的θn\theta^n
表现如何,例如下图中的ϕ\phi,在task 1和task 2上目前表现并不是最好的,但是在task 1上,如果顺着左边的黑色箭头梯度下降,最终可以得到θ^1\hat \theta^1;在task 2上,如果顺着右边的黑色箭头梯度下降,最终可以得到θ^2\hat \theta^2。这两个都是最好的结果。
李宏毅学习笔记44.Meta Learning

MAML的trick

李宏毅学习笔记44.Meta Learning
在上面讲的MAML更新参数的过程中,一般只会更新一次:
θ^=ϕϵϕl(ϕ)\hat \theta=\phi-\epsilon\triangledown_{\phi}l(\phi)
原因如下:
• Fast … Fast … Fast …更新次数少,速度就快,因为MAML跑一轮需要几个小时。
• Good to truly train a model with one step. 就用一次更新就可以达到最佳作为目标来训练。
• When using the algorithm, still update many times.实作的时候在测试过程中可以更新个几次,没毛病。
• Few-shot learning has limited data.MAML对应的数据比较少,多次更新容易过拟合。

MAML Toy Example

Each task:
• 给定一个正弦函数 y=asin(x+b)y=a\text{sin}(x+b)作为target function;
• 从正弦函数中采样K个点作为样本;
• 用这K个样本来估计target function。
李宏毅学习笔记44.Meta Learning
Model Pre-training做出的结果如下图所示:由于Model Pre-training是在所有task都最好的初始化ϕ\phi,这里所有的正弦函数叠起来就是一条直线,所以它初始就是直线。
李宏毅学习笔记44.Meta Learning
MAML的结果就很不错:
李宏毅学习笔记44.Meta Learning

Warning of Math

先把手上的公式整理出来,GD更新公式为:
ϕϕηϕL(ϕ)(1)\phi\leftarrow\phi-\eta\triangledown_{\phi}L(\phi)\tag1
其中损失函数为每个任务的ll累加
L(ϕ)=n=1Nln(θ^n)(2)L(\phi)=\sum_{n=1}^Nl^n(\hat\theta^n)\tag2
其中参数θ^n\hat\theta^n的计算公式为一步更新:
θ^=ϕϵϕl(ϕ)(3)\hat \theta=\phi-\epsilon\triangledown_{\phi}l(\phi)\tag3

先来整理计算公式1中的梯度优化项,把公式2代入公式1的梯度优化项中,并且把梯度放到求和函数里面去:
ϕL(ϕ)=ϕn=1Nln(θ^n)=n=1Nϕln(θ^n)\triangledown_{\phi}L(\phi)=\triangledown_{\phi}\sum_{n=1}^Nl^n(\hat\theta^n)=\sum_{n=1}^N\triangledown_{\phi}l^n(\hat\theta^n)
下面来看梯度ϕl(θ^)\triangledown_{\phi}l(\hat\theta)的求法,实际上是对每一项求偏导:
ϕl(θ^)=[l(θ^)/ϕ1l(θ^)/ϕ2l(θ^)/ϕi]\triangledown_{\phi}l(\hat\theta)=\begin{bmatrix}\partial l(\hat\theta)/\partial \phi_1 \\ \partial l(\hat\theta)/\partial \phi_2 \\ \vdots \\ \partial l(\hat\theta)/\partial \phi_i \\ \vdots \end{bmatrix}
初始化参数ϕi\phi_i是通过很多个θ^\hat\theta来影响l(θ^)l(\hat\theta)
李宏毅学习笔记44.Meta Learning
根据链式法则:
l(θ^)ϕi=jl(θ^)θ^jθ^jϕi(4)\cfrac{\partial l(\hat\theta)}{\partial \phi_i}=\sum_j\cfrac{\partial l(\hat\theta)}{\partial \hat\theta_j}\cfrac{\partial \hat\theta_j}{\partial \phi_i}\tag4
上式中l(θ^)θ^j\cfrac{\partial l(\hat\theta)}{\partial \hat\theta_j}很好计算,根据损失函数l的形式直接求即可,例如l如果是交叉熵,就用交叉熵求偏导即可。重点来看后面这项:θ^jϕi\cfrac{\partial \hat\theta_j}{\partial \phi_i}
根据公式3可知,θ^\hat\theta是一个向量,所以我们可以找其中一个分量:θ^j\hat\theta_j,由公式3可得:
θ^j=ϕjϵϕjl(ϕ)=ϕjϵl(ϕ)ϕj(5)\hat\theta_j=\phi_j-\epsilon\triangledown_{\phi_j}l(\phi)=\phi_j-\epsilon\cfrac{\partial l(\phi)}{\partial\phi_j}\tag5
对公式5中求ϕi\phi_i的偏导:
iji\neq j
θ^jϕi=ϵl(ϕ)ϕiϕj\cfrac{\partial \hat\theta_j}{\partial \phi_i}=-\epsilon\cfrac{\partial l(\phi)}{\partial\phi_i\partial\phi_j}
i=ji= j
θ^jϕi=1ϵl(ϕ)ϕiϕj\cfrac{\partial \hat\theta_j}{\partial \phi_i}=1-\epsilon\cfrac{\partial l(\phi)}{\partial\phi_i\partial\phi_j}
算二次偏导很麻烦,原论文提出忽略二次偏导项:
iji\neq j
θ^jϕi=ϵl(ϕ)ϕiϕj0(6)\cfrac{\partial \hat\theta_j}{\partial \phi_i}=-\epsilon\cfrac{\partial l(\phi)}{\partial\phi_i\partial\phi_j}\approx 0\tag6
i=ji= j
θ^jϕi=1ϵl(ϕ)ϕiϕj1(7)\cfrac{\partial \hat\theta_j}{\partial \phi_i}=1-\epsilon\cfrac{\partial l(\phi)}{\partial\phi_i\partial\phi_j}\approx 1\tag7
把公式6和公式7代入公式4,由于当iji\neq j时,θ^jϕi=0\cfrac{\partial \hat\theta_j}{\partial \phi_i}=0,所以求和的时候只用考虑i=ji= j的情况,即公式4可以写为:
l(θ^)ϕi=jl(θ^)θ^jθ^jϕil(θ^)θ^i(8)\cfrac{\partial l(\hat\theta)}{\partial \phi_i}=\sum_j\cfrac{\partial l(\hat\theta)}{\partial \hat\theta_j}\cfrac{\partial \hat\theta_j}{\partial \phi_i}\approx\cfrac{\partial l(\hat\theta)}{\partial \hat\theta_i}\tag8
利用公式8的估计,梯度矩阵就变成了:
ϕl(θ^)=[l(θ^)/ϕ1l(θ^)/ϕ2l(θ^)/ϕi]=[l(θ^)/θ^1l(θ^)/θ^2l(θ^)/θ^i]=θ^l(θ^)\triangledown_{\phi}l(\hat\theta)=\begin{bmatrix}\partial l(\hat\theta)/\partial \phi_1 \\ \partial l(\hat\theta)/\partial \phi_2 \\ \vdots \\ \partial l(\hat\theta)/\partial \phi_i \\ \vdots \end{bmatrix}=\begin{bmatrix}\partial l(\hat\theta)/\partial \hat\theta_1 \\ \partial l(\hat\theta)/\partial \hat\theta_2 \\ \vdots \\ \partial l(\hat\theta)/\partial \hat\theta_i \\ \vdots \end{bmatrix}=\triangledown_{\hat\theta}l(\hat\theta)
最后我们的梯度优化项就变成了:
ϕL(ϕ)=ϕn=1Nln(θ^n)=n=1Nϕln(θ^n)=n=1Nθ^nln(θ^n)\triangledown_{\phi}L(\phi)=\triangledown_{\phi}\sum_{n=1}^Nl^n(\hat\theta^n)=\sum_{n=1}^N\triangledown_{\phi}l^n(\hat\theta^n)=\sum_{n=1}^N\triangledown_{\hat\theta^n}l^n(\hat\theta^n)

MAML – Real Implementation

先要有一个初始化参数,然后把一个任务task看做是一个sample,当然可以用多个任务组成mini-batch,然后做GD,这里不用batch,而是用SGD:
李宏毅学习笔记44.Meta Learning
先取一个任务m(Sample a training task m),然后更新参数得到θ^m\hat\theta^m
李宏毅学习笔记44.Meta Learning
虽然说好只更新一次,但是这里还是更新两次:
李宏毅学习笔记44.Meta Learning
这里我们计算θ^m\hat\theta^m的偏导,取其方向作为ϕ0\phi^0的梯度更新方向:
李宏毅学习笔记44.Meta Learning
这里需要注意,同向的绿色和蓝色箭头不一定等长,因为LR可能不一样。
然后取一个任务n(Sample a training task n)同样用ϕ1\phi^1计算出θ^n\hat\theta^n以及θ^n\hat\theta^n的下一次梯度方向
李宏毅学习笔记44.Meta Learning
取其方向作为ϕ1\phi^1的梯度更新方向:
这里需要注意,同向的黄色和蓝色箭头不一定等长,因为LR可能不一样。
李宏毅学习笔记44.Meta Learning
再次对比transfer learning的Model Pre-training在实现上和MAML有什么不一样:
现有一个初始化参数:
李宏毅学习笔记44.Meta Learning
然后计算θ^m\hat\theta^m
李宏毅学习笔记44.Meta Learning
然后沿着绿色箭头更新ϕ0\phi^0
李宏毅学习笔记44.Meta Learning
然后不断重复:
李宏毅学习笔记44.Meta Learning

MAML 应用:Translation

Meta-Learning for Low-Resource Neural Machine Translation
18 training tasks: 18 different languages translating to English
2 validation tasks: 2 different languages translating to English
实验结果中用的是BLEU来做评估,横轴是数据量,当然数据量越大效果越好。
Baseline是多任务学习。
先看验证集结果,罗马语翻译为英文
李宏毅学习笔记44.Meta Learning
测试任务结果,法语翻译英文
李宏毅学习笔记44.Meta Learning

Reptile(简单介绍)

https://openai.com/blog/reptile/
现有初始化参数ϕ0\phi^0
李宏毅学习笔记44.Meta Learning
取一个任务m(Sample a training task m),Reptile没有规定只能更新一次参数,因此:
李宏毅学习笔记44.Meta Learning
ϕ0\phi^0θ^m\hat\theta^m方向就是ϕ0\phi^0更新的方向:
李宏毅学习笔记44.Meta Learning
计算出ϕ1\phi^1后,取一个任务n(Sample a training task n)同样用ϕ1\phi^1计算出θ^n\hat\theta^n并更新多次,取ϕ1\phi^1θ^n\hat\theta^n的方向作为ϕ1\phi^1的更新方向:
李宏毅学习笔记44.Meta Learning
把pre-train,MAML,Reptile都放在一起看下有什么区别:
下面g1g_1是pre-train的更新方向
g2g_2是MAML的更新方向
g1+g2g_1+g_2是Reptile的更新方向,当然还可以更新更多次
李宏毅学习笔记44.Meta Learning

实验结果比较:
李宏毅学习笔记44.Meta Learning
MAML和Reptile其实不分上下,但是最下面那个蓝色是Transfer learning。

More about Meta Learning

上面讲的MAML和Reptile都是关于用Meta Learning来找初始化参数这个事情,那我们在介绍Meta Learning的时候还有很多红色框框,这些也是可以用Meta Learning来进行研究如何学习的。
不过弹幕提示:只有初始化参数这里可以用GD
下图是用network来设计Architecture & Activation,以及如何更新参数。
李宏毅学习笔记44.Meta Learning
下面套娃预警:
We learn the initialization parameter ϕ\phi by gradient descent.
What is the initialization parameter ϕ0\phi^0 for initialization parameter ϕ\phi?
李宏毅学习笔记44.Meta Learning
How about learning algorithm beyond gradient descent?
李宏毅学习笔记44.Meta Learning