├── .gitignore ├── README.md ├── cmds ├── execute_hist │ └── camera_ready.sh └── shell_scrips │ ├── cifar-10-LT.sh │ ├── cifar-100-LT.sh │ ├── imagenet-100-res50-LT.sh │ └── imagenet-res50-LT.sh ├── data ├── LT_Dataset.py ├── augmentation.py ├── cifar10.py ├── cifar100.py └── imagenet_extract.sh ├── exp_analyse.py ├── imgs └── pipeline.png ├── models ├── __init__.py ├── meta_net │ ├── __init__.py │ ├── meta.py │ └── meta_resnet.py ├── resnet.py ├── resnet_prune.py ├── resnet_prune_multibn.py ├── resnet_s_cifar.py ├── resnet_s_cifar_prune.py ├── resnet_s_cifar_prune_multibn.py ├── ride_resnet_s_cifar.py ├── ride_resnet_s_cifar_prune_multibn.py └── utils.py ├── optimizer └── lars.py ├── prune ├── mask.py └── prune_simCLR.py ├── split ├── ImageNet_LT │ ├── ImageNet_BL_train.txt │ ├── ImageNet_LT_test.txt │ ├── ImageNet_LT_train.txt │ ├── ImageNet_LT_val.txt │ ├── ImageNet_train.txt │ ├── imageNet_BL_exp_train.txt │ ├── imageNet_LT_exp_train.txt │ └── imageNet_sub_balance_train_0.01.txt ├── cifar10 │ ├── trainIdxList.npy │ └── valIdxList.npy ├── cifar100 │ ├── cifar100_trainIdxList.npy │ └── cifar100_valIdxList.npy ├── cifar100_imbSub_with_subsets │ ├── cifar100_split1_D_b.npy │ ├── cifar100_split1_D_i.npy │ ├── cifar100_split1_S_b.npy │ ├── cifar100_split2_D_b.npy │ ├── cifar100_split2_D_i.npy │ ├── cifar100_split2_S_b.npy │ ├── cifar100_split3_D_b.npy │ ├── cifar100_split3_D_i.npy │ ├── cifar100_split3_S_b.npy │ ├── cifar100_split4_D_b.npy │ ├── cifar100_split4_D_i.npy │ ├── cifar100_split4_S_b.npy │ ├── cifar100_split5_D_b.npy │ ├── cifar100_split5_D_i.npy │ └── cifar100_split5_S_b.npy ├── cifar10_imbSub_with_subsets │ ├── split1_D_b.npy │ ├── split1_D_i.npy │ ├── split1_S_b.npy │ ├── split2_D_b.npy │ ├── split2_D_i.npy │ ├── split2_S_b.npy │ ├── split3_D_b.npy │ ├── split3_D_i.npy │ ├── split3_S_b.npy │ ├── split4_D_b.npy │ ├── split4_D_i.npy │ ├── split4_S_b.npy │ ├── split5_D_b.npy │ ├── split5_D_i.npy │ └── split5_S_b.npy └── imagenet-100 │ ├── ImageNet_100_test.txt │ ├── ImageNet_100_train.txt │ ├── ImageNet_100_val.txt │ ├── imageNet_100_BL_train.txt │ ├── imageNet_100_LT_train.txt │ └── imageNet_100_sub_balance_train_0.01.txt ├── train_cifar.py ├── train_imagenet.py ├── train_simCLR.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea 2 | *.pyc 3 | /analyse/TSNE_Draw/featureList/ 4 | *.slurm 5 | checkpoints_tune/ 6 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/README.md -------------------------------------------------------------------------------- /cmds/execute_hist/camera_ready.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/cmds/execute_hist/camera_ready.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/cifar-10-LT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/cmds/shell_scrips/cifar-10-LT.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/cifar-100-LT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/cmds/shell_scrips/cifar-100-LT.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/imagenet-100-res50-LT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/cmds/shell_scrips/imagenet-100-res50-LT.sh -------------------------------------------------------------------------------- /cmds/shell_scrips/imagenet-res50-LT.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/cmds/shell_scrips/imagenet-res50-LT.sh -------------------------------------------------------------------------------- /data/LT_Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/data/LT_Dataset.py -------------------------------------------------------------------------------- /data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/data/augmentation.py -------------------------------------------------------------------------------- /data/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/data/cifar10.py -------------------------------------------------------------------------------- /data/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/data/cifar100.py -------------------------------------------------------------------------------- /data/imagenet_extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/data/imagenet_extract.sh -------------------------------------------------------------------------------- /exp_analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/exp_analyse.py -------------------------------------------------------------------------------- /imgs/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/imgs/pipeline.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/meta_net/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/meta_net/meta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/meta_net/meta.py -------------------------------------------------------------------------------- /models/meta_net/meta_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/meta_net/meta_resnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet_prune.py -------------------------------------------------------------------------------- /models/resnet_prune_multibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet_prune_multibn.py -------------------------------------------------------------------------------- /models/resnet_s_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet_s_cifar.py -------------------------------------------------------------------------------- /models/resnet_s_cifar_prune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet_s_cifar_prune.py -------------------------------------------------------------------------------- /models/resnet_s_cifar_prune_multibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/resnet_s_cifar_prune_multibn.py -------------------------------------------------------------------------------- /models/ride_resnet_s_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/ride_resnet_s_cifar.py -------------------------------------------------------------------------------- /models/ride_resnet_s_cifar_prune_multibn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/ride_resnet_s_cifar_prune_multibn.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/models/utils.py -------------------------------------------------------------------------------- /optimizer/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/optimizer/lars.py -------------------------------------------------------------------------------- /prune/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/prune/mask.py -------------------------------------------------------------------------------- /prune/prune_simCLR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/prune/prune_simCLR.py -------------------------------------------------------------------------------- /split/ImageNet_LT/ImageNet_BL_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/ImageNet_BL_train.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/ImageNet_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/ImageNet_LT_test.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/ImageNet_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/ImageNet_LT_train.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/ImageNet_LT_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/ImageNet_LT_val.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/ImageNet_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/ImageNet_train.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/imageNet_BL_exp_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/imageNet_BL_exp_train.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/imageNet_LT_exp_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/imageNet_LT_exp_train.txt -------------------------------------------------------------------------------- /split/ImageNet_LT/imageNet_sub_balance_train_0.01.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/ImageNet_LT/imageNet_sub_balance_train_0.01.txt -------------------------------------------------------------------------------- /split/cifar10/trainIdxList.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10/trainIdxList.npy -------------------------------------------------------------------------------- /split/cifar10/valIdxList.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10/valIdxList.npy -------------------------------------------------------------------------------- /split/cifar100/cifar100_trainIdxList.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100/cifar100_trainIdxList.npy -------------------------------------------------------------------------------- /split/cifar100/cifar100_valIdxList.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100/cifar100_valIdxList.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split1_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split1_D_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split1_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split1_D_i.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split1_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split1_S_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split2_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split2_D_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split2_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split2_D_i.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split2_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split2_S_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split3_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split3_D_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split3_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split3_D_i.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split3_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split3_S_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split4_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split4_D_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split4_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split4_D_i.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split4_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split4_S_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split5_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split5_D_b.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split5_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split5_D_i.npy -------------------------------------------------------------------------------- /split/cifar100_imbSub_with_subsets/cifar100_split5_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar100_imbSub_with_subsets/cifar100_split5_S_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split1_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split1_D_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split1_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split1_D_i.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split1_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split1_S_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split2_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split2_D_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split2_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split2_D_i.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split2_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split2_S_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split3_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split3_D_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split3_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split3_D_i.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split3_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split3_S_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split4_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split4_D_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split4_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split4_D_i.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split4_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split4_S_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split5_D_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split5_D_b.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split5_D_i.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split5_D_i.npy -------------------------------------------------------------------------------- /split/cifar10_imbSub_with_subsets/split5_S_b.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/cifar10_imbSub_with_subsets/split5_S_b.npy -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/ImageNet_100_test.txt -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/ImageNet_100_train.txt -------------------------------------------------------------------------------- /split/imagenet-100/ImageNet_100_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/ImageNet_100_val.txt -------------------------------------------------------------------------------- /split/imagenet-100/imageNet_100_BL_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/imageNet_100_BL_train.txt -------------------------------------------------------------------------------- /split/imagenet-100/imageNet_100_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/imageNet_100_LT_train.txt -------------------------------------------------------------------------------- /split/imagenet-100/imageNet_100_sub_balance_train_0.01.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/split/imagenet-100/imageNet_100_sub_balance_train_0.01.txt -------------------------------------------------------------------------------- /train_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/train_cifar.py -------------------------------------------------------------------------------- /train_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/train_imagenet.py -------------------------------------------------------------------------------- /train_simCLR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/train_simCLR.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VITA-Group/SDCLR/HEAD/utils.py --------------------------------------------------------------------------------