Skip to main navigation Skip to search Skip to main content

Model-agnostic multi-stage loss optimization meta learning

  • Xiao Yao*
  • , Jianlong Zhu
  • , Guanying Huo*
  • , Ning Xu
  • , Xiaofeng Liu
  • , Ce Zhang
  • *Corresponding author for this work
  • Hohai University Changzhou

Research output: Contribution to journalArticlepeer-review

Abstract

Model Agnostic Meta Learning (MAML) has become the most representative meta learning algorithm to solve few-shot learning problems. This paper mainly discusses MAML framework, focusing on the key problem of solving few-shot learning through meta learning. However, MAML is sensitive to the base model for the inner loop, and training instability occur during the training process, resulting in an increase of the training difficulty of the model in the process of training and verification process, causing degradation of model performance. In order to solve these problems, we propose a multi-stage loss optimization meta-learning algorithm. By discussing a learning mechanism for inner and outer loops, it improves the training stability and accelerates the convergence for the model. The generalization ability of MAML has been enhanced.

Original languageEnglish
Pages (from-to)2349-2363
Number of pages15
JournalInternational Journal of Machine Learning and Cybernetics
Volume12
Issue number8
DOIs
StatePublished - Aug 2021
Externally publishedYes

Keywords

  • Few-shot learning
  • Meta learning
  • Multi-stage loss optimization
  • Training instability

Cite this