中文

Sign-MAML:基于 SignSGD 的高效模型无关元学习

机器学习 2021-12-22 v2 人工智能 计算机视觉与模式识别

摘要

我们提出一种计算高效的一阶模型无关元学习(MAML)新算法。关键使能技术是将 MAML 解释为双层优化(BLO)问题,并利用基于符号的 SGD(signSGD)作为 BLO 的底层优化器。我们表明,通过面向 signSGD 的 BLO 视角,MAML 自然产生一种交替优化方案,仅需所学元模型的一阶梯度。我们将所得 MAML 算法称为 Sign-MAML。与常规一阶 MAML(FO-MAML)算法相比,Sign-MAML 具有理论依据,因为它不施加元训练期间不存在二阶导数的任何假设。在实践中,我们表明 Sign-MAML 在各种少样本图像分类任务上优于 FO-MAML,且与 MAML 相比,它在分类精度与计算效率之间实现了更优雅的权衡。

关键词

引用

@article{arxiv.2109.07497,
  title  = {Sign-MAML: Efficient Model-Agnostic Meta-Learning by SignSGD},
  author = {Chen Fan and Parikshit Ram and Sijia Liu},
  journal= {arXiv preprint arXiv:2109.07497},
  year   = {2021}
}