Task Affinity with Maximum Bipartite Matching in Few-Shot Learning

Cat Phuoc Le, Juncheng Dong, Mohammadreza Soltani, Vahid Tarokh

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