LibFewShot: A Comprehensive Library for Few-Shot Learning

Wenbin Li, Ziyi Wang, Xuesong Yang, Chuanqi Dong, Pinzhuo Tian, Tiexin Qin, Jing Huo, Yinghuan Shi, Lei Wang, Yang Gao*, Jiebo Luo

*Corresponding author for this work

Research output: Journal Publications and ReviewsRGC 21 - Publication in refereed journalpeer-review

49 Citations (Scopus)

Abstract

Few-shot learning, especially few-shot image classification, has received increasing attention and witnessed significant advances in recent years. Some recent studies implicitly show that many generic techniques or “tricks”, such as data augmentation, pre-training, knowledge distillation, and self-supervision, may greatly boost the performance of a few-shot learning method. Moreover, different works may employ different software platforms, backbone architectures and input image sizes, making fair comparisons difficult and practitioners struggle with reproducibility. To address these situations, we propose a comprehensive library for few-shot learning (LibFewShot) by re-implementing eighteen state-of-the-art few-shot learning methods in a unified framework with the same single codebase in PyTorch. Furthermore, based on LibFewShot, we provide comprehensive evaluations on multiple benchmarks with various backbone architectures to evaluate common pitfalls and effects of different training tricks. In addition, with respect to the recent doubts on the necessity of meta- or episodic-training mechanism, our evaluation results confirm that such a mechanism is still necessary especially when combined with pre-training. We hope our work can not only lower the barriers for beginners to enter the area of few-shot learning but also elucidate the effects of nontrivial tricks to facilitate intrinsic research on few-shot learning.

© 2023 IEEE. Personal use is permitted, but republication/redistribution requires IEEE permission.
Original languageEnglish
Pages (from-to)14938-14955
JournalIEEE Transactions on Pattern Analysis and Machine Intelligence
Volume45
Issue number12
Online published5 Sept 2023
DOIs
Publication statusPublished - Dec 2023

Research Keywords

  • Benchmark testing
  • Deep learning
  • Fair comparison
  • few-shot learning
  • image classification
  • Libraries
  • Metalearning
  • Software
  • Task analysis
  • Training
  • unified framework

Fingerprint

Dive into the research topics of 'LibFewShot: A Comprehensive Library for Few-Shot Learning'. Together they form a unique fingerprint.

Cite this