CVLGMar 13, 2024

Unleashing the Power of Meta-tuning for Few-shot Generalization Through Sparse Interpolated Experts

arXiv:2403.08477v34 citationsh-index: 14Has CodeICML
Originality Incremental advance
AI Analysis

This addresses the issue of poor generalization in meta-tuning for vision foundation models, offering a solution for researchers and practitioners in few-shot learning, though it appears incremental as it builds on existing meta-tuning and sparse expert methods.

The paper tackled the problem of meta-tuning's underperformance on out-of-distribution tasks in few-shot learning by introducing SMAT, a method that uses sparse mixture-of-experts to isolate parameters for meta-tuning, achieving new state-of-the-art results on a challenging Meta-Dataset with OOD tasks.

Recent successes suggest that parameter-efficient fine-tuning of foundation models as the state-of-the-art method for transfer learning in vision, replacing the rich literature of alternatives such as meta-learning. In trying to harness the best of both worlds, meta-tuning introduces a subsequent optimization stage of foundation models but has so far only shown limited success and crucially tends to underperform on out-of-distribution (OOD) tasks. In this paper, we introduce Sparse MetA-Tuning (SMAT), a method inspired by sparse mixture-of-experts approaches and trained to isolate subsets of pre-trained parameters automatically for meta-tuning on each task. SMAT successfully overcomes OOD sensitivity and delivers on the promise of enhancing the transfer abilities of vision foundation models beyond parameter-efficient fine-tuning. We establish new state-of-the-art results on a challenging combination of Meta-Dataset augmented with additional OOD tasks in both zero-shot and gradient-based adaptation settings. In addition, we provide a thorough analysis of the superiority of learned over hand-designed sparsity patterns for sparse expert methods and the pivotal importance of the sparsity level in balancing between in-distribution and out-of-distribution generalization. Our code is publicly available.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes