├── .gitignore ├── README.md ├── config ├── cifar10 │ ├── Res_cifar10_imb0005.yaml │ ├── Res_cifar10_imb0005_ETF_DR.yaml │ ├── Res_cifar10_imb001.yaml │ ├── Res_cifar10_imb001_ETF_DR.yaml │ ├── Res_cifar10_imb002.yaml │ └── Res_cifar10_imb002_ETF_DR.yaml ├── cifar100 │ ├── Res_cifar100_imb0005.yaml │ ├── Res_cifar100_imb0005_ETF_DR.yaml │ ├── Res_cifar100_imb001.yaml │ ├── Res_cifar100_imb001_ETF_DR.yaml │ ├── Res_cifar100_imb002.yaml │ └── Res_cifar100_imb002_ETF_DR.yaml ├── imagenet │ ├── Res50_imagenetLT.yaml │ └── Res50_imagenetLT_ETF_DR.yaml ├── stl │ ├── Res_stl_imb0005.yaml │ ├── Res_stl_imb0005_ETF_DR.yaml │ ├── Res_stl_imb001.yaml │ ├── Res_stl_imb001_ETF_DR.yaml │ ├── Res_stl_imb002.yaml │ └── Res_stl_imb002_ETF_DR.yaml └── svhn │ ├── Res_svhn_imb0005.yaml │ ├── Res_svhn_imb0005_ETF_DR.yaml │ ├── Res_svhn_imb001.yaml │ ├── Res_svhn_imb001_ETF_DR.yaml │ ├── Res_svhn_imb002.yaml │ └── Res_svhn_imb002_ETF_DR.yaml ├── datasets ├── cifar10.py ├── cifar100.py ├── data_txt │ ├── ImageNet_LT_test.txt │ └── ImageNet_LT_train.txt ├── imagenet.py ├── sampler.py ├── stl10.py └── svhn.py ├── main_train.sh ├── methods.py ├── models ├── densenet.py ├── resnet.py └── resnet_cifar.py ├── train.py └── utils ├── __init__.py ├── logger.py ├── meter.py └── metric.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/README.md -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb0005.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb0005.yaml -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb0005_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb0005_ETF_DR.yaml -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb001.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb001.yaml -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb001_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb001_ETF_DR.yaml -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb002.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb002.yaml -------------------------------------------------------------------------------- /config/cifar10/Res_cifar10_imb002_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar10/Res_cifar10_imb002_ETF_DR.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb0005.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb0005.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb0005_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb0005_ETF_DR.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb001.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb001.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb001_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb001_ETF_DR.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb002.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb002.yaml -------------------------------------------------------------------------------- /config/cifar100/Res_cifar100_imb002_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/cifar100/Res_cifar100_imb002_ETF_DR.yaml -------------------------------------------------------------------------------- /config/imagenet/Res50_imagenetLT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/imagenet/Res50_imagenetLT.yaml -------------------------------------------------------------------------------- /config/imagenet/Res50_imagenetLT_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/imagenet/Res50_imagenetLT_ETF_DR.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb0005.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb0005.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb0005_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb0005_ETF_DR.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb001.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb001.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb001_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb001_ETF_DR.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb002.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb002.yaml -------------------------------------------------------------------------------- /config/stl/Res_stl_imb002_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/stl/Res_stl_imb002_ETF_DR.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb0005.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb0005.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb0005_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb0005_ETF_DR.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb001.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb001.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb001_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb001_ETF_DR.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb002.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb002.yaml -------------------------------------------------------------------------------- /config/svhn/Res_svhn_imb002_ETF_DR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/config/svhn/Res_svhn_imb002_ETF_DR.yaml -------------------------------------------------------------------------------- /datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/cifar10.py -------------------------------------------------------------------------------- /datasets/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/cifar100.py -------------------------------------------------------------------------------- /datasets/data_txt/ImageNet_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/data_txt/ImageNet_LT_test.txt -------------------------------------------------------------------------------- /datasets/data_txt/ImageNet_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/data_txt/ImageNet_LT_train.txt -------------------------------------------------------------------------------- /datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/imagenet.py -------------------------------------------------------------------------------- /datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/sampler.py -------------------------------------------------------------------------------- /datasets/stl10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/stl10.py -------------------------------------------------------------------------------- /datasets/svhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/datasets/svhn.py -------------------------------------------------------------------------------- /main_train.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=$1 python ./train.py --cfg $2 -------------------------------------------------------------------------------- /methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/methods.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/models/resnet_cifar.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NeuralCollapseApplications/ImbalancedLearning/HEAD/utils/metric.py --------------------------------------------------------------------------------