Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. RMAML: Riemannian meta-learning with orthogonality constraints
 
research article

RMAML: Riemannian meta-learning with orthogonality constraints

Tabealhojeh, Hadi
•
Adibi, Peyman
•
Karshenas, Hossein
Show more
April 12, 2023
Pattern Recognition

Meta-learning is the core capability that enables intelligent systems to rapidly generalize their prior ex-perience to learn new tasks. In general, the optimization-based methods formalize the meta-learning as a bi-level optimization problem, that is a nested optimization framework, in which meta-parameters are optimized (or learned) at the outer-level, while the inner-level optimizes the task-specific parameters. In this paper, we introduce RMAML, a meta-learning method that enforces orthogonality constraints to the bi-level optimization problem. We develop a geometry aware framework that generalizes the bi-level optimization problem to the Riemannian (constrained) setting. Using the Riemannian operations such as orthogonal projection, retraction and parallel transport, the bi-level optimization is reformulated so that it respects the Riemannian geometry. Moreover, we observe that a superior stable optimization and an im-proved generalization ability can be achieved when the parameters and meta-parameters of the method are modeled using a Stiefel Manifold. We empirically show that RMAML can easily reach competitive performances against several state of the art algorithms for few-shot classification and consistently out-performs its Euclidean counterpart, MAML. For example, by using the geometry of the Stiefel manifold to structure the fully-connected layers in a deep neural network, a 7% increase in single-domain few-shot classification accuracy is achieved. For the cross-domain few-shot learning, RMAML outperforms MAML by up to 9% of accuracy. Our ablation study also demonstrates the effectiveness of RMAML over MAML in terms of higher accuracy with a reduced number of tasks and (or) inner-level updates.(c) 2023 Elsevier Ltd. All rights reserved.

  • Details
  • Metrics
Type
research article
DOI
10.1016/j.patcog.2023.109563
Web of Science ID

WOS:000984928200001

Author(s)
Tabealhojeh, Hadi
Adibi, Peyman
Karshenas, Hossein
Roy, Soumava Kumar  
Harandi, Mehrtash
Date Issued

2023-04-12

Publisher

ELSEVIER SCI LTD

Published in
Pattern Recognition
Volume

140

Article Number

109563

Subjects

Computer Science, Artificial Intelligence

•

Engineering, Electrical & Electronic

•

Computer Science

•

Engineering

•

meta -learning

•

geometry -aware optimization

•

riemannian manifolds

•

few -shot image classification

•

gradient descent

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
CVLAB  
Available on Infoscience
June 5, 2023
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/197991
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés