ZooKT: Task-adaptive knowledge transfer of Model Zoo for few-shot learning

被引:0
|
作者
Zhang, Baoquan [1 ]
Shan, Bingqi [1 ]
Li, Aoxue [2 ]
Luo, Chuyao [1 ]
Ye, Yunming [1 ]
Li, Zhenguo [2 ]
机构
[1] Harbin Inst Technol, Sch Comp Sci & Technol, Shenzhen 518055, Guangdong, Peoples R China
[2] Noahs Ark Lab, Beijing, Peoples R China
关键词
Few-shot learning; Meta-learning; Model ZOO;
D O I
10.1016/j.patcog.2024.110960
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot learning (FSL) aims to recognize novel classes with few examples. It is challenging since it suffers from a data scarcity issue. Although existing methods have shown superior performance, they neglect some pretrained model priors from various datasets/tasks, which are usually free or cheap to collect from some open source platforms (e.g., Github). Inspired by this, in this paper, we focus on a new FSL setting, namely, Model-Zoo-based FSL (MZFSL), which transfers knowledge learned from not only base classes but also a zoo of deep models (called Model Zoo) pretrained on various tasks/datasets. To fully exploit the Model Zoo avoiding the risk of overfitting, we propose a novel knowledge transfer framework, called ZooKT, which amalgamates knowledge of model zoo, instead of finetuning the model zoo, to learn to extract transferable visual features for novel classes. Specifically, we first regard the model zoo as a prior of feature extractor. Then a meta weight prediction network is designed to leverage the prior to predict the weights of target feature extractor for novel classes. Finally, we integrate the target feature extractor to existing FSL methods for performing novel class prediction. Experimental results on few-shot classification and detection scenarios show that our ZooKT outperforms not only single-model finetuning methods but also state-of-the-art multiple-model transfer learning methods, with comparable inference time with a single model.
引用
收藏
页数:12
相关论文
共 50 条
  • [1] Task-adaptive Few-shot Learning on Sphere Manifold
    He, Xi
    Li, Fanzhang
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 2949 - 2956
  • [2] Learning to Learn Task-Adaptive Hyperparameters for Few-Shot Learning
    Baik, Sungyong
    Choi, Myungsub
    Choi, Janghoon
    Kim, Heewon
    Lee, Kyoung Mu
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2024, 46 (03) : 1441 - 1454
  • [3] Task-adaptive Relation Dependent Network for Few-shot Learning
    He, Xi
    Li, Fanzhang
    Liu, Li
    2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [4] Few-shot classification with task-adaptive semantic feature learning
    Pan, Mei-Hong
    Xin, Hong-Yi
    Xia, Chun-Qiu
    Shen, Hong -Bin
    PATTERN RECOGNITION, 2023, 141
  • [5] Task-Adaptive Few-shot Node Classification
    Wang, Song
    Ding, Kaize
    Zhang, Chuxu
    Chen, Chen
    Li, Jundong
    PROCEEDINGS OF THE 28TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, KDD 2022, 2022, : 1910 - 1919
  • [6] Meta-Learning with Task-Adaptive Loss Function for Few-Shot Learning
    Baik, Sungyong
    Choi, Janghoon
    Kim, Heewon
    Cho, Dohee
    Min, Jaesik
    Lee, Kyoung Mu
    2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 9445 - 9454
  • [7] TapNet: Neural Network Augmented with Task-Adaptive Projection for Few-Shot Learning
    Yoon, Sung Whan
    Seo, Jun
    Moon, Jaekyun
    INTERNATIONAL CONFERENCE ON MACHINE LEARNING, VOL 97, 2019, 97
  • [8] Task-adaptive Label Dependency Transfer for Few-shot Named Entity Recognition
    Zhang, Shan
    Cao, Bin
    Zhang, Tianming
    Liu, Yuqi
    Fan, Jing
    FINDINGS OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS, ACL 2023, 2023, : 3280 - 3293
  • [9] Task-Adaptive Prompted Transformer for Cross-Domain Few-Shot Learning
    Wu, Jiamin
    Liu, Xin
    Yin, Xiaotian
    Zhang, Tianzhu
    Zhang, Yongdong
    THIRTY-EIGHTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 6, 2024, : 6012 - 6020
  • [10] Task-Adaptive Feature Disentanglement and Hallucination for Few-Shot Classification
    Hu, Zixuan
    Shen, Li
    Lai, Shenqi
    Yuan, Chun
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2023, 33 (08) : 3638 - 3648