Skip to content

Class Incremental Learning (iCaRL, EEIL, BiC) reproduce github repository.

License

Notifications You must be signed in to change notification settings

3neutronstar/CIL-Zoo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

47 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Class Incremental Learning Zoo - pytorch version

Unofficial pytorch version of reproducing the method of class incremental learning (Working) in classification

Currently only CIFAR100 dataset is available.

  • How to Implement (iCaRL)
    python main.py train --train-mode icarl --gpu-ids 0 --model resnet32
  • How to Implement (EEIL, with BiC setting)
    python main.py train --train-mode eeil --gpu-ids 2 --task-size 10 --model resnet32 --batch-size 128 --lr 0.1 --gamma 0.1 --epochs 250 --lr-steps 100,150,200 --weight-decay 0.0002
  • How to Implement (EEIL, with EEIL setting)
    python main.py train --train-mode eeil --gpu-ids 2 --task-size 10 --model resnet32 --batch-size 128 --lr 0.1 --gamma 0.1 --epochs 40 --lr-steps 10,20,30 --weight-decay 0.0001
  • How to Implement (BiC)
    python main.py train --train-mode bic --gpu-ids 2 --task-size 10 --model resnet32 --batch-size 128 --lr 0.1 --gamma 0.1 --epochs 250 --lr-steps 100,150,200 --weight-decay 0.0002
  • 5 task (Top-1 Accuracy)
5 task 1 2 3 4 5
iCaRL 78.35 65.25 54.15 47.94 38.7
EEIL 82.5 72.12 65.75 59.08 54.23
BiC 81.15 71.43 64.28 58.56 54.91

EEIL use BiC settings for reproducing, we provide EEIL settings also.

  • 10 task
10 task 1 2 3 4 5 6 7 8 9 10
iCaRL 80.6 66.7 59.67 52.88 48.36 44.68 41.76 38.19 36.03 33.19
EEIL 87 77.7 73.7 66.5 61.66 58.37 55.27 51.21 48.59 45.47
BiC 87.7 80.4 72.17 66.45 61.92 57.78 54.61 51.4 50.28 47.78
  • 20 task
20 task 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
iCaRL 79.6 73.2 67.8 63.35 63.92 58.93 57.6 53.9 50.51 49 47.22 42.97 41.46 38.04 32.43 30.28 27.13 25.53 23.06 20
EEIL 84.4 83.3 75.4 69.65 70.36 65.6 63.86 56.8 54.29 52.3 50.42 48.32 47.95 45.39 43.12 42.81 40.58 40.46 39.65 36.12
BiC 87.4 85.4 78.47 76.4 73.84 68.87 65.54 61.68 57.67 54.8 53 50.68 51.21 49.62 48.77 45.76 45.29 43.67 41.27 39.64

Reference