Skip to content

IJCAI 2019 : Prototype Propagation Networks (PPN) for Weakly-supervised Few-shot Learning on Category Graph

Notifications You must be signed in to change notification settings

liulu112601/Prototype-Propagation-Net

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Prototype-Propagation-Networks

This is the official code for IJCAI 2019 Paper: Prototype Propagation Networks (PPN) for Weakly-supervised Few-shot Learning on Category Graph

We find weakly-labeled data as well as the propagation mechanism improve the performance of few-shot learning a lot.

If you find this project helpful, please consider to cite the following paper:

@inproceedings{liu2019ppn,
title={Prototype Propagation Networks (PPN) for Weakly-supervised Few-shot Learning on Category Graph},
author={Liu, Lu and Zhou, Tianyi and Long, Guodong and Jiang, Jing and Yao, Lina and Zhang, Chengqi},
booktitle={International Joint Conference on Artificial Intelligence (IJCAI)},
year={2019}
}

Dependencies

  • Python 3.6
  • Pytorch 1.0.0

Datasets

  • Download datasets from Google Drive
  • Enter the dir of the downloaded datasets. Extract the datasets by tar -xvf tiered-imagenet-pure.tar --directory ~/datasets/ or tar -xvf tiered-imagenet-mix.tar --directory ~/datasets/

Training

  • For 5 way 1 shot experiment on tiered-imagenet-pure: bash scripts/pp_buffer/train_anc.sh 0 5 1 pure all_level_avg_single 1, where in order 0 is for which GPU to use, 5 is way, 1 is shot, pure is dataset (mix otherwise), all_level_avg_single is training strategy used in our paper, 1 is the number of hops for propagation.

  • For 5 way 5 shot experiment on tiered-imagenet-mix, where each parameter follows the same setup: bash scripts/pp_buffer/train_base_anc.sh 0 5 1 mix all_level_avg_single 1

Testing

  • For 5 way 1 shot experiment on tiered-imagenet-pure: bash scripts/pp_buffer/test_anc_all_level.sh 0 5 1 pure all_level_avg_single 1 SEED, where SEED is the random seed used in training (check the name of the training logs) and the other parameters follow the same setup.

  • For 5 way 5 shot experiment on tiered-imagenet-pure, where each parameter follows the same setup: bash scripts/pp_buffer/test_base_anc_all_level.sh 0 5 1 mix all_level_avg_single 1 SEED

About

IJCAI 2019 : Prototype Propagation Networks (PPN) for Weakly-supervised Few-shot Learning on Category Graph

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published