Task Affinity with Maximum Bipartite Matching in Few-Shot LearningDownload PDF

29 Sept 2021, 00:31 (edited 04 May 2022)ICLR 2022 PosterReaders: Everyone
  • Keywords: Task Affinity, Transfer Learning, Few-Shot Learning
  • Abstract: We propose an asymmetric affinity score for representing the complexity of utilizing the knowledge of one task for learning another one. Our method is based on the maximum bipartite matching algorithm and utilizes the Fisher Information matrix. We provide theoretical analyses demonstrating that the proposed score is mathematically well-defined, and subsequently use the affinity score to propose a novel algorithm for the few-shot learning problem. In particular, using this score, we find relevant training data labels to the test data and leverage the discovered relevant data for episodically fine-tuning a few-shot model. Results on various few-shot benchmark datasets demonstrate the efficacy of the proposed approach by improving the classification accuracy over the state-of-the-art methods even when using smaller models.
  • One-sentence Summary: Task affinity and its application in few-shot learning
7 Replies