├── README.md ├── __init__.py ├── data ├── __pycache__ │ ├── MNIST.cpython-36.pyc │ ├── MNIST.cpython-37.pyc │ ├── create_dataset.cpython-36.pyc │ ├── create_dataset.cpython-37.pyc │ ├── dataLoader.cpython-36.pyc │ └── dataLoader.cpython-37.pyc ├── create_dataset.py └── dataLoader.py ├── networks ├── EncoderNN.py ├── MAF_flow_multiple.py ├── __init__.py ├── __pycache__ │ ├── EncoderNN.cpython-36.pyc │ ├── EncoderNN.cpython-37.pyc │ ├── MAF_flow_likelihood.cpython-36.pyc │ ├── MAF_flow_likelihood.cpython-37.pyc │ ├── MAF_flow_likelihood_debugging.cpython-36.pyc │ ├── MAF_flow_likelihood_debugging.cpython-37.pyc │ ├── MAF_flow_likelihood_multiple.cpython-36.pyc │ ├── MAF_flow_likelihood_multiple.cpython-37.pyc │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── cGAN_MNIST.cpython-37.pyc │ ├── gcl_torch.cpython-36.pyc │ ├── gcl_torch.cpython-37.pyc │ ├── gcl_torch_multiple.cpython-36.pyc │ └── gcl_torch_multiple.cpython-37.pyc ├── fdv_multiple.py └── gcl_multiple.py ├── pdf ├── NeurIPS-2021-supercharging-imbalanced-data-learning-with-energy-based-contrastive-representation-transfer-Paper.pdf └── workflow.png ├── train_CRT_toy.py ├── trainer ├── FDV_multiple_trainer.py ├── GCL_multiple_trainer.py ├── GCL_trainer.py ├── __init__.py ├── __pycache__ │ ├── GCL_multiple_trainer.cpython-36.pyc │ ├── GCL_multiple_trainer.cpython-37.pyc │ ├── GCL_trainer.cpython-36.pyc │ ├── GCL_trainer.cpython-37.pyc │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── pre_trainer.cpython-36.pyc │ └── pre_trainer.cpython-37.pyc └── pre_trainer.py └── utils ├── Visualization.py ├── __pycache__ ├── Visualization.cpython-37.pyc ├── dataLoader.cpython-36.pyc ├── dataLoader.cpython-37.pyc ├── plot.cpython-36.pyc ├── plot.cpython-37.pyc ├── regularizer.cpython-37.pyc ├── trainer_util.cpython-36.pyc └── trainer_util.cpython-37.pyc ├── plot.py ├── regularizer.py ├── trainer_util.py └── validity_check.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/__pycache__/MNIST.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/MNIST.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/MNIST.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/MNIST.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/create_dataset.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/create_dataset.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/create_dataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/create_dataset.cpython-37.pyc -------------------------------------------------------------------------------- /data/__pycache__/dataLoader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/dataLoader.cpython-36.pyc -------------------------------------------------------------------------------- /data/__pycache__/dataLoader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/__pycache__/dataLoader.cpython-37.pyc -------------------------------------------------------------------------------- /data/create_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/create_dataset.py -------------------------------------------------------------------------------- /data/dataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/data/dataLoader.py -------------------------------------------------------------------------------- /networks/EncoderNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/EncoderNN.py -------------------------------------------------------------------------------- /networks/MAF_flow_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/MAF_flow_multiple.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/__pycache__/EncoderNN.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/EncoderNN.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/EncoderNN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/EncoderNN.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood_debugging.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood_debugging.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood_debugging.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood_debugging.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood_multiple.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood_multiple.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/MAF_flow_likelihood_multiple.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/MAF_flow_likelihood_multiple.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/cGAN_MNIST.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/cGAN_MNIST.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/gcl_torch.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/gcl_torch.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/gcl_torch.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/gcl_torch.cpython-37.pyc -------------------------------------------------------------------------------- /networks/__pycache__/gcl_torch_multiple.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/gcl_torch_multiple.cpython-36.pyc -------------------------------------------------------------------------------- /networks/__pycache__/gcl_torch_multiple.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/__pycache__/gcl_torch_multiple.cpython-37.pyc -------------------------------------------------------------------------------- /networks/fdv_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/fdv_multiple.py -------------------------------------------------------------------------------- /networks/gcl_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/networks/gcl_multiple.py -------------------------------------------------------------------------------- /pdf/NeurIPS-2021-supercharging-imbalanced-data-learning-with-energy-based-contrastive-representation-transfer-Paper.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/pdf/NeurIPS-2021-supercharging-imbalanced-data-learning-with-energy-based-contrastive-representation-transfer-Paper.pdf -------------------------------------------------------------------------------- /pdf/workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/pdf/workflow.png -------------------------------------------------------------------------------- /train_CRT_toy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/train_CRT_toy.py -------------------------------------------------------------------------------- /trainer/FDV_multiple_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/FDV_multiple_trainer.py -------------------------------------------------------------------------------- /trainer/GCL_multiple_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/GCL_multiple_trainer.py -------------------------------------------------------------------------------- /trainer/GCL_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/GCL_trainer.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/__pycache__/GCL_multiple_trainer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/GCL_multiple_trainer.cpython-36.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/GCL_multiple_trainer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/GCL_multiple_trainer.cpython-37.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/GCL_trainer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/GCL_trainer.cpython-36.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/GCL_trainer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/GCL_trainer.cpython-37.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/pre_trainer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/pre_trainer.cpython-36.pyc -------------------------------------------------------------------------------- /trainer/__pycache__/pre_trainer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/__pycache__/pre_trainer.cpython-37.pyc -------------------------------------------------------------------------------- /trainer/pre_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/trainer/pre_trainer.py -------------------------------------------------------------------------------- /utils/Visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/Visualization.py -------------------------------------------------------------------------------- /utils/__pycache__/Visualization.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/Visualization.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dataLoader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/dataLoader.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dataLoader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/dataLoader.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/plot.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/plot.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/plot.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/plot.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/regularizer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/regularizer.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/trainer_util.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/trainer_util.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/trainer_util.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/__pycache__/trainer_util.cpython-37.pyc -------------------------------------------------------------------------------- /utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/plot.py -------------------------------------------------------------------------------- /utils/regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/regularizer.py -------------------------------------------------------------------------------- /utils/trainer_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/trainer_util.py -------------------------------------------------------------------------------- /utils/validity_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZidiXiu/ECRT/HEAD/utils/validity_check.py --------------------------------------------------------------------------------