MAML算法概述
什么是MAML
1. 论文地址:Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
2. 要解决的问题
- 小样本问题
- 模型收敛过慢
3. 算法描述
MAML期望通过训练一组初始化参数
,使得模型透过训练出的初始化参数,未来在少量样本基础上实现快速收敛。该初始化参数 在训练集上未必是最优解,但可以通过训练出的参数在新的任务上快速收敛,找到最优解。
4. V.S. Pre-train
- Pre-train:训练集上的全局最优参数,但放到测试集上未必可以训练出全局最优,可能只会找到局部最优点。
- MAML:在训练集和测试集上未必全局最优参数,但通过少量迭代,便可收敛到全局最优。
算法描述
- 随机初始化一个权重θ
- Setp3 ~ Step10:一个epoch
- 随机采样一个batch的Task
- 遍历所有Task
- 从Support Set中取出一个batch的Task中的Label和Image
- Setp6 ~ Step7:前向传播,计算梯度后反向传播,更新θ′这个权重
- 从Query Set中取出所有Task前向传播,但不更新模型
- Step10:所有Task结束后,计算Loss,计算梯度,更新θ的权重